test: Only run pythia8 tests if enabled (#1739)

acts-project · Dec 14, 2022 · 0aa3259 · 0aa3259
1 parent d4aac95
commit 0aa3259
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/Examples/Python/tests/helpers/__init__.py b/Examples/Python/tests/helpers/__init__.py
@@ -47,6 +47,14 @@
 except ImportError:
     edm4hepEnabled = False
 
+
+try:
+    import acts.examples
+
+    pythia8Enabled = hasattr(acts.examples, "pythia8")
+except ImportError:
+    pythia8Enabled = False
+
 isCI = os.environ.get("CI", "false") == "true"
 
 

diff --git a/Examples/Python/tests/test_examples.py b/Examples/Python/tests/test_examples.py
@@ -11,6 +11,7 @@
     rootEnabled,
     dd4hepEnabled,
     hepmc3Enabled,
+    pythia8Enabled,
     AssertCollectionExistsAlg,
     isCI,
     doHashChecks,
@@ -65,6 +66,7 @@ def assert_entries(root_file, tree_name, exp):
 
 
 @pytest.mark.slow
+@pytest.mark.skipif(not pythia8Enabled, reason="Pythia8 not set up")
 def test_pythia8(tmp_path, seq, assert_root_hash):
     from pythia8 import runPythia8
 
@@ -1091,6 +1093,7 @@ def test_ckf_tracks_example(
 
 
 @pytest.mark.skipif(not dd4hepEnabled, reason="DD4hep not set up")
+@pytest.mark.skipif(not pythia8Enabled, reason="Pythia8 not set up")
 @pytest.mark.slow
 @pytest.mark.filterwarnings("ignore::UserWarning")
 def test_vertex_fitting(tmp_path):