def __setitem__(self, key, item):
if key is None:
return
if isinstance(key, str):
key = ep.get_atomic_number(key)
if key < 1:
raise ValueError('Atomic number cannot be less than hydrogen')
if key > 118:
raise ValueError('Atomic number cannot be greater than Uuo')
self._data[np.uint8(key)] = convert_value(item)
def setSelection(self, selection):
def _uncheckedAll():
for widget in self._group.buttons():
widget.setChecked(False)
if selection is None:
_uncheckedAll()
self.selectionChanged.emit()
return
if isinstance(selection, (int, six.string_types)):
selection = [selection]
if not self.isMultipleSelection() and len(selection) > 1:
raise ValueError('Multiple selection mode is off. Cannot select more than one element')
_uncheckedAll()
for z in selection:
if isinstance(z, six.string_types):
z = get_atomic_number(z)
self._group.button(z).setChecked(True)
self.selectionChanged.emit()
def testatomic_number(self):
self.assertEqual(1, get_atomic_number('H'))
self.assertEqual(118, get_atomic_number('Uuo'))
def __delitem__(self, key):
if isinstance(key, str):
key = ep.get_atomic_number(key)
del self._data[key]
def __getitem__(self, key):
if isinstance(key, str):
key = ep.get_atomic_number(key)
return self._data[key]
def atomicNumber(self):
symbol = self.text()
if symbol == str(None):
return None
return get_atomic_number(symbol)
def set_symbol(self, symbol):
self.atomic_number = get_atomic_number(symbol)
