Fix deprecations

xavctn · Dec 16, 2023 · f19f89b · f19f89b
1 parent 05d718b
commit f19f89b
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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']))