From e580741d3801dde56e97f714d6ef4218893396d3 Mon Sep 17 00:00:00 2001 From: Eric Jolibois Date: Wed, 17 Jul 2024 15:02:13 +0200 Subject: [PATCH] test: rewrite a bit --- python/tests/test_fastexcel.py | 15 ++++++--------- 1 file changed, 6 insertions(+), 9 deletions(-) diff --git a/python/tests/test_fastexcel.py b/python/tests/test_fastexcel.py index 6ffb471..28b5b0e 100644 --- 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))