first checkpointing test to only compare active population

Helmholtz-AI-Energy · Sep 18, 2024 · 91738de · 91738de
1 parent 40c688d
commit 91738de
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_island.py b/tests/test_island.py
@@ -176,7 +176,7 @@ def test_islands_checkpointing(
     # Run actual optimization.
     islands.propulate(debug=2)
 
-    old_population = copy.deepcopy(islands.propulator.population)
+    old_population = copy.deepcopy(islands.propulator._get_active_individuals()[0])
     del islands
 
     islands = Islands(
@@ -190,7 +190,7 @@ def test_islands_checkpointing(
         checkpoint_path=mpi_tmp_path,
     )
 
-    assert len(deepdiff.DeepDiff(old_population, islands.propulator.population, ignore_order=True)) == 0
+    assert len(deepdiff.DeepDiff(old_population, list(islands.propulator.population.values()), ignore_order=True)) == 0
     islands.propulate()
     log.handlers.clear()