removed cheating, normal log levels

Helmholtz-AI-Energy · Oct 23, 2023 · e42db3d · e42db3d
1 parent bbda900
commit e42db3d
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 4 deletions.
diff --git a/tests/test_cmaes.py b/tests/test_cmaes.py
@@ -32,7 +32,6 @@ def test_PSO():
     """
     Test single worker using Propulator to optimize sphere using a PSO propagator.
     """
-    return
     rng = random.Random(42)  # Separate random number generator for optimization.
     limits = {
         "a": (-5.12, 5.12),

diff --git a/tests/test_propulator_sphere.py b/tests/test_propulator_sphere.py
@@ -40,7 +40,7 @@ def test_Propulator():
     }
     with tempfile.TemporaryDirectory() as checkpoint_path:
         set_logger_config(
-            level=logging.DEBUG,
+            level=logging.INFO,
             log_file=checkpoint_path + "/propulate.log",
             log_to_stdout=True,
             log_rank=False,
@@ -66,7 +66,7 @@ def test_Propulator():
         )
 
         # Run optimization and print summary of results.
-        propulator.propulate(debug=2)
+        propulator.propulate()
         propulator.summarize()
         best = min(propulator.population, key=attrgetter("loss"))
 

diff --git a/tests/test_pso.py b/tests/test_pso.py
@@ -33,7 +33,6 @@ def test_PSO():
     """
     Test single worker using Propulator to optimize sphere using a PSO propagator.
     """
-    return
     rng = random.Random(42)  # Separate random number generator for optimization.
     limits = {
         "a": (-5.12, 5.12),