def test_element_data(Z):
# Cycle through the elements and check that
# the info returned from the functions is consistent
data = lut.element_data_from_Z(Z)
assert data[1] == Z
assert data == lut.element_data_from_sym(data[0])
assert data == lut.element_data_from_name(data[2])
assert data[0] == lut.element_sym_from_Z(Z)
assert data[2] == lut.element_name_from_Z(Z)
nsym = lut.element_sym_from_Z(Z, True)
nname = lut.element_name_from_Z(Z, True)
assert nsym[0] == data[0][0].upper()
assert nname[0] == data[2][0].upper()
def test_element_data_fail(Z):
with pytest.raises(KeyError, match=r'No element data for Z'):
lut.element_data_from_Z(Z)