diff --git a/array_api_strict/tests/test_flags.py b/array_api_strict/tests/test_flags.py index ebee415..dcfc20d 100644 --- 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())