Even more elegant error check on element names

This commit is contained in:
Paul Cosgrove 2020-02-27 12:50:44 +00:00
parent d9b2c4dccc
commit 70b2f968f8

View file

@ -530,9 +530,8 @@ class Material(IDManagerMixin):
# Allow for element identifier to be given as a symbol or name
if len(element)>2:
el = element.lower()
if el in openmc.data.ELEMENT_SYMBOL:
element = openmc.data.ELEMENT_SYMBOL[el]
else:
element = openmc.data.ELEMENT_SYMBOL.get(el)
if element is None:
msg = 'Element name "{}" not recognised'.format(el)
raise ValueError(msg)