fix schema tests

Eventual-Inc · Dec 19, 2024 · 6552fa9 · 6552fa9
1 parent 93e9d9f
commit 6552fa9
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/connect/test_print_schema.py b/tests/connect/test_print_schema.py
@@ -6,7 +6,7 @@ def test_print_schema_range(spark_session, capsys) -> None:
     df.printSchema()
 
     captured = capsys.readouterr()
-    expected = "root\n" " |-- id: integer (nullable = true)\n\n"
+    expected = "root\n" " |-- id: long (nullable = true)\n\n"
     assert captured.out == expected
 
 
@@ -16,7 +16,7 @@ def test_print_schema_simple_df(spark_session, capsys) -> None:
     df.printSchema()
 
     captured = capsys.readouterr()
-    expected = "root\n" " |-- value: integer (nullable = true)\n\n"
+    expected = "root\n" " |-- value: long (nullable = true)\n\n"
     assert captured.out == expected
 
 
@@ -28,7 +28,7 @@ def test_print_schema_multiple_columns(spark_session, capsys) -> None:
     captured = capsys.readouterr()
     expected = (
         "root\n"
-        " |-- id: integer (nullable = true)\n"
+        " |-- id: long (nullable = true)\n"
         " |-- name: string (nullable = true)\n"
         " |-- flag: boolean (nullable = true)\n\n"
     )
@@ -51,5 +51,5 @@ def test_print_schema_with_nulls(spark_session, capsys) -> None:
     df.printSchema()
 
     captured = capsys.readouterr()
-    expected = "root\n" " |-- id: integer (nullable = true)\n" " |-- value: string (nullable = true)\n\n"
+    expected = "root\n" " |-- id: long (nullable = true)\n" " |-- value: string (nullable = true)\n\n"
     assert captured.out == expected