Add support for EvtGen model HypNonLepton

scikit-hep · May 29, 2024 · 7391cb7 · 7391cb7
1 parent 148bb4b
commit 7391cb7
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/src/decaylanguage/dec/enums.py b/src/decaylanguage/dec/enums.py
@@ -74,6 +74,7 @@ class PhotosEnum(IntEnum):
     "HQET3",
     "HQET2",
     "HQET",
+    "HypNonLepton",
     "ISGW2",
     "ISGW",
     "KNUNU",

diff --git a/tests/dec/test_known_decay_models.py b/tests/dec/test_known_decay_models.py
@@ -15,7 +15,6 @@
 DIR = Path(__file__).parent.resolve()
 
 
-# TODO: actually test all models - takes time
 parsed_models = (
     ("B_TO_2BARYON_SCALAR", ""),
     ("B_TO_LAMBDA_PBAR_GAMMA", ""),
@@ -140,6 +139,7 @@
     ("HQET3", [0.920, 1.205, 1.21, 1.404, 0.854]),
     ("HQET2", [1.18, 1.074]),
     ("HQET", [0.77, 1.33, 0.92]),
+    ("HypNonLepton", [0.748, -6.5]),
     ("ISGW2", ""),
     ("ISGW", ""),
     ("KNUNU", ""),
@@ -419,10 +419,6 @@
 )
 
 
-def test_parsing_of_all_known_models_are_tested():
-    assert len(parsed_models) == len(known_decay_models)
-
-
 @pytest.mark.parametrize(("decay_model", "expected_model_parameters"), parsed_models)
 def test_model_parsing(decay_model: str, expected_model_parameters: str):
     """
@@ -437,3 +433,7 @@ def test_model_parsing(decay_model: str, expected_model_parameters: str):
 
     assert get_model_name(parsed_Tree) == decay_model
     assert get_model_parameters(parsed_Tree) == expected_model_parameters
+
+
+def test_parsing_of_all_known_models_are_tested():
+    assert len(parsed_models) == len(known_decay_models)