typing

Author: keewis

xarray_array_testing/base.py

@@ -3,6 +3,7 @@
 from types import ModuleType
 
 import numpy.testing as npt
+from xarray.namedarray._typing import duckarray
 
 
 class DuckArrayTestMixin(ABC):
@@ -13,7 +14,7 @@ def xp() -> ModuleType:
 
     @property
     @abc.abstractmethod
-    def array_type(self) -> type:
+    def array_type(self) -> type[duckarray]:
         pass
 
     @staticmethod

xarray_array_testing/tests/test_numpy.py

@@ -1,3 +1,5 @@
+from types import ModuleType
+
 import hypothesis.strategies as st
 import numpy as np
 
@@ -12,11 +14,11 @@ def create_numpy_array(*, shape, dtype):
 
 class NumpyTestMixin(DuckArrayTestMixin):
     @property
-    def xp(self):
+    def xp(self) -> ModuleType:
         return np
 
     @property
-    def array_type(self):
+    def array_type(self) -> type[np.ndarray]:
         return np.ndarray
 
     @staticmethod