def test_DecayMode_charge_conjugate(): dd = DaughtersDict('pi- pi0 nu(tau)') dm = DecayMode(0.2551, dd, model='TAUHADNU', model_params=[-0.108, 0.775, 0.149, 1.364, 0.400]) dm_cc = dm.charge_conjugate() assert dm_cc.daughters == DaughtersDict('pi+ pi0 nu(tau)~') assert 'BF: 0.2551' in dm.describe() assert 'Decay model: TAUHADNU [-0.108, 0.775, 0.149, 1.364, 0.4]' in dm.describe( )
def test_DecayMode_constructor_from_pdgids(): dm = DecayMode.from_pdgids( 0.5, [321, -321], model="TAUHADNU", model_params=[-0.108, 0.775, 0.149, 1.364, 0.400], ) assert dm.daughters == DaughtersDict("K+ K-")
def test_DecayMode_constructor_with_user_model_info(): dd = DaughtersDict('K+ K-') dm = DecayMode(0.5, dd, model='PHSP', study='toy', year=2019) assert dm.metadata == { 'model': 'PHSP', 'model_params': '', 'study': 'toy', 'year': 2019 }
def test_DecayMode_describe_simple(): dd = DaughtersDict('pi- pi0 nu_tau') dm = DecayMode(0.2551, dd, model='TAUHADNU', model_params=[-0.108, 0.775, 0.149, 1.364, 0.400]) assert 'BF: 0.2551' in dm.describe() assert 'Decay model: TAUHADNU [-0.108, 0.775, 0.149, 1.364, 0.4]' in dm.describe( )
def test_DecayMode_describe_simple(): dd = DaughtersDict("pi- pi0 nu_tau") dm = DecayMode(0.2551, dd, model="TAUHADNU", model_params=[-0.108, 0.775, 0.149, 1.364, 0.400]) assert "BF: 0.2551" in dm.describe() assert "Decay model: TAUHADNU [-0.108, 0.775, 0.149, 1.364, 0.4]" in dm.describe( )
def test_DecayMode_constructor_with_user_metadata(): dd = DaughtersDict("K+ K-") dm = DecayMode(0.5, dd, model="PHSP", study="toy", year=2019) assert dm.metadata == { "model": "PHSP", "model_params": "", "study": "toy", "year": 2019, }
def test_DecayMode_constructor_with_model_info(): dd = DaughtersDict("pi- pi0 nu_tau") dm = DecayMode(0.2551, dd, model="TAUHADNU", model_params=[-0.108, 0.775, 0.149, 1.364, 0.400]) assert dm.metadata == { "model": "TAUHADNU", "model_params": [-0.108, 0.775, 0.149, 1.364, 0.4], }
def test_DecayMode_constructor_with_model_info(): dd = DaughtersDict('pi- pi0 nu_tau') dm = DecayMode(0.2551, dd, model='TAUHADNU', model_params=[-0.108, 0.775, 0.149, 1.364, 0.400]) assert dm.metadata == { 'model': 'TAUHADNU', 'model_params': [-0.108, 0.775, 0.149, 1.364, 0.4] }
def test_DecayChain_flatten_complex(): dm1 = DecayMode(1.0, "D0 K_S0 pi+ pi0") dm2 = DecayMode(0.0124, "K_S0 pi0 pi0") dm3 = DecayMode(0.692, "pi+ pi-") dm4 = DecayMode(0.98823, "gamma gamma") dc = DecayChain("X+", {"X+": dm1, "D0": dm2, "K_S0": dm3, "pi0": dm4}) dc_flatten = dc.flatten() assert dc_flatten.decays[dc_flatten.mother].daughters == DaughtersDict({ "gamma": 6, "pi+": 3, "pi-": 2 }) assert dc_flatten.bf == approx(1.0 * 0.0124 * (0.692**2) * (0.98823**3))
def test_DecayMode_charge_conjugate(): dd = DaughtersDict("pi- pi0 nu_tau") dm = DecayMode(0.2551, dd, model="TAUHADNU", model_params=[-0.108, 0.775, 0.149, 1.364, 0.400]) dm_cc = dm.charge_conjugate() assert dm_cc.daughters == DaughtersDict("pi+ pi0 anti-nu_tau") assert "BF: 0.2551" in dm.describe() assert "Decay model: TAUHADNU [-0.108, 0.775, 0.149, 1.364, 0.4]" in dm.describe( ) dd = DaughtersDict("pi- pi0 nu(tau)") assert dd.charge_conjugate( pdg_name=True) == DaughtersDict("pi+ pi0 nu(tau)~")
def test_DecayChain_flatten_with_stable_particles(): dm1 = DecayMode(0.5, "D0 anti-D0 pi+ pi0 pi0") dm2 = DecayMode(0.0124, "K_S0 pi0") dm3 = DecayMode(0.692, "pi+ pi-") dm4 = DecayMode(0.98823, "gamma gamma") dc = DecayChain("X+", { "X+": dm1, "D0": dm2, "anti-D0": dm2, "K_S0": dm3, "pi0": dm4 }) dc_flatten = dc.flatten(stable_particles=["pi0"]) assert dc_flatten.decays[dc_flatten.mother].daughters == DaughtersDict({ "pi0": 4, "pi+": 3, "pi-": 2 }) assert dc_flatten.bf == approx(0.5 * (0.0124**2) * (0.692**2))
def test_DaughtersDict_add(): dd1 = DaughtersDict({'K+': 1, 'K-': 2, 'pi0': 3}) dd2 = DaughtersDict({'K+': 1, 'K-': 1}) dd3 = dd1 + dd2 assert len(dd3) == 8
def test_DaughtersDict_to_string(): dd1 = DaughtersDict({'K+': 1, 'K-': 2, 'pi0': 3}) assert dd1.to_string() == 'K+ K- K- pi0 pi0 pi0'
def test_DaughtersDict_string_repr(): dd = DaughtersDict(['K+', 'K-', 'K-', 'pi+', 'pi0']) assert dd.__str__() == "<DaughtersDict: ['K+', 'K-', 'K-', 'pi+', 'pi0']>"
def test_DaughtersDict_len(): dd = DaughtersDict({'K+': 1, 'K-': 3, 'pi0': 1}) assert len(dd) == 5
def test_DaughtersDict_constructor_from_list(): dd = DaughtersDict(['K+', 'K-', 'K-', 'pi+', 'pi0']) assert dd == {'K+': 1, 'K-': 2, 'pi+': 1, 'pi0': 1}
def test_DaughtersDict_constructor_from_string(): dd = DaughtersDict('K+ K- pi0') assert dd == {'K+': 1, 'K-': 1, 'pi0': 1}
def test_DecayMode_string_repr(): dd = DaughtersDict('p p~ K+ pi-') dm = DecayMode(1.e-6, dd, model='PHSP') assert str(dm) == "<DecayMode: daughters=K+ p pi- p~, BF=1e-06>"
def test_DecayMode_number_of_final_states(): dd = DaughtersDict('p p~ K+ pi-') dm = DecayMode(1.e-6, dd, model='PHSP') assert len(dm) == 4
def test_DaughtersDict_to_string(): dd1 = DaughtersDict({"K+": 1, "K-": 2, "pi0": 3}) assert dd1.to_string() == "K+ K- K- pi0 pi0 pi0"
def test_DaughtersDict_add(): dd1 = DaughtersDict({"K+": 1, "K-": 2, "pi0": 3}) dd2 = DaughtersDict({"K+": 1, "K-": 1}) dd3 = dd1 + dd2 assert len(dd3) == 8
def test_DaughtersDict_len(): dd = DaughtersDict({"K+": 1, "K-": 3, "pi0": 1}) assert len(dd) == 5
def test_DecayMode_constructor_default(): dm = DecayMode() assert dm.bf == 0 assert dm.daughters == DaughtersDict() assert dm.metadata == dict(model='', model_params='')
def test_DaughtersDict_charge_conjugate(): dd = DaughtersDict({"K+": 2, "pi0": 1}) assert dd.charge_conjugate() == {"K-": 2, "pi0": 1}
def test_DaughtersDict_charge_conjugate_pdg_names(): dd = DaughtersDict({"K(S)0": 1, "pi+": 1}) # PDG names! assert dd == {"K(S)0": 1, "pi+": 1} # PDG names kept as-is assert dd.charge_conjugate() == {"ChargeConj(K(S)0)": 1, "pi-": 1} assert dd.charge_conjugate(pdg_name=True) == {"K(S)0": 1, "pi-": 1}
def test_DecayMode_constructor_simplest(): dm = DecayMode(0.1234, "K+ K-") assert dm.bf == 0.1234 assert dm.daughters == DaughtersDict("K+ K-") assert dm.metadata == dict(model="", model_params="")
def test_DecayMode_constructor_simple(): dd = DaughtersDict('K+ K-') dm = DecayMode(0.1234, dd) assert dm.bf == 0.1234 assert dm.daughters == DaughtersDict('K+ K-') assert dm.metadata == dict(model='', model_params='')
def test_DecayMode_describe_with_extra_info(): dd = DaughtersDict('K+ K-') dm = DecayMode(1.e-6, dd, model='PHSP', study='toy', year=2019) assert 'Extra info:' in dm.describe() assert 'study: toy' in dm.describe() assert 'year: 2019' in dm.describe()
def test_DaughtersDict_constructor_from_dict(): dd = DaughtersDict({'K+': 1, 'K-': 2, 'pi+': 1, 'pi0': 1}) assert dd == {'K+': 1, 'K-': 2, 'pi+': 1, 'pi0': 1}
def test_DecayChain_flatten(dc2): dc2_flatten = dc2.flatten() assert dc2_flatten.mother == dc2.mother assert dc2_flatten.bf == dc2_flatten.visible_bf assert dc2_flatten.decays[dc2.mother].daughters == DaughtersDict( ["gamma", "gamma", "pi+", "pi+", "pi-"])