def test_Lark_DecayModelParamValueReplacement_Visitor_single_value():
t = Tree('decay', [
Tree('particle', [Token('LABEL', 'Upsilon(4S)')]),
Tree('decayline', [
Tree('value', [Token('SIGNED_NUMBER', '1.0')]),
Tree('particle', [Token('LABEL', 'B0')]),
Tree('particle', [Token('LABEL', 'anti-B0')]),
Tree('model', [
Token('MODEL_NAME', 'VSS_BMIX'),
Tree('model_options', [Token('LABEL', 'dm')])
])
])
])
DecayModelParamValueReplacement().visit(t)
# Nothing done since model parameter name has no corresponding
# 'Define' statement from which the actual value can be inferred
tree_decayline = list(t.find_data('decayline'))[0]
assert get_model_name(tree_decayline) == 'VSS_BMIX'
assert get_model_parameters(tree_decayline) == ['dm']
dict_define_defs = {'dm': 0.507e12}
DecayModelParamValueReplacement(define_defs=dict_define_defs).visit(t)
# The model parameter 'dm' should now be replaced by its value
assert get_model_name(tree_decayline) == 'VSS_BMIX'
assert get_model_parameters(tree_decayline) == [507000000000.0]
def test_Lark_DecayModelParamValueReplacement_Visitor_list():
t = Tree('decay', [
Tree('particle', [Token('LABEL', 'B0sig')]),
Tree('decayline', [
Tree('value', [Token('SIGNED_NUMBER', '1.000')]),
Tree('particle', [Token('LABEL', 'MyFirstD*-')]),
Tree('particle', [Token('LABEL', 'MySecondD*+')]),
Tree('model', [
Token('MODEL_NAME', 'SVV_HELAMP'),
Tree('model_options', [
Tree('value', [Token('SIGNED_NUMBER', '0.0')]),
Tree('value', [Token('SIGNED_NUMBER', '0.0')]),
Tree('value', [Token('SIGNED_NUMBER', '0.0')]),
Tree('value', [Token('SIGNED_NUMBER', '0.0')]),
Tree('value', [Token('SIGNED_NUMBER', '1.0')]),
Tree('value', [Token('SIGNED_NUMBER', '0.0')])
])
])
])
])
DecayModelParamValueReplacement().visit(t)
# The visitor should do nothing in this case
tree_decayline = list(t.find_data('decayline'))[0]
assert get_model_name(tree_decayline) == 'SVV_HELAMP'
assert get_model_parameters(tree_decayline) == [
0.0, 0.0, 0.0, 0.0, 1.0, 0.0
]
def test_Lark_DecayModelParamValueReplacement_Visitor_single_value():
t = Tree(
"decay",
[
Tree("particle", [Token("LABEL", "Upsilon(4S)")]),
Tree(
"decayline",
[
Tree("value", [Token("SIGNED_NUMBER", "1.0")]),
Tree("particle", [Token("LABEL", "B0")]),
Tree("particle", [Token("LABEL", "anti-B0")]),
Tree(
"model",
[
Token("MODEL_NAME", "VSS_BMIX"),
Tree("model_options", [Token("LABEL", "dm")]),
],
),
],
),
],
)
DecayModelParamValueReplacement().visit(t)
# Nothing done since model parameter name has no corresponding
# 'Define' statement from which the actual value can be inferred
tree_decayline = list(t.find_data("decayline"))[0]
assert get_model_name(tree_decayline) == "VSS_BMIX"
assert get_model_parameters(tree_decayline) == ["dm"]
dict_define_defs = {"dm": 0.507e12}
DecayModelParamValueReplacement(define_defs=dict_define_defs).visit(t)
# The model parameter 'dm' should now be replaced by its value
assert get_model_name(tree_decayline) == "VSS_BMIX"
assert get_model_parameters(tree_decayline) == [507000000000.0]
def test_Lark_DecayModelParamValueReplacement_Visitor_list():
t = Tree(
"decay",
[
Tree("particle", [Token("LABEL", "B0sig")]),
Tree(
"decayline",
[
Tree("value", [Token("SIGNED_NUMBER", "1.000")]),
Tree("particle", [Token("LABEL", "MyFirstD*-")]),
Tree("particle", [Token("LABEL", "MySecondD*+")]),
Tree(
"model",
[
Token("MODEL_NAME", "SVV_HELAMP"),
Tree(
"model_options",
[
Tree("value",
[Token("SIGNED_NUMBER", "0.0")]),
Tree("value",
[Token("SIGNED_NUMBER", "0.0")]),
Tree("value",
[Token("SIGNED_NUMBER", "0.0")]),
Tree("value",
[Token("SIGNED_NUMBER", "0.0")]),
Tree("value",
[Token("SIGNED_NUMBER", "1.0")]),
Tree("value",
[Token("SIGNED_NUMBER", "0.0")]),
],
),
],
),
],
),
],
)
DecayModelParamValueReplacement().visit(t)
# The visitor should do nothing in this case
tree_decayline = list(t.find_data("decayline"))[0]
assert get_model_name(tree_decayline) == "SVV_HELAMP"
assert get_model_parameters(tree_decayline) == [
0.0, 0.0, 0.0, 0.0, 1.0, 0.0
]
def test_Lark_DecayModelParamValueReplacement_Visitor_no_params():
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')])
])
])
DecayModelParamValueReplacement().visit(t)
# The visitor should do nothing in this case
tree_decayline = list(t.find_data('decayline'))[0]
assert get_model_name(tree_decayline) == 'PHSP'
assert get_model_parameters(tree_decayline) == ''
def test_Lark_DecayModelParamValueReplacement_Visitor_no_params():
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")]),
],
),
],
)
DecayModelParamValueReplacement().visit(t)
# The visitor should do nothing in this case
tree_decayline = list(t.find_data("decayline"))[0]
assert get_model_name(tree_decayline) == "PHSP"
assert get_model_parameters(tree_decayline) == ""