def set_dac_unit(self, unit: str) -> None:
"""Set the unit of dac parameters"""
allowed_values = Enum('mV', 'V')
allowed_values.validate(unit)
self._gain = {'V': 1, 'mV': 1e3}[unit]
for i in range(1, self._number_dacs + 1):
setattr(self.parameters[f'dac{i}'], 'unit', unit)
def test_good(self):
for enum in self.enums:
e = Enum(*enum)
for v in enum:
e.validate(v)
for v in [22, 'bad data', [44, 55]]:
with self.assertRaises((ValueError, TypeError)):
e.validate(v)
self.assertEqual(repr(e), '<Enum: {}>'.format(repr(set(enum))))
# Enum is never numeric, even if its members are all numbers
# because the use of is_numeric is for sweeping
self.assertFalse(e.is_numeric)
def test_good():
for enum in enums:
e = Enum(*enum)
for v in enum:
e.validate(v)
for v in [22, 'bad data', [44, 55]]:
with pytest.raises((ValueError, TypeError)):
e.validate(v)
assert repr(e) == f"<Enum: {repr(set(enum))}>"
# Enum is never numeric, even if its members are all numbers
# because the use of is_numeric is for sweeping
assert not e.is_numeric
def test_valid_values():
for enum in enums:
e = Enum(*enum)
for val in e._valid_values:
e.validate(val)