def test_validate_choices(): validate = cli._validate_choices(['choice1', 'choice2']) assert validate('choice1') assert validate('choice2') with pytest.raises(click.BadParameter): assert validate('choice3')