Sort test result (#1081)

snowflakedb · Oct 9, 2023 · ff2a386 · ff2a386
1 parent eceeb36
commit ff2a386
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tests/integ/scala/test_complex_dataframe_suite.py b/tests/integ/scala/test_complex_dataframe_suite.py
@@ -69,8 +69,12 @@ def test_combination_of_multiple_operators_with_filters(session):
 
     assert df1.filter(col("a") < 6).join(df2, ["a"], "left_anti").collect() == []
 
-    df = df1.filter(col("a") < 6).join(df2, "a").union(df2.filter(col("a") > 5))
-    # don't sort
+    df = (
+        df1.filter(col("a") < 6)
+        .join(df2, "a")
+        .union(df2.filter(col("a") > 5))
+        .sort("a")
+    )
     assert df.collect() == [Row(i, f"test{i}") for i in range(1, 11)]