diff --git a/Lib/test/output/test_getopt b/Lib/test/output/test_getopt new file mode 100644 index 00000000000..b632f716fe0 --- /dev/null +++ b/Lib/test/output/test_getopt @@ -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. diff --git a/Lib/test/test_getopt.py b/Lib/test/test_getopt.py new file mode 100644 index 00000000000..1580a79f252 --- /dev/null +++ b/Lib/test/test_getopt.py @@ -0,0 +1,101 @@ +# test_getopt.py +# David Goodger 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."