test: implement tests/* for parallel rom

kshitij-maths · kshitij-maths · commit 009a4e0c6129 · 2026-04-02T23:53:50.000+02:00
diff --git a/tests/test_parallel/__init__.py b/tests/test_parallel/__init__.py
@@ -0,0 +1,3 @@
+import matplotlib
+
+matplotlib.use("Agg")
diff --git a/tests/test_parallel/test_ae.py b/tests/test_parallel/test_ae.py
@@ -0,0 +1,20 @@
+import unittest
+import pytest
+import sys
+from unittest import mock
+
+try:
+    from ezyrb.parallel import ReducedOrderModel as ParallelROM
+    import tests.test_ae as original_module
+    HAS_PARALLEL = True
+except ImportError:
+    HAS_PARALLEL = False
+
+if not HAS_PARALLEL:
+    raise unittest.SkipTest("Parallel dependencies missing")
+
+original_module.ROM = ParallelROM
+if hasattr(original_module, 'ReducedOrderModel'):
+    original_module.ReducedOrderModel = ParallelROM
+
+from tests.test_ae import *
diff --git a/tests/test_parallel/test_ann.py b/tests/test_parallel/test_ann.py
@@ -0,0 +1,20 @@
+import unittest
+import pytest
+import sys
+from unittest import mock
+
+try:
+    from ezyrb.parallel import ReducedOrderModel as ParallelROM
+    import tests.test_ann as original_module
+    HAS_PARALLEL = True
+except ImportError:
+    HAS_PARALLEL = False
+
+if not HAS_PARALLEL:
+    raise unittest.SkipTest("Parallel dependencies missing")
+
+original_module.ROM = ParallelROM
+if hasattr(original_module, 'ReducedOrderModel'):
+    original_module.ReducedOrderModel = ParallelROM
+
+from tests.test_ann import *
diff --git a/tests/test_parallel/test_approximation.py b/tests/test_parallel/test_approximation.py
@@ -0,0 +1,20 @@
+import unittest
+import pytest
+import sys
+from unittest import mock
+
+try:
+    from ezyrb.parallel import ReducedOrderModel as ParallelROM
+    import tests.test_approximation as original_module
+    HAS_PARALLEL = True
+except ImportError:
+    HAS_PARALLEL = False
+
+if not HAS_PARALLEL:
+    raise unittest.SkipTest("Parallel dependencies missing")
+
+original_module.ROM = ParallelROM
+if hasattr(original_module, 'ReducedOrderModel'):
+    original_module.ReducedOrderModel = ParallelROM
+
+from tests.test_approximation import *
diff --git a/tests/test_parallel/test_database.py b/tests/test_parallel/test_database.py
@@ -0,0 +1,20 @@
+import unittest
+import pytest
+import sys
+from unittest import mock
+
+try:
+    from ezyrb.parallel import ReducedOrderModel as ParallelROM
+    import tests.test_database as original_module
+    HAS_PARALLEL = True
+except ImportError:
+    HAS_PARALLEL = False
+
+if not HAS_PARALLEL:
+    raise unittest.SkipTest("Parallel dependencies missing")
+
+original_module.ROM = ParallelROM
+if hasattr(original_module, 'ReducedOrderModel'):
+    original_module.ReducedOrderModel = ParallelROM
+
+from tests.test_database import *
diff --git a/tests/test_parallel/test_gpr.py b/tests/test_parallel/test_gpr.py
@@ -0,0 +1,20 @@
+import unittest
+import pytest
+import sys
+from unittest import mock
+
+try:
+    from ezyrb.parallel import ReducedOrderModel as ParallelROM
+    import tests.test_gpr as original_module
+    HAS_PARALLEL = True
+except ImportError:
+    HAS_PARALLEL = False
+
+if not HAS_PARALLEL:
+    raise unittest.SkipTest("Parallel dependencies missing")
+
+original_module.ROM = ParallelROM
+if hasattr(original_module, 'ReducedOrderModel'):
+    original_module.ReducedOrderModel = ParallelROM
+
+from tests.test_gpr import *
diff --git a/tests/test_parallel/test_k_neighbors_regressor.py b/tests/test_parallel/test_k_neighbors_regressor.py
@@ -0,0 +1,20 @@
+import unittest
+import pytest
+import sys
+from unittest import mock
+
+try:
+    from ezyrb.parallel import ReducedOrderModel as ParallelROM
+    import tests.test_k_neighbors_regressor as original_module
+    HAS_PARALLEL = True
+except ImportError:
+    HAS_PARALLEL = False
+
+if not HAS_PARALLEL:
+    raise unittest.SkipTest("Parallel dependencies missing")
+
+original_module.ROM = ParallelROM
+if hasattr(original_module, 'ReducedOrderModel'):
+    original_module.ReducedOrderModel = ParallelROM
+
+from tests.test_k_neighbors_regressor import *
diff --git a/tests/test_parallel/test_linear.py b/tests/test_parallel/test_linear.py
@@ -0,0 +1,20 @@
+import unittest
+import pytest
+import sys
+from unittest import mock
+
+try:
+    from ezyrb.parallel import ReducedOrderModel as ParallelROM
+    import tests.test_linear as original_module
+    HAS_PARALLEL = True
+except ImportError:
+    HAS_PARALLEL = False
+
+if not HAS_PARALLEL:
+    raise unittest.SkipTest("Parallel dependencies missing")
+
+original_module.ROM = ParallelROM
+if hasattr(original_module, 'ReducedOrderModel'):
+    original_module.ReducedOrderModel = ParallelROM
+    
+from tests.test_linear import *
diff --git a/tests/test_parallel/test_nnshift.py b/tests/test_parallel/test_nnshift.py
@@ -0,0 +1,20 @@
+import unittest
+import pytest
+import sys
+from unittest import mock
+
+try:
+    from ezyrb.parallel import ReducedOrderModel as ParallelROM
+    import tests.test_nnshift as original_module
+    HAS_PARALLEL = True
+except ImportError:
+    HAS_PARALLEL = False
+
+if not HAS_PARALLEL:
+    raise unittest.SkipTest("Parallel dependencies missing")
+
+original_module.ROM = ParallelROM
+if hasattr(original_module, 'ReducedOrderModel'):
+    original_module.ReducedOrderModel = ParallelROM
+
+from tests.test_nnshift import *
diff --git a/tests/test_parallel/test_parameter.py b/tests/test_parallel/test_parameter.py
@@ -0,0 +1,20 @@
+import unittest
+import pytest
+import sys
+from unittest import mock
+
+try:
+    from ezyrb.parallel import ReducedOrderModel as ParallelROM
+    import tests.test_parameter as original_module
+    HAS_PARALLEL = True
+except ImportError:
+    HAS_PARALLEL = False
+
+if not HAS_PARALLEL:
+    raise unittest.SkipTest("Parallel dependencies missing")
+
+original_module.ROM = ParallelROM
+if hasattr(original_module, 'ReducedOrderModel'):
+    original_module.ReducedOrderModel = ParallelROM
+
+from tests.test_parameter import *
diff --git a/tests/test_parallel/test_pod.py b/tests/test_parallel/test_pod.py
@@ -0,0 +1,20 @@
+import unittest
+import pytest
+import sys
+from unittest import mock
+
+try:
+    from ezyrb.parallel import ReducedOrderModel as ParallelROM
+    import tests.test_pod as original_module
+    HAS_PARALLEL = True
+except ImportError:
+    HAS_PARALLEL = False
+
+if not HAS_PARALLEL:
+    raise unittest.SkipTest("Parallel dependencies missing")
+
+original_module.ROM = ParallelROM
+if hasattr(original_module, 'ReducedOrderModel'):
+    original_module.ReducedOrderModel = ParallelROM
+
+from tests.test_pod import *
diff --git a/tests/test_parallel/test_podae.py b/tests/test_parallel/test_podae.py
@@ -0,0 +1,24 @@
+import unittest
+import pytest
+import sys
+from unittest import mock
+
+try:
+    from ezyrb.parallel import ReducedOrderModel as ParallelROM
+    import tests.test_podae as original_module
+    HAS_PARALLEL = True
+except ImportError:
+    HAS_PARALLEL = False
+
+if not HAS_PARALLEL:
+    raise unittest.SkipTest("Parallel dependencies missing")
+
+# 2. The "Lazy" Swap (Monkeypatching)
+# We replace the ROM in the original module's namespace 
+# so that decorators like @pytest.mark.parametrize use the Parallel version.
+original_module.ROM = ParallelROM
+if hasattr(original_module, 'ReducedOrderModel'):
+    original_module.ReducedOrderModel = ParallelROM
+
+# 3. Re-export the tests so pytest finds them here
+from tests.test_podae import *
diff --git a/tests/test_parallel/test_radius_neighbors_regressor.py b/tests/test_parallel/test_radius_neighbors_regressor.py
@@ -0,0 +1,20 @@
+import unittest
+import pytest
+import sys
+from unittest import mock
+
+try:
+    from ezyrb.parallel import ReducedOrderModel as ParallelROM
+    import tests.test_radius_neighbors_regressor as original_module
+    HAS_PARALLEL = True
+except ImportError:
+    HAS_PARALLEL = False
+
+if not HAS_PARALLEL:
+    raise unittest.SkipTest("Parallel dependencies missing")
+
+original_module.ROM = ParallelROM
+if hasattr(original_module, 'ReducedOrderModel'):
+    original_module.ReducedOrderModel = ParallelROM
+
+from tests.test_radius_neighbors_regressor import *
diff --git a/tests/test_parallel/test_reducedordermodel.py b/tests/test_parallel/test_reducedordermodel.py
@@ -0,0 +1,20 @@
+import unittest
+import pytest
+import sys
+from unittest import mock
+
+try:
+    from ezyrb.parallel import ReducedOrderModel as ParallelROM
+    import tests.test_reducedordermodel as original_module
+    HAS_PARALLEL = True
+except ImportError:
+    HAS_PARALLEL = False
+
+if not HAS_PARALLEL:
+    raise unittest.SkipTest("Parallel dependencies missing")
+
+original_module.ROM = ParallelROM
+if hasattr(original_module, 'ReducedOrderModel'):
+    original_module.ReducedOrderModel = ParallelROM
+    
+from tests.test_reducedordermodel import *
diff --git a/tests/test_parallel/test_regular_grid.py b/tests/test_parallel/test_regular_grid.py
@@ -0,0 +1,20 @@
+import unittest
+import pytest
+import sys
+from unittest import mock
+
+try:
+    from ezyrb.parallel import ReducedOrderModel as ParallelROM
+    import tests.test_regular_grid as original_module
+    HAS_PARALLEL = True
+except ImportError:
+    HAS_PARALLEL = False
+
+if not HAS_PARALLEL:
+    raise unittest.SkipTest("Parallel dependencies missing")
+
+original_module.ROM = ParallelROM
+if hasattr(original_module, 'ReducedOrderModel'):
+    original_module.ReducedOrderModel = ParallelROM
+
+from tests.test_regular_grid import *
diff --git a/tests/test_parallel/test_scaler.py b/tests/test_parallel/test_scaler.py
@@ -0,0 +1,20 @@
+import unittest
+import pytest
+import sys
+from unittest import mock
+
+try:
+    from ezyrb.parallel import ReducedOrderModel as ParallelROM
+    import tests.test_scaler as original_module
+    HAS_PARALLEL = True
+except ImportError:
+    HAS_PARALLEL = False
+
+if not HAS_PARALLEL:
+    raise unittest.SkipTest("Parallel dependencies missing")
+
+original_module.ROM = ParallelROM
+if hasattr(original_module, 'ReducedOrderModel'):
+    original_module.ReducedOrderModel = ParallelROM
+
+from tests.test_scaler import *
diff --git a/tests/test_parallel/test_shift.py b/tests/test_parallel/test_shift.py
@@ -0,0 +1,21 @@
+import unittest
+import pytest
+import sys
+from unittest import mock
+
+# 1. Shield the imports
+try:
+    from ezyrb.parallel import ReducedOrderModel as ParallelROM
+    import tests.test_shift as original_module
+    HAS_PARALLEL = True
+except ImportError:
+    HAS_PARALLEL = False
+
+if not HAS_PARALLEL:
+    raise unittest.SkipTest("Parallel dependencies missing")
+
+original_module.ROM = ParallelROM
+if hasattr(original_module, 'ReducedOrderModel'):
+    original_module.ReducedOrderModel = ParallelROM
+
+from tests.test_shift import *
diff --git a/tests/test_parallel/test_sklearn_wrappers.py b/tests/test_parallel/test_sklearn_wrappers.py
@@ -0,0 +1,20 @@
+import unittest
+import pytest
+import sys
+from unittest import mock
+
+try:
+    from ezyrb.parallel import ReducedOrderModel as ParallelROM
+    import tests.test_sklearn_wrappers as original_module
+    HAS_PARALLEL = True
+except ImportError:
+    HAS_PARALLEL = False
+
+if not HAS_PARALLEL:
+    raise unittest.SkipTest("Parallel dependencies missing")
+
+original_module.ROM = ParallelROM
+if hasattr(original_module, 'ReducedOrderModel'):
+    original_module.ReducedOrderModel = ParallelROM
+
+from tests.test_sklearn_wrappers import *
diff --git a/tests/test_parallel/test_snapshot.py b/tests/test_parallel/test_snapshot.py
@@ -0,0 +1,20 @@
+import unittest
+import pytest
+import sys
+from unittest import mock
+
+try:
+    from ezyrb.parallel import ReducedOrderModel as ParallelROM
+    import tests.test_snapshot as original_module
+    HAS_PARALLEL = True
+except ImportError:
+    HAS_PARALLEL = False
+
+if not HAS_PARALLEL:
+    raise unittest.SkipTest("Parallel dependencies missing")
+
+original_module.ROM = ParallelROM
+if hasattr(original_module, 'ReducedOrderModel'):
+    original_module.ReducedOrderModel = ParallelROM
+
+from tests.test_snapshot import *

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+import matplotlib`
	`2`	`+`
	`3`	`+matplotlib.use("Agg")`