Fix broken doctest and tests on accessors (#46)

* Fix doctest on cupy_xarray.accessors.CupyDataArrayAccessor.as_cupy

Output type is now just `cupy.ndarray` instead of `cupy.core.core.ndarray`. Also updated pytest command in contributing.rst to include `--doctest-modules` flag.

* Fix ModuleNotFoundError: No module named 'xarray.core.pycompat'

Xref https://github.com/pydata/xarray/blob/v2024.06.0/xarray/namedarray/pycompat.py#L21-L43

* Use try-except instead of xr.namedarray.pycompat to get dask_array_type

---------

Co-authored-by: Deepak Cherian <[email protected]>

Author: weiji14

cupy_xarray/accessors.py

@@ -5,10 +5,13 @@
     register_dataarray_accessor,
     register_dataset_accessor,
 )
-from xarray.namedarray.pycompat import DuckArrayModule
 
-dsk = DuckArrayModule("dask")
-dask_array_type = dsk.type
+try:
+    import dask.array
+
+    dask_array_type = dask.array.Array
+except ImportError:
+    dask_array_type = None
 
 
 @register_dataarray_accessor("cupy")
@@ -55,7 +58,7 @@ def as_cupy(self):
         >>> da = xr.tutorial.load_dataset("air_temperature").air
         >>> gda = da.cupy.as_cupy()
         >>> type(gda.data)
-        <class 'cupy.core.core.ndarray'>
+        <class 'cupy.ndarray'>
 
         """
         if isinstance(self.da.data, dask_array_type):

cupy_xarray/tests/test_accessors.py

@@ -1,10 +1,16 @@
 import numpy as np
 import pytest
 import xarray as xr
-from xarray.core.pycompat import dask_array_type
 
 import cupy_xarray  # noqa: F401
 
+try:
+    import dask.array
+
+    dask_array_type = dask.array.Array
+except ImportError:
+    dask_array_type = None
+
 
 @pytest.fixture
 def tutorial_ds_air():

docs/source/contributing.rst

@@ -132,7 +132,7 @@ Running the test suite
 *cupy-xarray* uses the `pytest <https://docs.pytest.org/en/latest/contents.html>`_
 framework for testing. You can run the test suite using::
 
-    pytest cupy-xarray
+    pytest --doctest-modules cupy_xarray
 
 Contributing documentation
 ==========================