BEHAVIOR: return mass with index for State (#172)

* MAINT: update test hashes

Author: redeboer

src/ampform_dpd/__init__.py

@@ -31,6 +31,7 @@
     get_decay_product_ids,
     to_particle,
 )
+from ampform_dpd.dynamics.builder import create_mass_symbol
 from ampform_dpd.spin import create_spin_range
 
 if TYPE_CHECKING:
@@ -485,7 +486,7 @@ def formulate_non_resonant(
 
 def create_mass_symbol_mapping(decay: ThreeBodyDecay) -> dict[sp.Symbol, float]:
     return {
-        sp.Symbol(f"m{i}", nonnegative=True): decay.states[i].mass
+        create_mass_symbol(decay.states[i]): decay.states[i].mass
         for i in sorted(decay.states)  # ensure that dict keys are sorted by state ID
     }
 

src/ampform_dpd/dynamics/builder.py

@@ -107,8 +107,10 @@ def _create_meson_radius_symbol(isobar: IsobarNode) -> sp.Symbol:
     return sp.Symbol(R"R_\mathrm{res}", nonnegative=True)
 
 
-def create_mass_symbol(particle: IsobarNode | Particle) -> sp.Symbol:
+def create_mass_symbol(particle: IsobarNode | Particle | State) -> sp.Symbol:
     particle = to_particle(particle)
+    if isinstance(particle, State):
+        return sp.Symbol(f"m{particle.index}", nonnegative=True)
     return sp.Symbol(f"m_{{{particle.latex}}}", nonnegative=True)
 
 

tests/test_cached.py

@@ -18,8 +18,8 @@
 @pytest.mark.parametrize(
     ("min_ls", "expected_hashes"),
     [
-        pytest.param(True, {"ba9b5fd"}, id="min-ls"),
-        pytest.param(False, {"8039460"}, id="all-ls"),
+        pytest.param(True, {"544cb15"}, id="min-ls"),
+        pytest.param(False, {"01968bc", "7300a10"}, id="all-ls"),
     ],
 )
 def test_hashes(reaction: ReactionInfo, min_ls: bool, expected_hashes: set[str]):
@@ -51,30 +51,30 @@ def test_amplitude_doit_hashes(reaction: ReactionInfo):
     }
     print(hashes)
     assert hashes == {
-        "A2[-1,0,-1/2,-1/2]": "6770935",
-        "A3[-1,0,-1/2,-1/2]": "5b5001a",
-        "A2[-1,0,-1/2,1/2]": "716e629",
-        "A3[-1,0,-1/2,1/2]": "d338945",
-        "A2[-1,0,1/2,-1/2]": "78bf694",
-        "A3[-1,0,1/2,-1/2]": "b72a7e2",
-        "A2[-1,0,1/2,1/2]": "b1e428d",
-        "A3[-1,0,1/2,1/2]": "8730459",
-        "A2[0,0,-1/2,-1/2]": "c9cb5e9",
-        "A3[0,0,-1/2,-1/2]": "a62c07e",
-        "A2[0,0,-1/2,1/2]": "b87152c",
-        "A3[0,0,-1/2,1/2]": "8961c3b",
-        "A2[0,0,1/2,-1/2]": "73f76f4",
-        "A3[0,0,1/2,-1/2]": "82d805d",
-        "A2[0,0,1/2,1/2]": "36dd5d0",
-        "A3[0,0,1/2,1/2]": "ac73d2d",
-        "A2[1,0,-1/2,-1/2]": "7f8513f",
-        "A3[1,0,-1/2,-1/2]": "3691bba",
-        "A2[1,0,-1/2,1/2]": "165a2ad",
-        "A3[1,0,-1/2,1/2]": "7c7b3cf",
-        "A2[1,0,1/2,-1/2]": "e3ff1a7",
-        "A3[1,0,1/2,-1/2]": "06027b1",
-        "A2[1,0,1/2,1/2]": "eb57212",
-        "A3[1,0,1/2,1/2]": "3ef2476",
+        "A2[-1,0,-1/2,-1/2]": "652522d",
+        "A3[-1,0,-1/2,-1/2]": "5d8b698",
+        "A2[-1,0,-1/2,1/2]": "0a944fe",
+        "A3[-1,0,-1/2,1/2]": "a564151",
+        "A2[-1,0,1/2,-1/2]": "4732f96",
+        "A3[-1,0,1/2,-1/2]": "67bc82d",
+        "A2[-1,0,1/2,1/2]": "98370c7",
+        "A3[-1,0,1/2,1/2]": "00d5d44",
+        "A2[0,0,-1/2,-1/2]": "aef2ca4",
+        "A3[0,0,-1/2,-1/2]": "bcb25df",
+        "A2[0,0,-1/2,1/2]": "00c9500",
+        "A3[0,0,-1/2,1/2]": "7f73b98",
+        "A2[0,0,1/2,-1/2]": "f3ade16",
+        "A3[0,0,1/2,-1/2]": "96ec036",
+        "A2[0,0,1/2,1/2]": "81dcaa6",
+        "A3[0,0,1/2,1/2]": "6a547e5",
+        "A2[1,0,-1/2,-1/2]": "09ca576",
+        "A3[1,0,-1/2,-1/2]": "b3ed335",
+        "A2[1,0,-1/2,1/2]": "beed520",
+        "A3[1,0,-1/2,1/2]": "115597f",
+        "A2[1,0,1/2,-1/2]": "f6e2775",
+        "A3[1,0,1/2,-1/2]": "0d898ee",
+        "A2[1,0,1/2,1/2]": "9276158",
+        "A3[1,0,1/2,1/2]": "f3fc8c5",
     }