From 6552fa9ab3bf1698bc6247f86e9ca23f88752156 Mon Sep 17 00:00:00 2001 From: Andrew Gazelka Date: Thu, 19 Dec 2024 10:23:09 -0800 Subject: [PATCH] fix schema tests --- tests/connect/test_print_schema.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/connect/test_print_schema.py b/tests/connect/test_print_schema.py index 23d2a8693a..c1a9f3aa9c 100644 --- 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