reduce number of generations in tests

Helmholtz-AI-Energy · Aug 15, 2024 · a0681a8 · a0681a8
1 parent 66bc773
commit a0681a8
Show file tree

Hide file tree

Showing 6 changed files with 14 additions and 14 deletions.
diff --git a/tests/test_cmaes.py b/tests/test_cmaes.py
@@ -38,7 +38,7 @@ def test_cmaes_basic(cma_adapter: CMAAdapter, mpi_tmp_path: pathlib.Path) -> Non
         loss_fn=benchmark_function,
         propagator=propagator,
         rng=rng,
-        generations=100,
+        generations=10,
         checkpoint_path=mpi_tmp_path,
     )
     # Run optimization and print summary of results.

diff --git a/tests/test_island.py b/tests/test_island.py
@@ -70,7 +70,7 @@ def test_islands(
         loss_fn=benchmark_function,
         propagator=propagator,
         rng=rng,
-        generations=100,
+        generations=10,
         num_islands=2,
         migration_probability=0.9,
         pollination=pollination,
@@ -109,7 +109,7 @@ def test_checkpointing_isolated(
         loss_fn=benchmark_function,
         propagator=propagator,
         rng=rng,
-        generations=100,
+        generations=10,
         num_islands=2,
         migration_probability=0.0,
         checkpoint_path=mpi_tmp_path,
@@ -126,7 +126,7 @@ def test_checkpointing_isolated(
         loss_fn=benchmark_function,
         propagator=propagator,
         rng=rng,
-        generations=100,
+        generations=10,
         num_islands=2,
         migration_probability=0.0,
         checkpoint_path=mpi_tmp_path,
@@ -170,7 +170,7 @@ def test_checkpointing(
         loss_fn=benchmark_function,
         propagator=propagator,
         rng=rng,
-        generations=100,
+        generations=10,
         num_islands=2,
         migration_probability=0.9,
         pollination=pollination,
@@ -188,7 +188,7 @@ def test_checkpointing(
         loss_fn=benchmark_function,
         propagator=propagator,
         rng=rng,
-        generations=100,
+        generations=10,
         num_islands=2,
         migration_probability=0.9,
         pollination=pollination,
@@ -233,7 +233,7 @@ def test_checkpointing_unequal_populations(
         loss_fn=benchmark_function,
         propagator=propagator,
         rng=rng,
-        generations=100,
+        generations=10,
         num_islands=2,
         island_sizes=np.array([3, 5]),
         migration_probability=0.9,
@@ -252,7 +252,7 @@ def test_checkpointing_unequal_populations(
         loss_fn=benchmark_function,
         propagator=propagator,
         rng=rng,
-        generations=100,
+        generations=10,
         num_islands=2,
         island_sizes=np.array([3, 5]),
         migration_probability=0.9,

diff --git a/tests/test_multi_rank_workers.py b/tests/test_multi_rank_workers.py
@@ -69,7 +69,7 @@ def test_multi_rank_workers(mpi_tmp_path: pathlib.Path) -> None:
         loss_fn=parallel_sphere,  # Loss function to be minimized
         propagator=propagator,  # Propagator, i.e., evolutionary operator to be used
         rng=rng,  # Separate random number generator for Propulate optimization
-        generations=100,  # Overall number of generations
+        generations=10,  # Overall number of generations
         num_islands=2,  # Number of islands
         migration_probability=0.9,  # Migration probability
         pollination=False,  # Whether to use pollination or migration

diff --git a/tests/test_nm.py b/tests/test_nm.py
@@ -61,7 +61,7 @@ def test_cmaes(
         loss_fn=function,
         propagator=propagator,
         rng=rng,
-        generations=100,
+        generations=10,
         checkpoint_path=mpi_tmp_path,
     )
     # Run optimization and print summary of results.

diff --git a/tests/test_propulator.py b/tests/test_propulator.py
@@ -61,7 +61,7 @@ def test_propulator(function_name: str, mpi_tmp_path: pathlib.Path) -> None:
         loss_fn=benchmark_function,
         propagator=propagator,
         rng=rng,
-        generations=100,
+        generations=10,
         checkpoint_path=mpi_tmp_path,
     )  # Set up propulator performing actual optimization.
     propulator.propulate()  # Run optimization and print summary of results.
@@ -91,7 +91,7 @@ def test_propulator_checkpointing(mpi_tmp_path: pathlib.Path) -> None:
     propulator = Propulator(
         loss_fn=benchmark_function,
         propagator=propagator,
-        generations=100,
+        generations=10,
         checkpoint_path=mpi_tmp_path,
         rng=rng,
     )  # Set up propulator performing actual optimization.
@@ -107,7 +107,7 @@ def test_propulator_checkpointing(mpi_tmp_path: pathlib.Path) -> None:
     propulator = Propulator(
         loss_fn=benchmark_function,
         propagator=propagator,
-        generations=20,
+        generations=5,
         checkpoint_path=mpi_tmp_path,
         rng=rng,
     )  # Set up new propulator starting from checkpoint.

diff --git a/tests/test_pso.py b/tests/test_pso.py
@@ -77,7 +77,7 @@ def test_pso(pso_propagator: Propagator, mpi_tmp_path: pathlib.Path) -> None:
         loss_fn=sphere,
         propagator=propagator,
         rng=rng,
-        generations=100,
+        generations=10,
         checkpoint_path=mpi_tmp_path,
     )