Merge pull request #321 from kayjan/feature/new-polars

Fix unit test to work with new polars

Author: kayjan

CHANGELOG.md

@@ -15,6 +15,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - Misc: Documentation to include tips and tricks on working with custom classes.
 - Tree Export: Mermaid diagram title to add newline.
 - Tree Helper: Get tree diff string replacement bug when the path change is substring of another path.
+- Tree Export: Polars test to work with old (<=1.9.0) and new polars version.
 
 ## [0.22.1] - 2024-11-03
 ### Added:

pyproject.toml

@@ -58,6 +58,7 @@ path = "bigtree/__init__.py"
 dependencies = [
   "black",
   "coverage",
+  "isort",
   "matplotlib",
   "mypy",
   "pandas",

tests/tree/test_export.py

@@ -1173,11 +1173,20 @@ def test_tree_to_polars_name_col_missing(tree_node):
         tuple(map(int, pl.__version__.split(".")[:2])) > (1, 9),
         reason="Not compatible with polars>1.9.0",
     )
+    def test_tree_to_polars_name_path_col_missing_old_polars(tree_node):
+        actual = export.tree_to_polars(tree_node, name_col="", path_col="")
+        assert actual.is_empty()
+        assert actual.shape == (0, 0)
+
+    @staticmethod
+    @unittest.skipIf(
+        tuple(map(int, pl.__version__.split(".")[:2])) <= (1, 9),
+        reason="Not compatible with polars<=1.9.0",
+    )
     def test_tree_to_polars_name_path_col_missing(tree_node):
-        expected = pl.DataFrame()
-        expected.index = range(8)
         actual = export.tree_to_polars(tree_node, name_col="", path_col="")
-        assert expected.equals(actual)
+        assert actual.is_empty()
+        assert actual.shape == (8, 0)
 
     @staticmethod
     def test_tree_to_polars_parent_col(tree_node):