feat: enhance DataWrapper to support dims and coords properties for custom data types (#228)

Author: tlambert03

src/ndv/models/_data_wrapper.py

@@ -129,12 +129,17 @@ def supports(cls, obj: Any) -> TypeGuard[Any]:
     @property
     def dims(self) -> tuple[Hashable, ...]:
         """Return the dimension labels for the data."""
+        if hasattr(self._data, "dims"):
+            return tuple(self._data.dims)
         # type ignore is asserted in the __init__ method
         return tuple(range(len(self._data.shape)))  # type: ignore [attr-defined]
 
     @property
     def coords(self) -> Mapping[Hashable, Sequence]:
         """Return the coordinates for the data."""
+        if hasattr(self._data, "coords") and isinstance(self._data.coords, Mapping):
+            return self._data.coords
+
         dims = self.dims
         # type ignore is asserted in the __init__ method
         return {i: range(s) for i, s in zip(dims, self._data.shape)}  # type: ignore [attr-defined]
@@ -399,16 +404,6 @@ def _asarray(self, data: cl_array.Array) -> np.ndarray:
 class XarrayWrapper(DataWrapper["xr.DataArray"]):
     """Wrapper for xarray DataArray objects."""
 
-    @property
-    def dims(self) -> tuple[Hashable, ...]:
-        """Return the dimension labels for the data."""
-        return tuple(self._data.dims)
-
-    @property
-    def coords(self) -> Mapping[Hashable, Sequence]:
-        """Return the coordinates for the data."""
-        return self.data.coords  # type: ignore [no-any-return]
-
     @classmethod
     def supports(cls, obj: Any) -> TypeGuard[xr.DataArray]:
         if (xr := sys.modules.get("xarray")) and isinstance(obj, xr.DataArray):