Merge branch 'main' into 5155_Bug_MultiIndex.lookup

Author: ajpotts

arkouda/pandas/extension/_arkouda_array.py

@@ -6,6 +6,7 @@
 import numpy as np
 
 from numpy import ndarray
+from numpy.typing import NDArray
 from pandas.api.extensions import ExtensionArray
 
 from arkouda.numpy.dtypes import dtype as ak_dtype
@@ -188,15 +189,17 @@ def astype(self, dtype, copy: bool = False):
         # Fallback: local cast
         return self.to_ndarray().astype(npdt, copy=copy)
 
-    def isna(self) -> ExtensionArray | ndarray[Any, Any]:
+    def isna(self) -> NDArray[np.bool_]:
         from arkouda.numpy import isnan
         from arkouda.numpy.pdarraycreation import full as ak_full
         from arkouda.numpy.util import is_float
 
         if not is_float(self._data):
-            return ak_full(self._data.size, False, dtype=bool)
+            return (
+                ak_full(self._data.size, False, dtype=bool).to_ndarray().astype(dtype=bool, copy=False)
+            )
 
-        return isnan(self._data)
+        return isnan(self._data).to_ndarray().astype(dtype=bool, copy=False)
 
     @property
     def dtype(self):

tests/pandas/extension/arkouda_array_extension.py

@@ -168,15 +168,15 @@ def test_isna(self):
         ak_data = ak.arange(10)
         arr = ArkoudaArray(ak_data)
         na = arr.isna()
-        assert ak.all(na == False)
+        assert np.all(na == False)
 
     def test_isna_with_nan(self):
         from arkouda.testing import assert_equal
 
         ak_data = ak.array([1, np.nan, 2])
         arr = ArkoudaArray(ak_data)
         na = arr.isna()
-        expected = ak.array([False, True, False])
+        expected = np.array([False, True, False])
         assert_equal(na, expected)
 
     def test_copy(self):