def test_Lark_ChargeConjugateReplacement_Visitor_with_aliases():
"""
Example with a D0 decay specified via an alias (MyD0).
As such, it is necessary to state what the particle-antiparticle match is,
which in decay files would mean the following lines:
Alias MyD0 D0
Alias MyAnti-D0 anti-D0
ChargeConj MyD0 MyAnti-D0
A dictionary of matches should be passed to the Lark Visitor instance.
"""
t = Tree('decay', [
Tree('particle', [Token('LABEL', 'MyD0')]),
Tree('decayline', [
Tree('value', [Token('SIGNED_NUMBER', '1.0')]),
Tree('particle', [Token('LABEL', 'K-')]),
Tree('particle', [Token('LABEL', 'pi+')]),
Tree('model', [Token('MODEL_NAME', 'PHSP')])
])
])
dict_ChargeConj_defs = {'MyD0': 'MyAnti-D0'}
ChargeConjugateReplacement(charge_conj_defs=dict_ChargeConj_defs).visit(t)
assert get_decay_mother_name(t) == 'MyAnti-D0'
assert get_final_state_particle_names(t.children[1]) == ['K+', 'pi-']
def test_Lark_ChargeConjugateReplacement_Visitor_with_aliases():
"""
Example with a D0 decay specified via an alias (MyD0).
As such, it is necessary to state what the particle-antiparticle match is,
which in decay files would mean the following lines:
Alias MyD0 D0
Alias MyAnti-D0 anti-D0
ChargeConj MyD0 MyAnti-D0
A dictionary of matches should be passed to the Lark Visitor instance.
"""
t = Tree(
"decay",
[
Tree("particle", [Token("LABEL", "MyD0")]),
Tree(
"decayline",
[
Tree("value", [Token("SIGNED_NUMBER", "1.0")]),
Tree("particle", [Token("LABEL", "K-")]),
Tree("particle", [Token("LABEL", "pi+")]),
Tree("model", [Token("MODEL_NAME", "PHSP")]),
],
),
],
)
dict_ChargeConj_defs = {"MyD0": "MyAnti-D0"}
ChargeConjugateReplacement(charge_conj_defs=dict_ChargeConj_defs).visit(t)
assert get_decay_mother_name(t) == "MyAnti-D0"
assert get_final_state_particle_names(t.children[1]) == ["K+", "pi-"]
def test_Lark_ChargeConjugateReplacement_Visitor():
"""
A simple example usage of the ChargeConjugateReplacement implementation
of a Lark's Visitor, here replacing all particles in a 'decay' Tree
by their antiparticles.
"""
t = Tree(
"decay",
[
Tree("particle", [Token("LABEL", "D0")]),
Tree(
"decayline",
[
Tree("value", [Token("SIGNED_NUMBER", "1.0")]),
Tree("particle", [Token("LABEL", "K-")]),
Tree("particle", [Token("LABEL", "pi+")]),
Tree("model", [Token("MODEL_NAME", "PHSP")]),
],
),
],
)
ChargeConjugateReplacement().visit(t)
assert get_decay_mother_name(t) == "anti-D0"
assert get_final_state_particle_names(t.children[1]) == ["K+", "pi-"]
def test_Lark_ChargeConjugateReplacement_Visitor():
"""
A simple example usage of the ChargeConjugateReplacement implementation
of a Lark's Visitor, here replacing all particles in a 'decay' Tree
by their antiparticles.
"""
t = Tree('decay', [
Tree('particle', [Token('LABEL', 'D0')]),
Tree('decayline', [
Tree('value', [Token('SIGNED_NUMBER', '1.0')]),
Tree('particle', [Token('LABEL', 'K-')]),
Tree('particle', [Token('LABEL', 'pi+')]),
Tree('model', [Token('MODEL_NAME', 'PHSP')])
])
])
ChargeConjugateReplacement().visit(t)
assert get_decay_mother_name(t) == 'anti-D0'
assert get_final_state_particle_names(t.children[1]) == ['K+', 'pi-']