TEST simplify qoi tests

DataResponsibly · Nov 13, 2024 · fc80181 · fc80181
1 parent 2903153
commit fc80181
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/sharp/qoi/tests/test_basic_usage.py b/sharp/qoi/tests/test_basic_usage.py
@@ -24,6 +24,8 @@ def score_function(X):
 qois_str = ["rank", "rank_score", "top_k", "diff", "flip", "likelihood"]
 qois_obj = [get_qoi(qoi) for qoi in qois_str]
 
+pairs1 = rng.randint(0, X.shape[0], size=3)
+pairs2 = rng.randint(0, X.shape[0], size=3)
 
 # @pytest.mark.parametrize("qoi_str, item_idx", product(qois_str, range(X.shape[0])))
 # def test_qoi_estimation(qoi_str, item_idx):
@@ -52,7 +54,7 @@ def test_qoi_estimation(qoi_str):
 
 @pytest.mark.parametrize(
     "qoi_str, item_idx1, item_idx2",
-    product(qois_str, range(X.shape[0]), range(X.shape[0])),
+    product(qois_str, pairs1, pairs2),
 )
 def test_qoi_calculation(qoi_str, item_idx1, item_idx2):
     item1 = X[item_idx1 : item_idx1 + 1]