Add ExtensionArray to array_like (#229)

Author: kroenlein

gemd/__version__.py

@@ -1 +1 @@
-__version__ = "2.2.3"
+__version__ = "2.2.4"

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

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",

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

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