summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--Lib/test/output/test_getopt7
-rw-r--r--Lib/test/test_getopt.py101
2 files changed, 108 insertions, 0 deletions
diff --git a/Lib/test/output/test_getopt b/Lib/test/output/test_getopt
new file mode 100644
index 0000000..b632f71
--- /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 0000000..1580a79
--- /dev/null
+++ b/Lib/test/test_getopt.py
@@ -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."