BUG: Add band_as_variable to xarray plugin (#603)

corteva · Nov 7, 2022 · 5952536 · 5952536
1 parent 59e97af
commit 5952536
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/rioxarray/xarray_plugin.py b/rioxarray/xarray_plugin.py
@@ -48,6 +48,7 @@ def open_dataset(
         default_name="band_data",
         decode_times=True,
         decode_timedelta=None,
+        band_as_variable=False,
         open_kwargs=None,
     ):
         if open_kwargs is None:
@@ -65,6 +66,7 @@ def open_dataset(
             default_name=default_name,
             decode_times=decode_times,
             decode_timedelta=decode_timedelta,
+            band_as_variable=band_as_variable,
             **open_kwargs,
         )
         if isinstance(rds, xr.DataArray):

diff --git a/test/integration/test_integration__io.py b/test/integration/test_integration__io.py
@@ -482,9 +482,11 @@ def test_utm():
             assert "y" not in rioda.coords
 
 
-def test_band_as_variable():
+def test_band_as_variable(open_rasterio):
     with create_tmp_geotiff() as (tmp_file, expected):
-        with rioxarray.open_rasterio(tmp_file, band_as_variable=True) as riods:
+        with open_rasterio(
+            tmp_file, band_as_variable=True, mask_and_scale=False
+        ) as riods:
             for band in (1, 2, 3):
                 band_name = f"band_{band}"
                 assert_allclose(riods[band_name], expected.sel(band=band).drop("band"))