added mpi test markers for skipping the tests without mpi

Helmholtz-AI-Energy · Nov 6, 2023 · b14a316 · b14a316
1 parent 4bc3e38
commit b14a316
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 0 deletions.
diff --git a/tests/test_cmaes.py b/tests/test_cmaes.py
@@ -2,11 +2,14 @@
 import tempfile
 from operator import attrgetter
 
+import pytest
+
 from propulate import Propulator
 from propulate.propagators import CMAPropagator, BasicCMA
 from propulate.utils import sphere
 
 
+@pytest.mark.mpi_skip
 def test_PSO():
     """
     Test single worker using Propulator to optimize sphere using a PSO propagator.

diff --git a/tests/test_population.py b/tests/test_population.py
@@ -1,6 +1,9 @@
 from propulate.population import Individual
 
+import pytest
 
+
+@pytest.mark.mpi_skip
 def test_individual():
     ind_map = {
         "float1": 0.1,
@@ -15,5 +18,6 @@ def test_individual():
     print(ind)
 
 
+@pytest.mark.mpi_skip
 def test_limits():
     raise
diff --git a/tests/test_propulator.py b/tests/test_propulator.py
@@ -4,10 +4,13 @@
 import logging
 from copy import deepcopy
 
+import pytest
+
 from propulate import Propulator
 from propulate.utils import get_default_propagator, set_logger_config, sphere
 
 
+@pytest.mark.mpi_skip
 def test_Propulator():
     """
     Test single worker using Propulator to optimize sphere.
@@ -52,6 +55,7 @@ def test_Propulator():
         assert best.loss < 0.8
 
 
+@pytest.mark.mpi_skip
 def test_checkpointing_propulator():
     rng = random.Random(42)  # Separate random number generator for optimization.
     limits = {