def test_init__invalid_choices(self):
with pytest.raises(ValueError, match="choices contains a non"):
argument_actions.EnumName(
option_strings="--colour",
dest="colour",
type=Colour,
choices=(Colour.Blue, "Pink"),
)
def test_init__valid_choices(self):
target = argument_actions.EnumName(
option_strings="--colour",
dest="colour",
type=Colour,
choices=(Colour.Blue, Colour.Red),
)
assert target.choices == ("Blue", "Red")
def test_init__type_not_an_enum(self):
with pytest.raises(TypeError, match="type must be an Enum"):
argument_actions.EnumName(option_strings="--colour",
type=str,
dest="colour")
def name_target(self):
return argument_actions.EnumName(option_strings="--colour",
dest="colour",
type=Colour)
def test_init__type_not_provided(self):
with pytest.raises(ValueError, match="type must be assigned an Enum"):
argument_actions.EnumName(option_strings="--colour", dest="colour")