Passing inspection_functions tests

Author: lshaw8317

src/blosc2/__init__.py

@@ -25,9 +25,10 @@
 if not IS_WASM:
     import numexpr
 
-from .version import __version__
+from .version import __array_api_version__, __version__
 
 __version__ = __version__
+__array_api_version__ = __array_api_version__
 """
 Python-Blosc2 version.
 """
@@ -226,34 +227,28 @@ def __array_namespace_info__() -> Info:
     """
     Return information about the array namespace following the Array API specification.
     """
+
+    def _raise(exc):
+        raise exc
+
     return Info(
-        capabilities={
+        capabilities=lambda: {
             "boolean indexing": True,
             "data-dependent shapes": False,
-            "max dimensions": 8,
+            "max dimensions": MAX_DIM,
         },
-        default_device=None,
-        default_dtypes={
+        default_device=lambda: "cpu",
+        default_dtypes=lambda device=None: {
             "real floating": DEFAULT_FLOAT,
             "complex floating": DEFAULT_COMPLEX,
             "integral": DEFAULT_INT,
             "indexing": DEFAULT_INDEX,
-        },
-        dtypes={
-            "bool": bool_,
-            "int8": int8,
-            "int16": int16,
-            "int32": int32,
-            "int64": int64,
-            "uint8": uint8,
-            "uint16": uint16,
-            "uint32": uint32,
-            "uint64": uint64,
-            "float32": float32,
-            "float64": float64,
-            "complex64": complex64,
-            "complex128": complex128,
-        },
+        }
+        if (device == "cpu" or device is None)
+        else _raise(ValueError("Only cpu devices allowed")),
+        dtypes=lambda device=None, kind=None: np.__array_namespace_info__().dtypes(kind=kind, device=device)
+        if (device == "cpu" or device is None)
+        else _raise(ValueError("Only cpu devices allowed")),
         devices=lambda: ["cpu"],
         name="blosc2",
         version=__version__,

src/blosc2/ndarray.py

@@ -991,6 +991,12 @@ def __complex__(self) -> complex:
             raise ValueError(f"Cannot convert array of shape {self.shape} to complex float.")
         return complex(self[()])
 
+    def item(self) -> float | bool | complex | int:
+        """
+        Copy an element of an array to a standard Python scalar and return it.
+        """
+        return self[()].item()
+
     @is_documented_by(sum)
     def sum(self, axis=None, dtype=None, keepdims=False, **kwargs):
         expr = blosc2.LazyExpr(new_op=(self, None, None))
@@ -1831,7 +1837,7 @@ def __setitem__(  # noqa : C901
 
         key_, mask = process_key(key, self.shape)  # internally handles key an integer
         if hasattr(value, "shape") and value.shape == ():
-            value = value[()]
+            value = value.item()
 
         def updater(sel_idx):
             return value[sel_idx]
@@ -4379,6 +4385,7 @@ def _check_ndarray_kwargs(**kwargs):
         "storage",
         "out",
     ]
+    _ = kwargs.pop("device", None)  # pop device (not used, but needs to be discarded)
     for key in kwargs:
         if key not in supported_keys:
             raise KeyError(

src/blosc2/version.py

@@ -1 +1,2 @@
 __version__ = "3.7.3.dev0"
+__array_api_version__ = "2024.12"