Update test_types.py

AngersZhuuuu · May 27, 2024 · c860edd · c860edd
1 parent cba05ff
commit c860edd
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/python/pyspark/sql/tests/test_types.py b/python/pyspark/sql/tests/test_types.py
@@ -251,7 +251,10 @@ def __init__(self):
                 df = self.spark.createDataFrame(data)
                 self.assertEqual(list(df.schema)[0].dataType.simpleString(), "timestamp_ntz")
                 self.assertEqual(df.first()[0], datetime.datetime(1970, 1, 1, 0, 0))
-                self.assertEqual(list(df.schema)[1].dataType.simpleString(), "timestamp_ntz")
+                self.assertEqual(
+                    list(df.schema)[1].dataType.simpleString(),
+                    "struct<a:struct<a:timestamp_ntz>>"
+                )
                 self.assertEqual(df.first()[1], Row(a=Row(a=datetime.datetime(1970, 1, 1, 0, 0))))
 
             df = self.spark.createDataFrame(