Skip to content

Commit 2fa267a

Browse files
committed
addition of test_exactextract
1 parent 8a23e04 commit 2fa267a

1 file changed

Lines changed: 31 additions & 0 deletions

File tree

xvec/tests/test_zonal_stats.py

Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -425,3 +425,34 @@ def test_variable_geometry_single(glaciers):
425425

426426
assert result.sizes == {"year": 3, "name": 5, "band": 11}
427427
assert result.statistics.mean() == 13168.585
428+
429+
430+
def test_exactextract_strategy():
431+
ds = xr.tutorial.open_dataset("eraint_uvz")
432+
world = gpd.read_file(geodatasets.get_path("naturalearth land"))
433+
434+
result_feature_sequential = ds.z.xvec.zonal_stats(
435+
world.geometry,
436+
"longitude",
437+
"latitude",
438+
method="exactextract",
439+
strategy="feature-sequential",
440+
)
441+
result_raster_sequential = ds.z.xvec.zonal_stats(
442+
world.geometry,
443+
"longitude",
444+
"latitude",
445+
method="exactextract",
446+
strategy="raster-sequential",
447+
)
448+
449+
xr.testing.assert_allclose(result_feature_sequential, result_raster_sequential)
450+
451+
with pytest.raises(KeyError):
452+
ds.z.xvec.zonal_stats(
453+
world.geometry,
454+
"longitude",
455+
"latitude",
456+
method="exactextract",
457+
strategy="invalid_strategy",
458+
)

0 commit comments

Comments
 (0)