add test for refit_models

alan-turing-institute · Feb 23, 2024 · b06e2c1 · b06e2c1
1 parent 0994de0
commit b06e2c1
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/test_compare.py b/tests/test_compare.py
@@ -157,3 +157,9 @@ def test_refit_model(ae_run):
     model = ae_run.get_model(rank=1)
     refitted_model = ae_run.refit_model(model)
     assert refitted_model is not None
+
+
+def test_refit_models(ae_run):
+    models = ae_run.refit_models()
+    assert models is not None
+    assert len(models) == len(ae_run.models)