def test_is_baryon(PDGIDs): _baryons = (PDGIDs.Proton, PDGIDs.AntiNeutron, PDGIDs.Lambda, PDGIDs.Sigma0, PDGIDs.SigmaPlus, PDGIDs.SigmaMinus, PDGIDs.Xi0, PDGIDs.AntiXiMinus,PDGIDs.OmegaMinus, PDGIDs.LcPlus, PDGIDs.Lb, PDGIDs.LtPlus, PDGIDs.RPlusPlus_GTildeUUU, PDGIDs.UCbarCUDPentaquark, PDGIDs.AntiUCbarCUDPentaquark) _non_baryons = [id for id in PDGIDs if id not in _baryons] for id in _baryons: assert is_baryon(id) == True for id in _non_baryons: assert is_baryon(id) == False
def test_is_hadron(PDGIDs): for id in PDGIDs: assert is_hadron(id) == (is_meson(id) or is_baryon(id))