fix metadata tests for polars > 0.18

danielgafni · Sep 7, 2023 · c6de3c8 · c6de3c8
1 parent 72e4418
commit c6de3c8
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 12 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/tests/test_upath_io_managers.py b/tests/test_upath_io_managers.py
@@ -62,8 +62,10 @@ def upstream() -> pl.DataFrame:
     from packaging.version import Version
 
     if Version(pl.__version__) >= Version("0.18.0"):
-        expected_stats["a"]["50%"] = expected_stats["a"].pop("median")
-        expected_stats["b"]["50%"] = expected_stats["b"].pop("median")
+        expected_stats["a"].pop("median")
+        expected_stats["a"]["50%"] = 1.0
+        expected_stats["b"].pop("median")
+        expected_stats["b"]["50%"] = "null"
 
     assert DeepDiff(stats, expected_stats) == {}