diff --git a/test/gw/result_test.py b/test/gw/result_test.py index a92cb104..d0f8ca56 100644 --- 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__":