Merge pull request #11 from jhnnsrs/main

xarray test and compatibility with latest xarray

Author: d-v-b

.gitignore

@@ -38,4 +38,5 @@ MANIFEST.in
 .venv
 /releases/*
 pip-wheel-metadata
-/poetry.toml
+/poetry.toml
+/venv

src/xarray_multiscale/multiscale.py

@@ -330,7 +330,9 @@ def downscale(
         to_downscale.data, reduction, scale_factors, **kwargs
     )
     downscaled_coords = downscale_coords(to_downscale, scale_factors)
-    return DataArray(downscaled_data, downscaled_coords, attrs=array.attrs)
+    return DataArray(
+        downscaled_data, downscaled_coords, attrs=array.attrs, dims=array.dims
+    )
 
 
 def downscale_coords(

tests/test_doc.py

@@ -0,0 +1,10 @@
+from xarray_multiscale import multiscale
+from xarray_multiscale.reducers import windowed_mean
+import numpy as np
+import xarray as xr
+
+
+def test_xarray_example():
+    data = xr.DataArray(np.zeros((1024, 1024)), dims=("x", "y"))
+    scaled_data = multiscale(data, windowed_mean, (2, 2))
+    assert len(scaled_data) == 11, "Incorrect number of arrays returned"