address comments

snowflakedb · Oct 27, 2024 · 27e0fe2 · 27e0fe2
1 parent 03eb621
commit 27e0fe2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/integ/scala/test_datatype_suite.py b/tests/integ/scala/test_datatype_suite.py
@@ -559,7 +559,8 @@ def test_struct_dtype_iceberg_lqb(
         union_df = union_df.select(
             array_sort("ARR", sort_ascending=False).alias("ARR"), "MAP", "A", "B"
         )
-        queries = union_df.queries
+
+        assert union_df.schema == expected_schema
 
         union_df.write.save_as_table(
             write_table,
@@ -568,6 +569,7 @@ def test_struct_dtype_iceberg_lqb(
             iceberg_config=ICEBERG_CONFIG,
         )
 
+        queries = union_df.queries
         # assert that the queries are broken down into 2 queries and 1 post action
         assert len(queries["queries"]) == 2, queries["queries"]
         assert len(queries["post_actions"]) == 1