rm pandas dependency in test

delta-io · Mar 25, 2024 · 8f71a79 · 8f71a79
1 parent f6c476b
commit 8f71a79
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/python/tests/test_writer.py b/python/tests/test_writer.py
@@ -1508,7 +1508,9 @@ def test_rust_decimal_cast(tmp_path: pathlib.Path):
 
     write_deltalake(tmp_path, data, mode="append", engine="rust")
 
-    assert DeltaTable(tmp_path).to_pandas()["x"][0] == Decimal("100.1")
+    assert DeltaTable("test-table").to_pyarrow_table()["x"][0].as_py() == Decimal(
+        "100.1"
+    )
 
     # Write smaller decimal,  works since it's fits in the previous decimal precision, scale
     data = pa.table({"x": pa.array([Decimal("10.1")])})