diff --git a/tests/test_cmaes.py b/tests/test_cmaes.py index b227add..8eef4e7 100644 --- 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 index 720c711..5639a9a 100644 --- 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 index 7bb99aa..ca7b524 100644 --- 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 index e833691..7a3def0 100644 --- 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 index fc9f6ec..16c4241 100644 --- 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 index 6a10ade..f6fb5e3 100644 --- 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, )