Fix tests after prior file name is different

nuclear-multimessenger-astronomy · Aug 28, 2023 · 4b74bc6 · 4b74bc6
1 parent 19778ee
commit 4b74bc6
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/nmma/tests/injections.py b/nmma/tests/injections.py
@@ -50,7 +50,10 @@ def create_injection_from_command_line(model_name):
         path to the injection file created by nmma_create_injection
         """
 
-        prior_path = os.path.join("./priors/", model_name + ".prior")
+        if model_name == "nugent-hyper":
+            prior_path = os.path.join("./priors/", "sncosmo-generic" + ".prior")
+        else:
+            prior_path = os.path.join("./priors/", model_name + ".prior")
         assert os.path.exists(prior_path), "prior file does not exist"
         injection_name = os.path.join(test_directory, model_name + "_injection.json")