TST: update cosmology test in result tests

bilby-dev · Nov 28, 2024 · 235992c · 235992c
1 parent c49c5d3
commit 235992c
Showing 1 changed file with 5 additions and 9 deletions.
diff --git a/test/gw/result_test.py b/test/gw/result_test.py
@@ -3,7 +3,6 @@
 import unittest
 
 import pandas as pd
-from parameterized import parameterized
 
 import bilby
 
@@ -201,14 +200,11 @@ def test_detector_injection_properties_no_injection(self):
             self.result.detector_injection_properties("not_a_detector"), None
         )
 
-    @parameterized.expand(["Planck18", "Planck15", "Planck15_LAL"])
-    def test_cosmology(self, cosmology):
-        from bilby.gw.cosmology import get_cosmology, set_cosmology
-        set_cosmology(cosmology)
-        cosmo = get_cosmology()
-        self.assertEqual(self.result.cosmology, cosmo)
-        # Reset the cosmology to the default
-        set_cosmology("Planck15")
+    def test_cosmology(self):
+        self.assertEqual(
+            self.result.cosmology,
+            self.meta_data["cosmology"],
+        )
 
 
 if __name__ == "__main__":