From 27e0fe27ad5cd52c361f10a207f867c9d7e7b38b Mon Sep 17 00:00:00 2001
From: Afroz Alam <afroz.alam@snowflake.com>
Date: Sat, 26 Oct 2024 19:11:21 -0700
Subject: [PATCH] address comments

---
 tests/integ/scala/test_datatype_suite.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tests/integ/scala/test_datatype_suite.py b/tests/integ/scala/test_datatype_suite.py
index ae704cb0ea..c4eb5e0aaf 100644
--- 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