TST: add cumulative_prod to test_flags

data-apis · Dec 16, 2024 · 3969d2f · 3969d2f
1 parent 0bff2f1
commit 3969d2f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/array_api_strict/tests/test_flags.py b/array_api_strict/tests/test_flags.py
@@ -308,6 +308,7 @@ def test_api_version_2023_12(func_name):
     'take_along_axis': lambda: xp.take_along_axis(xp.zeros((2, 3)),
                                                   xp.zeros((1, 4), dtype=xp.int64)),
     'count_nonzero': lambda: xp.count_nonzero(xp.arange(3)),
+    'cumulative_prod': lambda: xp.cumulative_prod(xp.arange(1, 5)),
 }
 
 @pytest.mark.parametrize('func_name', api_version_2024_12_examples.keys())