Added combine_discovery_validation() tests (#1)

ocbe-uio · Jan 14, 2025 · 2ff0e2d · 2ff0e2d
1 parent 0061c99
commit 2ff0e2d
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/test_pipeline.py b/tests/test_pipeline.py
@@ -120,3 +120,13 @@ def test_fit_line():
         assert math.isclose(
             fit_disc.loc[line, 'CI'].values[0], 0.0000193, rel_tol=1e-2
         )
+
+all_pat = pipeline.combine_discovery_validation(
+    disc_data_shap, valid_data_shap, fit_disc, fit_valid
+)
+def test_combine_discovery_validation():
+    assert 'Feature' in all_pat.columns
+    assert 'Fraction' in all_pat.columns
+    assert 'SHAP value' in all_pat.columns
+    if local:
+        all_pat.shape == (15888, 3)