Add ExtensionArray to array_like (#229)

kroenlein · web-flow · commit 54463ee4147f · 2026-02-27T13:44:50.000-07:00
diff --git a/gemd/__version__.py b/gemd/__version__.py
@@ -1 +1 @@
-__version__ = "2.2.3"
+__version__ = "2.2.4"
diff --git a/gemd/entity/util.py b/gemd/entity/util.py
@@ -88,7 +88,9 @@ def array_like():
         import numpy as np
         try:
             import pandas as pd
-            _array_like = (list, tuple, np.ndarray, pd.core.base.PandasObject)
+            _array_like = (list, tuple, np.ndarray,
+                           pd.core.base.PandasObject,
+                           pd.api.extensions.ExtensionArray)
         except ImportError:  # pragma: no cover
             _array_like = (list, tuple, np.ndarray)  # pragma: no cover
     except ImportError:  # pragma: no cover
diff --git a/pyproject.toml b/pyproject.toml
@@ -47,10 +47,10 @@ exclude = ["docs", "tests"]
 dev = [
     "flake8==7.0.0",
     "flake8-docstrings==1.7.0",
-    "numpy==1.24.4; python_version<'3.10'",
-    "pandas==2.0.3; python_version<'3.10'",
-    "numpy>=2.0.2,<=2.1.0; python_version>='3.10'",
-    "pandas==2.3.0; python_version>='3.10'",
+    "numpy>=1.24.4; python_version<'3.10'",
+    "pandas>=2.0.3; python_version<'3.10'",
+    "numpy>=2.0.2,<3; python_version>='3.10'",
+    "pandas>=2.3.0,<4; python_version>='3.10'",
     "pytest==8.4.2",
     "pytest-cov==7.0.0",
     "derp==0.1.1",
diff --git a/tests/entity/bounds/test_categorical_bounds.py b/tests/entity/bounds/test_categorical_bounds.py
@@ -62,7 +62,7 @@ def test_json():
 
 def test_numpy():
     """Test that ndarrays, Series work as well."""
-    assert len(array_like()) < 5  # In case we extend at some point
+    assert len(array_like()) < 6  # In case we extend at some point
 
     if len(array_like()) > 2:  # Test numpy
         import numpy as np
@@ -75,3 +75,6 @@ def test_numpy():
         pd_bounds = CategoricalBounds(pd.Series(["spam", "eggs"]))
         pd_copy = loads(dumps(pd_bounds))
         assert pd_copy == pd_bounds
+
+        pd_unique = CategoricalBounds(pd.Series(["spam", "eggs"]).unique())
+        assert pd_copy == pd_unique
diff --git a/tests/entity/bounds/test_composition_bounds.py b/tests/entity/bounds/test_composition_bounds.py
@@ -63,7 +63,7 @@ def test_json():
 
 def test_numpy():
     """Test that ndarrays, Series work as well."""
-    assert len(array_like()) < 5  # In case we extend at some point
+    assert len(array_like()) < 6  # In case we extend at some point
 
     if len(array_like()) > 2:  # Test numpy
         import numpy as np

Original file line number	Diff line number	Diff line change
`@@ -1 +1 @@`
`1`		`-__version__ = "2.2.3"`
	`1`	`+__version__ = "2.2.4"`