def test_raise_argument_error_custom_message(argument: cli.Argument, cli_argument_group): argument.register(cli_argument_group) with pytest.raises(argparse.ArgumentError) as error_info: argument.raise_argument_error('Custom error') error_msg = Colors.remove(str(error_info.value)) assert error_msg == f'argument {argument.key}: Custom error'
def test_raise_argument_error(argument: cli.Argument, cli_argument_group): argument.register(cli_argument_group) with pytest.raises(argparse.ArgumentError) as error_info: argument.raise_argument_error() error_msg = Colors.remove(str(error_info.value)) key = argument.key assert error_msg.startswith(f'argument {key}: ') assert any([ f'Value {key.upper()} not provided' in error_msg, f'Missing value {key.upper()}' in error_msg ])