unit test for setup_pipeline complete

LSSTDESC · Dec 5, 2023 · 80c9a0b · 80c9a0b
1 parent 1cd8e29
commit 80c9a0b
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/tests/test_run_cosmosis_2pt.py b/tests/test_run_cosmosis_2pt.py
@@ -61,9 +61,21 @@ def __init__(self):
 def test_setup_pipeline():
     # Setup
     inifile = './src/blind_2pt_cosmosis/cosmosis_files/default_blinding_template.ini'
-    angles_file = 'angles.fits'
-    nz_file = 'nz.fits'
+    angles_file = './src/blind_2pt_cosmosis/cosmosis_files/sim_fiducial.fits'
+    nz_file = './src/blind_2pt_cosmosis/cosmosis_files/sim_fiducial.fits'
 
     # Exercise
     pipeline = setup_pipeline(inifile)
+    assert isinstance(pipeline, LikelihoodPipeline)
+
+    # Exercise
+    pipeline = setup_pipeline(inifile, angles_file=angles_file)
+    assert isinstance(pipeline, LikelihoodPipeline)
+
+    # Exercise
+    pipeline = setup_pipeline(inifile, nz_file=angles_file)
+    assert isinstance(pipeline, LikelihoodPipeline)
+
+    # Exercise
+    pipeline = setup_pipeline(inifile, nz_file=angles_file, angles_file=angles_file)
     assert isinstance(pipeline, LikelihoodPipeline)