def test_keyword_style_search():
particles = Particle.from_search_list(pdgname='p')
assert len(particles) == 2
assert 2212 in particles
assert -2212 in particles
particles = Particle.from_search_list(name='p')
assert len(particles) == 1
assert 2212 in particles
assert Particle.from_search(pdgname='p', particle=True) == 2212
assert Particle.from_search(pdgname='p', particle=False) == -2212
assert Particle.from_search(name='p', particle=True) == 2212
assert Particle.from_search(name='p~', particle=False) == -2212
def test_keyword_lambda_style_search():
particles = Particle.from_search_list(pdgname=lambda x: 'p' == x)
assert len(particles) == 2
assert 2212 in particles
assert -2212 in particles
# Fuzzy name
particles = Particle.from_search_list(name=lambda x: 'p' in x)
assert len(particles) > 2
assert 2212 in particles
assert -2212 in particles
# Name and particle
assert Particle.from_search(name=lambda x: x == 'p', particle=True) == 2212
# Unit based comparison
assert 2212 in Particle.from_search_list(lifetime=lambda x: x > 1 * second)
def test_lambda_style_search():
particles = Particle.from_search_list(lambda p: p.pdgname == 'p')
assert len(particles) == 2
assert 2212 in particles
assert -2212 in particles
assert Particle.from_search(lambda p: p.pdgname == 'p' and p > 0) == 2212
assert Particle.from_search(lambda p: p.pdgname == 'p' and p < 0) == -2212
def test_fuzzy_name_search():
particles = Particle.from_search_list('p~')
assert len(particles) == 1
assert -2212 in particles
