def test_is_valid():
assert_equal(is_valid("Proton"), True)
assert_equal(is_valid("Protium"), True)
assert_equal(is_valid("Hydrogen"), True)
assert_equal(is_valid(2212), True)
assert_equal(is_valid(-2212), True)
assert_equal(is_valid("Neutron"), True)
assert_equal(is_valid("AntiProton"), True)
assert_equal(is_valid("AntiNeutron"), True)
def test_is_valid():
assert_equal(is_valid("Proton"),True)
assert_equal(is_valid("Protium"),True)
assert_equal(is_valid("Hydrogen"),True)
assert_equal(is_valid(2212),True)
assert_equal(is_valid(-2212),True)
assert_equal(is_valid("Neutron"),True)
assert_equal(is_valid("AntiProton"),True)
assert_equal(is_valid("AntiNeutron"),True)
def particle_split(tally_group_name):
"""
Function that splits group name on the basis of ':'
returns tally particle
"""
try:
group_name = tally_group_name.split('/')
tally_particle = group_name[0].split(':')[1]
except:
raise Exception(
"':' is missing in %s" % tally_group_name)
if (tally_particle == ''):
raise Exception("Tally particle is missing in %s" % tally_group_name)
# chack the validity of the particle name
if (is_valid(tally_particle) == False):
raise Exception(
"Particle included in group %s is not a valid particle name!" % tally_group_name)
return tally_particle
