From cd94fb9f6bb35fc35a808160483c858bd6dd80b1 Mon Sep 17 00:00:00 2001 From: mj-will Date: Thu, 26 Sep 2024 12:47:24 +0200 Subject: [PATCH] TST: update tests for reparam changes --- .../test_reparameterisations/test_get_reparameterisation.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/test_reparameterisations/test_get_reparameterisation.py b/tests/test_reparameterisations/test_get_reparameterisation.py index 2ddbd0dc..1f11c98b 100644 --- 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}