diff --git a/tests/ocr/doctr/test_doctr.py b/tests/ocr/doctr/test_doctr.py index b19103a..5f5afbb 100644 --- a/tests/ocr/doctr/test_doctr.py +++ b/tests/ocr/doctr/test_doctr.py @@ -58,4 +58,4 @@ def test_doctr_document(): expected = OCRDataframe(df=pl.read_csv("test_data/ocr_df.csv", separator=";").lazy()) - assert result.df.collect().frame_equal(expected.df.collect()) + assert result.df.collect().equals(expected.df.collect()) diff --git a/tests/ocr/easyocr/test_easyocr.py b/tests/ocr/easyocr/test_easyocr.py index 49cc78e..cc0611e 100644 --- a/tests/ocr/easyocr/test_easyocr.py +++ b/tests/ocr/easyocr/test_easyocr.py @@ -65,4 +65,4 @@ def test_easyocr_document(): expected = OCRDataframe(df=pl.read_csv("test_data/ocr_df.csv", separator=";").lazy()) - assert result.df.drop("confidence").collect(streaming=True).frame_equal(expected.df.drop("confidence").collect(streaming=True)) + assert result.df.drop("confidence").collect(streaming=True).equals(expected.df.drop("confidence").collect(streaming=True)) diff --git a/tests/tables/processing/bordered_tables/cells/test_deduplication_cells.py b/tests/tables/processing/bordered_tables/cells/test_deduplication_cells.py index 8bdc815..9352a4e 100644 --- a/tests/tables/processing/bordered_tables/cells/test_deduplication_cells.py +++ b/tests/tables/processing/bordered_tables/cells/test_deduplication_cells.py @@ -10,4 +10,4 @@ def test_deduplicate_cells(): result = deduplicate_cells(df_cells=df_cells).collect() expected = pl.read_csv("test_data/expected.csv", separator=";", encoding="utf-8") - assert result.frame_equal(expected) + assert result.equals(expected) diff --git a/tests/tables/processing/bordered_tables/cells/test_identification_cells.py b/tests/tables/processing/bordered_tables/cells/test_identification_cells.py index 30d9dab..8230d2a 100644 --- a/tests/tables/processing/bordered_tables/cells/test_identification_cells.py +++ b/tests/tables/processing/bordered_tables/cells/test_identification_cells.py @@ -19,7 +19,7 @@ def test_get_potential_cells_from_h_lines(): expected = pl.read_csv("test_data/expected_potential_cells.csv", separator=";", encoding="utf-8") - assert result.frame_equal(expected) + assert result.equals(expected) def test_get_cells_dataframe(): @@ -32,4 +32,4 @@ def test_get_cells_dataframe(): vertical_lines=v_lines).collect() expected = pl.read_csv("test_data/expected_ident_cells.csv", separator=";", encoding="utf-8") - assert result.frame_equal(expected.sort(['x1', 'y1', 'x2', 'y2'])) + assert result.equals(expected.sort(['x1', 'y1', 'x2', 'y2']))