Add product to reduction statistics (for raster.Place) (#62)

nens · Aug 25, 2020 · 0f5a136 · 0f5a136
1 parent 2435d1a
commit 0f5a136
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/CHANGES.rst b/CHANGES.rst
@@ -6,6 +6,7 @@ Changelog of dask-geomodeling
 
 - Check for matching time resolutions in raster.Clip.
 
+- Added 'product' to raster.reduction STATISTICS.
 
 2.2.10 (2020-07-29)
 -------------------

diff --git a/dask_geomodeling/raster/reduction.py b/dask_geomodeling/raster/reduction.py
@@ -19,6 +19,7 @@
     "std": np.nanstd,
     "var": np.nanvar,
     "median": np.nanmedian,
+    "product": np.nanprod,
     # "p<number>" uses np.nanpercentile
 }
 

diff --git a/dask_geomodeling/tests/test_raster_reduction.py b/dask_geomodeling/tests/test_raster_reduction.py
@@ -51,6 +51,7 @@ def stack_nodata_only():
         ("std", [[[np.sqrt(2 / 3), np.sqrt(2 / 3), np.sqrt(2 / 3)], [0, 0, 255]]]),
         ("var", [[[2 / 3, 2 / 3, 2 / 3], [0, 0, 255]]]),
         ("median", [[[2, 2, 2], [4, 5, 255]]]),
+        ("product", [[[6, 6, 6], [16, 25, 255]]]),
         ("p99", [[[2.98, 2.98, 2.98], [4, 5, 255]]]),
     ],
 )
@@ -61,7 +62,7 @@ def test_reduce(statistic, expected, dtype, stack):
 
 
 @pytest.mark.parametrize(
-    "statistic,expected_value",
+    "statistic, expected_value",
     [
         ("first", 255),
         ("last", 255),
@@ -75,6 +76,7 @@ def test_reduce(statistic, expected, dtype, stack):
         ("std", 255),
         ("var", 255),
         ("median", 255),
+        ("product", 255),
         ("p99", 255),
     ],
 )