コード例 #1
0
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'
コード例 #2
0
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
    ])