Skip to content

Commit 20172a7

Browse files
committed
final fix
1 parent 8ab4a61 commit 20172a7

File tree

2 files changed

+6
-7
lines changed

2 files changed

+6
-7
lines changed

intake_xarray/tests/data/catalog.yaml

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -60,8 +60,7 @@ sources:
6060
description: "https://github.com/mapbox/rasterio/blob/master/tests/data/<red|green>.tif"
6161
driver: rasterio
6262
args:
63-
urlpath: '{{ CATALOG_DIR }}little_{band}.tif'
64-
concat_dim: band
63+
urlpath: '{{ CATALOG_DIR }}little_{color}.tif'
6564
pattern_tiff_source_path_pattern_as_str:
6665
description: "https://github.com/mapbox/rasterio/blob/master/tests/data/<red|green>.tif"
6766
driver: rasterio

intake_xarray/tests/test_intake_xarray.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -125,15 +125,15 @@ def test_read_pattern_field_as_band():
125125
cat = intake.open_catalog(os.path.join(here, 'data', 'catalog.yaml'))
126126
colors = cat.pattern_tiff_source_path_pattern_field_as_band
127127

128-
da = colors.read()
129-
assert len(da.band) == 6
130-
assert set(da.band.data) == set(['red', 'green'])
131-
assert da[da.band == 'red'].shape == (3, 64, 64)
128+
da = colors.read().band_data
129+
assert da.shape == (2, 3, 64, 64)
130+
assert set(da.color.data) == set(['red', 'green'])
131+
assert da[da.color == 'red'].squeeze().shape == (3, 64, 64)
132132

133133
rgb = {'red': [204, 17, 17], 'green': [17, 204, 17]}
134134
for color, values in rgb.items():
135135
for i, v in enumerate(values):
136-
assert (da[da.band == color][i].values == v).all()
136+
assert (da[da.color == color][0][i].values == v).all()
137137

138138

139139
def test_read_pattern_path_as_pattern_as_str_with_list_of_urlpaths():

0 commit comments

Comments
 (0)