Add corresponding test

scikit-hep · Jul 21, 2023 · 923db49 · 923db49
1 parent 5c97942
commit 923db49
Showing 1 changed file with 27 additions and 2 deletions.
diff --git a/tests/dec/test_dec.py b/tests/dec/test_dec.py
@@ -227,11 +227,36 @@ def test_jetset_definitions_parsing():
     }
 
 
-def test_list_lineshape_definitions():
+def test_dict_lineshape_settings():
     p = DecFileParser(DIR / "../data/defs-aliases-chargeconj.dec")
     p.parse()
 
-    assert p.list_lineshape_definitions() == [
+    assert p.dict_lineshape_settings() == {
+        "MyK*0": {
+            "lineshape": "LSNONRELBW",
+            "BlattWeisskopf": 0.0,
+            "ChangeMassMin": 0.5,
+            "ChangeMassMax": 3.5,
+        },
+        "MyPhi": {
+            "lineshape": "LSNONRELBW",
+            "BlattWeisskopf": 0.0,
+            "ChangeMassMin": 1.0,
+            "ChangeMassMax": 1.04,
+        },
+        "MyKS0pipi": {
+            "lineshape": "LSFLAT",
+            "ChangeMassMin": 1.1,
+            "ChangeMassMax": 2.4,
+        },
+    }
+
+
+def test_list_lineshapePW_definitions():
+    p = DecFileParser(DIR / "../data/defs-aliases-chargeconj.dec")
+    p.parse()
+
+    assert p.list_lineshapePW_definitions() == [
         (["D_1+", "D*+", "pi0"], 2),
         (["D_1+", "D*0", "pi+"], 2),
         (["D_1-", "D*-", "pi0"], 2),