diff --git a/src/decaylanguage/dec/enums.py b/src/decaylanguage/dec/enums.py index abf6da9f..9ecdbb14 100644 --- 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 index 290acd7e..c813f910 100644 --- 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)