Expand tests

emdgroup · Jan 20, 2025 · 0baf9b7 · 0baf9b7
1 parent 8a27738
commit 0baf9b7
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/tests/insights/test_shap.py b/tests/insights/test_shap.py
@@ -47,9 +47,18 @@ def batch_size():
 @pytest.fixture(
     params=[
         ["Conti_finite1", "Conti_finite2"],
-        ["Categorical_1", "SomeSetting", "Num_disc_1", "Conti_finite1"],
+        ["Num_disc_1", "Fraction_1"],
+        ["Num_disc_1", "Conti_finite1"],
+        ["Num_disc_1", "Categorical_1"],
+        ["Conti_finite1", "Categorical_1"],
+    ],
+    ids=[
+        "params_conti",
+        "params_disc_num",
+        "params_hybrid_num",
+        "params_disc_cat",
+        "params_hybrid_cat",
     ],
-    ids=["conti_params", "hybrid_params"],
 )
 def parameter_names(request):
     return request.param