def source(self, src_type):
if src_type is None:
self._source = self.SourceType.undefined
else:
self._source = get_from_enum(src_type, self.SourceType)
def kind(self, int_type):
if int_type is None:
self._interaction_type = self.Type.undefined
else:
self._interaction_type = get_from_enum(int_type, self.Type)
def value_type(self, val_type):
if val_type is None:
self._value_type = self.Type.undefined
else:
self._value_type = get_from_enum(val_type, self.Type)
def id(self, particle_id):
if particle_id is None:
self._id = self.Type.undefined
else:
self._id = get_from_enum(particle_id, self.Type)
def test_get_ambiguous(self, enum):
"""Test that getting ambiguous values from any method works"""
assert get_from_enum(enum.rouge, enum) == enum.red
assert get_from_enum(2, enum) == enum.vert
assert get_from_enum("bleu", enum) == enum.blue
def test_get_str(self, enum):
"""Test that getting value from name works"""
assert get_from_enum("red", enum) == enum.red
assert get_from_enum("green", enum) == enum.green
assert get_from_enum("blue", enum) == enum.blue
def test_get_int(self, enum):
"""Test that getting vlue from integer works"""
assert get_from_enum(1, enum) == enum.red
assert get_from_enum(2, enum) == enum.green
assert get_from_enum(3, enum) == enum.blue
def test_get_value(self, enum):
"""Test that getting value directly from value works"""
assert get_from_enum(enum.red, enum) == enum.red
assert get_from_enum(enum.green, enum) == enum.green
assert get_from_enum(enum.blue, enum) == enum.blue
