test: rewrite a bit

ToucanToco · Jul 17, 2024 · e580741 · e580741
1 parent b2ee363
commit e580741
Showing 1 changed file with 6 additions and 9 deletions.
diff --git a/python/tests/test_fastexcel.py b/python/tests/test_fastexcel.py
@@ -584,12 +584,9 @@ def test_null_values_in_cells() -> None:
 
 def test_bool_casting_to_string_for_polars() -> None:
     excel_reader = fastexcel.read_excel(path_for_fixture("sheet-bool.xlsx"))
-
-    actual_polars_df = excel_reader.load_sheet(0, header_row=None, column_names=["0"]).to_polars()
-    expected_polars_df = pl.DataFrame(
-        {
-            "0": ["true", "false", "some string"],
-        }
-    )
-
-    pl_assert_frame_equal(actual_polars_df, expected_polars_df)
+    sheet = excel_reader.load_sheet(0, column_names=["col1"])
+    expected = {
+        "col1": ["true", "false", "some string"],
+    }
+    pl_assert_frame_equal(sheet.to_polars(), pl.DataFrame(expected))
+    pd_assert_frame_equal(sheet.to_pandas(), pd.DataFrame(expected))