diff --git a/python/tests/test_fastexcel.py b/python/tests/test_fastexcel.py index 6ffb471..aebde70 100644 --- a/python/tests/test_fastexcel.py +++ b/python/tests/test_fastexcel.py @@ -584,12 +584,13 @@ 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"], - } + 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), ) - - pl_assert_frame_equal(actual_polars_df, expected_polars_df) + pd_expected = pd.DataFrame(expected) + pd_assert_frame_equal(sheet.to_pandas(), pd_expected)