use parametrize

UXARRAY · Nov 19, 2024 · a3c87b8 · a3c87b8
1 parent 8c36fa0
commit a3c87b8
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/test/test_weighted_mean.py b/test/test_weighted_mean.py
@@ -1,6 +1,8 @@
 import os
 from pathlib import Path
 
+import pytest
+
 import dask.array as da
 import numpy as np
 import numpy.testing as nt
@@ -122,5 +124,9 @@ def test_csne30_equal_area():
 
 
 # TODO for Rachel
-def test_csne30_equal_area_dask():
+@pytest.mark.parametrize("chunk_size", [1, 2, 4])
+def test_csne30_equal_area_dask(chunk_size):
+
+    # ... .chunk(n_face=chunk_size)
+
     pass