minor test update

Bears-R-Us · Nov 8, 2024 · 19fb1a2 · 19fb1a2
1 parent d97c5b6
commit 19fb1a2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/pdarrayclass_test.py b/tests/pdarrayclass_test.py
@@ -13,7 +13,7 @@
 import arkouda.pdarrayclass
 
 REDUCTION_OPS = list(set(ak.pdarrayclass.SUPPORTED_REDUCTION_OPS) - set(["isSorted", "isSortedLocally"]))
-
+INDEX_REDUCTION_OPS = ak.pdarrayclass.SUPPORTED_INDEX_REDUCTION_OPS
 
 DTYPES = ["int64", "float64", "bool", "uint64"]
 
@@ -151,7 +151,7 @@ def assert_reduction_ops_match(
 
         ak_assert_equivalent(ak_result, np_op(nda, axis=axis))
 
-    @pytest.mark.parametrize("op", ["argmin", "argmax"])
+    @pytest.mark.parametrize("op", INDEX_REDUCTION_OPS)
     @pytest.mark.parametrize("size", pytest.prob_size)
     @pytest.mark.parametrize("dtype", DTYPES)
     @pytest.mark.parametrize("arry_gen", [ak.zeros, ak.ones, ak.arange])
@@ -165,7 +165,7 @@ def test_index_reduction_1D(self, op, dtype, arry_gen, size, axis):
         ak_assert_equivalent(ak_result, np_op(nda, axis=axis))
 
     @pytest.mark.skip_if_max_rank_less_than(3)
-    @pytest.mark.parametrize("op", ["argmin", "argmax"])
+    @pytest.mark.parametrize("op", INDEX_REDUCTION_OPS)
     @pytest.mark.parametrize("size", pytest.prob_size)
     @pytest.mark.parametrize("dtype", DTYPES)
     @pytest.mark.parametrize("arry_gen", [ak.zeros, ak.ones, ak.arange])