mirror of https://github.com/python/cpython.git
David Goodger's new getopt test module (thanks, David!).
https://sourceforge.net/patch/?func=detailpatch&patch_id=101110&group_id=5470 Accepted as-is, except for purging an "import *".
This commit is contained in:
parent
8bad612881
commit
571bb8fc72
|
@ -0,0 +1,7 @@
|
|||
test_getopt
|
||||
Running tests on getopt.short_has_arg
|
||||
Running tests on getopt.long_has_args
|
||||
Running tests on getopt.do_shorts
|
||||
Running tests on getopt.do_longs
|
||||
Running tests on getopt.getopt
|
||||
Module getopt: tests completed successfully.
|
|
@ -0,0 +1,101 @@
|
|||
# test_getopt.py
|
||||
# David Goodger <dgoodger@bigfoot.com> 2000-08-19
|
||||
|
||||
import getopt
|
||||
from getopt import GetoptError
|
||||
from test.test_support import verbose
|
||||
|
||||
def expectException(teststr, expected, failure=AssertionError):
|
||||
"""Executes a statement passed in teststr, and raises an exception
|
||||
(failure) if the expected exception is *not* raised."""
|
||||
try:
|
||||
exec teststr
|
||||
except expected:
|
||||
pass
|
||||
else:
|
||||
raise failure
|
||||
|
||||
if verbose:
|
||||
print 'Running tests on getopt.short_has_arg'
|
||||
assert getopt.short_has_arg('a', 'a:')
|
||||
assert not getopt.short_has_arg('a', 'a')
|
||||
expectException("tmp = getopt.short_has_arg('a', 'b')", GetoptError)
|
||||
expectException("tmp = getopt.short_has_arg('a', '')", GetoptError)
|
||||
|
||||
if verbose:
|
||||
print 'Running tests on getopt.long_has_args'
|
||||
has_arg, option = getopt.long_has_args('abc', ['abc='])
|
||||
assert has_arg
|
||||
assert option == 'abc'
|
||||
has_arg, option = getopt.long_has_args('abc', ['abc'])
|
||||
assert not has_arg
|
||||
assert option == 'abc'
|
||||
has_arg, option = getopt.long_has_args('abc', ['abcd'])
|
||||
assert not has_arg
|
||||
assert option == 'abcd'
|
||||
expectException("has_arg, option = getopt.long_has_args('abc', ['def'])",
|
||||
GetoptError)
|
||||
expectException("has_arg, option = getopt.long_has_args('abc', [])",
|
||||
GetoptError)
|
||||
expectException("has_arg, option = " + \
|
||||
"getopt.long_has_args('abc', ['abcd','abcde'])",
|
||||
GetoptError)
|
||||
|
||||
if verbose:
|
||||
print 'Running tests on getopt.do_shorts'
|
||||
opts, args = getopt.do_shorts([], 'a', 'a', [])
|
||||
assert opts == [('-a', '')]
|
||||
assert args == []
|
||||
opts, args = getopt.do_shorts([], 'a1', 'a:', [])
|
||||
assert opts == [('-a', '1')]
|
||||
assert args == []
|
||||
#opts, args = getopt.do_shorts([], 'a=1', 'a:', [])
|
||||
#assert opts == [('-a', '1')]
|
||||
#assert args == []
|
||||
opts, args = getopt.do_shorts([], 'a', 'a:', ['1'])
|
||||
assert opts == [('-a', '1')]
|
||||
assert args == []
|
||||
opts, args = getopt.do_shorts([], 'a', 'a:', ['1', '2'])
|
||||
assert opts == [('-a', '1')]
|
||||
assert args == ['2']
|
||||
expectException("opts, args = getopt.do_shorts([], 'a1', 'a', [])",
|
||||
GetoptError)
|
||||
expectException("opts, args = getopt.do_shorts([], 'a', 'a:', [])",
|
||||
GetoptError)
|
||||
|
||||
if verbose:
|
||||
print 'Running tests on getopt.do_longs'
|
||||
opts, args = getopt.do_longs([], 'abc', ['abc'], [])
|
||||
assert opts == [('--abc', '')]
|
||||
assert args == []
|
||||
opts, args = getopt.do_longs([], 'abc=1', ['abc='], [])
|
||||
assert opts == [('--abc', '1')]
|
||||
assert args == []
|
||||
opts, args = getopt.do_longs([], 'abc=1', ['abcd='], [])
|
||||
assert opts == [('--abcd', '1')]
|
||||
assert args == []
|
||||
expectException("opts, args = getopt.do_longs([], 'abc=1', ['abc'], [])",
|
||||
GetoptError)
|
||||
expectException("opts, args = getopt.do_longs([], 'abc', ['abc='], [])",
|
||||
GetoptError)
|
||||
|
||||
# note: the empty string between '-a' and '--beta' is significant:
|
||||
# it simulates an empty string option argument ('-a ""') on the command line.
|
||||
cmdline = ['-a', '1', '-b', '--alpha=2', '--beta', '-a', '3', '-a', '',
|
||||
'--beta', 'arg1', 'arg2']
|
||||
|
||||
if verbose:
|
||||
print 'Running tests on getopt.getopt'
|
||||
opts, args = getopt.getopt(cmdline, 'a:b', ['alpha=', 'beta'])
|
||||
assert opts == [('-a', '1'), ('-b', ''), ('--alpha', '2'), ('--beta', ''),
|
||||
('-a', '3'), ('-a', ''), ('--beta', '')]
|
||||
# Note ambiguity of ('-b', '') and ('-a', '') above. This must be
|
||||
# accounted for in the code that calls getopt().
|
||||
assert args == ['arg1', 'arg2']
|
||||
|
||||
expectException(
|
||||
"opts, args = getopt.getopt(cmdline, 'a:b', ['alpha', 'beta'])",
|
||||
GetoptError)
|
||||
|
||||
if verbose:
|
||||
print "Module getopt: tests completed successfully."
|
Loading…
Reference in New Issue