Fix tests

juaml · Apr 24, 2024 · c187b80 · c187b80
1 parent 7719120
commit c187b80
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 0 deletions.
diff --git a/julearn/pipeline/test/test_merger.py → julearn/pipeline/tests/test_merger.py b/julearn/pipeline/test/test_merger.py → julearn/pipeline/tests/test_merger.py
@@ -104,3 +104,12 @@ def test_merger_errors() -> None:
         match="must have the same named steps.",
     ):
         merge_pipelines(pipe1, pipe4, search_params=None)
+
+    search_params = {"kind": "grid"}
+    pipe5 = creator2.to_pipeline(search_params={"kind": "bayes"})
+
+    with pytest.raises(
+        ValueError,
+        match="one of the pipelines to merge is a BayesSearchCV",
+    ):
+        merge_pipelines(pipe1, pipe5, search_params=search_params)
diff --git a/...rn/pipeline/test/test_pipeline_creator.py → ...n/pipeline/tests/test_pipeline_creator.py b/...rn/pipeline/test/test_pipeline_creator.py → ...n/pipeline/tests/test_pipeline_creator.py
diff --git a/...arn/pipeline/test/test_target_pipeline.py → ...rn/pipeline/tests/test_target_pipeline.py b/...arn/pipeline/test/test_target_pipeline.py → ...rn/pipeline/tests/test_target_pipeline.py
diff --git a/...line/test/test_target_pipeline_creator.py → ...ine/tests/test_target_pipeline_creator.py b/...line/test/test_target_pipeline_creator.py → ...ine/tests/test_target_pipeline_creator.py