TST: update tests for reparam changes

mj-will · Sep 26, 2024 · cd94fb9 · cd94fb9
1 parent cedb7d4
commit cd94fb9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_reparameterisations/test_get_reparameterisation.py b/tests/test_reparameterisations/test_get_reparameterisation.py
@@ -127,7 +127,10 @@ def test_get_reparameterisation_invalid_input():
 
 def test_get_reparameterisation_defaults():
     """Test using an updated defaults dictionary."""
-    defaults = {"default": ("Class", {"x": 2})}
+    from nessai.reparameterisations import ReparameterisationDict
+
+    defaults = ReparameterisationDict()
+    defaults.add_reparameterisation("default", "Class", {"x": 2})
     cls, kwargs = get_reparameterisation("default", defaults=defaults)
     assert cls == "Class"
     assert kwargs == {"x": 2}