def test_excess_args(self): argv = ['ngram.py', '1', 'arg2', 'arg3', 'arg4', 'arg5'] ret = ['1', 'arg2', 'arg3', 'arg4'] assert ngram.parse_args(argv) == ret
def test_invalid_mode_arg(self): argv = ['ngram.py', 'arg1', 'arg2', 'arg3', 'arg4'] with pytest.raises(ngram.UsageError): ngram.parse_args(argv)
def test_not_enough_args(self): argv = ['ngram.py', '1', 'arg2', 'arg3'] with pytest.raises(ngram.UsageError): ngram.parse_args(argv)
def test_correct_usage(self): argv = ['ngram.py', '1', 'arg2', 'arg3', 'arg4'] ret = ['1', 'arg2', 'arg3', 'arg4'] assert ngram.parse_args(argv) == ret
def test_empty_args(self): argv = [] with pytest.raises(ngram.UsageError): ngram.parse_args(argv)