Merge pull request #123 from intake/fix_122

fix `xarray_image` `to_zarr` if `xr.DataArray`

Author: martindurant

intake_xarray/tests/test_image.py

@@ -184,3 +184,14 @@ def test_read_images_and_exif_as_glob_with_coerce():
     ds = source.read()
     assert ds['raster'].shape == (3, 256, 256, 3)
     assert ds['EXIF Image ImageWidth'].shape == (3,)
+
+
+def test_read_images_and_persist():
+    pytest.importorskip('skimage')
+    urlpath = os.path.join(here, 'data', 'images', '*')
+    source = ImageSource(urlpath=urlpath, coerce_shape=(256, 256))
+    import tempfile
+    exported = tempfile.mkdtemp()
+    source.export(exported)
+    import xarray as xr
+    assert xr.open_dataset(exported, engine="zarr")

intake_xarray/xarray_container.py

@@ -1,5 +1,6 @@
 import itertools
 import os
+import xarray as xr
 from dask.delayed import Delayed
 from intake.container.base import RemoteSource, get_partition
 from intake.source.base import Schema
@@ -155,6 +156,9 @@ def _persist(source, path, **kwargs):
         http://xarray.pydata.org/en/stable/generated/xarray.Dataset.to_zarr.html
         """
         from intake_xarray import ZarrSource
-        source.to_dask().to_zarr(path, **kwargs)
+        ds = source.to_dask()
+        if isinstance(ds, xr.DataArray):
+            ds = ds.to_dataset(name=ds.name if ds.name else "variable")
+        ds.to_zarr(path, **kwargs)
         return ZarrSource(path)