test fix

microsoft · Mar 11, 2021 · e1dee72 · e1dee72
1 parent 2089026
commit e1dee72
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/test/scala/com/microsoft/hyperspace/index/DeltaLakeIntegrationTest.scala b/src/test/scala/com/microsoft/hyperspace/index/DeltaLakeIntegrationTest.scala
@@ -242,12 +242,14 @@ class DeltaLakeIntegrationTest extends QueryTest with HyperspaceSuite {
             spark.disableHyperspace()
             val dfWithHyperspaceDisabled = query()
             val basePlan = dfWithHyperspaceDisabled.queryExecution.optimizedPlan
-            val resultDisabled = dfWithHyperspaceDisabled.collect().toSeq.toSet
+            val resultDisabled = dfWithHyperspaceDisabled.collect().toSeq.sortBy(_.hashCode())
             spark.enableHyperspace()
             val dfWithHyperspaceEnabled = query()
-            assert(!basePlan.equals(dfWithHyperspaceEnabled.queryExecution.optimizedPlan))
-            val resultEnabled = dfWithHyperspaceEnabled.collect().toSeq.toSet
+            val updatedPlan = dfWithHyperspaceEnabled.queryExecution.optimizedPlan
+            val resultEnabled = dfWithHyperspaceEnabled.collect().toSeq.sortBy(_.hashCode())
+            assert(!basePlan.equals(updatedPlan))
             assert(resultEnabled.equals(resultDisabled))
+            checkAnswer(dfWithHyperspaceDisabled, dfWithHyperspaceEnabled)
           }
         }
       }