tests

pola-rs · Oct 1, 2024 · f3f727d · f3f727d
1 parent 018ec08
commit f3f727d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/py-polars/polars/testing/parametric/strategies/dtype.py b/py-polars/polars/testing/parametric/strategies/dtype.py
@@ -334,9 +334,9 @@ def _time_zones() -> SearchStrategy[str]:
     )
 
 
-def _categorical_orderings() -> SearchStrategy[CategoricalOrdering]:
+def _categorical_orderings() -> SearchStrategy[CategoricalOrdering | None]:
     """Create a strategy for generating valid ordering types for categorical data."""
-    return st.sampled_from(["physical", "lexical"])
+    return st.sampled_from([None, "lexical"])
 
 
 @st.composite

diff --git a/py-polars/tests/unit/operations/namespaces/test_categorical.py b/py-polars/tests/unit/operations/namespaces/test_categorical.py
@@ -1,3 +1,5 @@
+import warnings
+
 import pytest
 
 import polars as pl
@@ -146,5 +148,6 @@ def test_cat_uses_lexical_ordering() -> None:
     s = s.cast(pl.Categorical("lexical"))
     assert s.cat.uses_lexical_ordering() is True
 
-    s = s.cast(pl.Categorical("physical"))
-    assert s.cat.uses_lexical_ordering() is False
+    with warnings.catch_warnings():
+        s = s.cast(pl.Categorical("physical"))
+        assert s.cat.uses_lexical_ordering() is False