[mirror_edk2.git] / AppPkg / Applications / Python / Python-2.7.2 / Lib / test / test_getopt.py

# test_getopt.py\r
# David Goodger <dgoodger@bigfoot.com> 2000-08-19\r
\r
from test.test_support import verbose, run_doctest, run_unittest, EnvironmentVarGuard\r
import unittest\r
\r
import getopt\r
\r
sentinel = object()\r
\r
class GetoptTests(unittest.TestCase):\r
    def setUp(self):\r
        self.env = EnvironmentVarGuard()\r
        if "POSIXLY_CORRECT" in self.env:\r
            del self.env["POSIXLY_CORRECT"]\r
\r
    def tearDown(self):\r
        self.env.__exit__()\r
        del self.env\r
\r
    def assertError(self, *args, **kwargs):\r
        self.assertRaises(getopt.GetoptError, *args, **kwargs)\r
\r
    def test_short_has_arg(self):\r
        self.assertTrue(getopt.short_has_arg('a', 'a:'))\r
        self.assertFalse(getopt.short_has_arg('a', 'a'))\r
        self.assertError(getopt.short_has_arg, 'a', 'b')\r
\r
    def test_long_has_args(self):\r
        has_arg, option = getopt.long_has_args('abc', ['abc='])\r
        self.assertTrue(has_arg)\r
        self.assertEqual(option, 'abc')\r
\r
        has_arg, option = getopt.long_has_args('abc', ['abc'])\r
        self.assertFalse(has_arg)\r
        self.assertEqual(option, 'abc')\r
\r
        has_arg, option = getopt.long_has_args('abc', ['abcd'])\r
        self.assertFalse(has_arg)\r
        self.assertEqual(option, 'abcd')\r
\r
        self.assertError(getopt.long_has_args, 'abc', ['def'])\r
        self.assertError(getopt.long_has_args, 'abc', [])\r
        self.assertError(getopt.long_has_args, 'abc', ['abcd','abcde'])\r
\r
    def test_do_shorts(self):\r
        opts, args = getopt.do_shorts([], 'a', 'a', [])\r
        self.assertEqual(opts, [('-a', '')])\r
        self.assertEqual(args, [])\r
\r
        opts, args = getopt.do_shorts([], 'a1', 'a:', [])\r
        self.assertEqual(opts, [('-a', '1')])\r
        self.assertEqual(args, [])\r
\r
        #opts, args = getopt.do_shorts([], 'a=1', 'a:', [])\r
        #self.assertEqual(opts, [('-a', '1')])\r
        #self.assertEqual(args, [])\r
\r
        opts, args = getopt.do_shorts([], 'a', 'a:', ['1'])\r
        self.assertEqual(opts, [('-a', '1')])\r
        self.assertEqual(args, [])\r
\r
        opts, args = getopt.do_shorts([], 'a', 'a:', ['1', '2'])\r
        self.assertEqual(opts, [('-a', '1')])\r
        self.assertEqual(args, ['2'])\r
\r
        self.assertError(getopt.do_shorts, [], 'a1', 'a', [])\r
        self.assertError(getopt.do_shorts, [], 'a', 'a:', [])\r
\r
    def test_do_longs(self):\r
        opts, args = getopt.do_longs([], 'abc', ['abc'], [])\r
        self.assertEqual(opts, [('--abc', '')])\r
        self.assertEqual(args, [])\r
\r
        opts, args = getopt.do_longs([], 'abc=1', ['abc='], [])\r
        self.assertEqual(opts, [('--abc', '1')])\r
        self.assertEqual(args, [])\r
\r
        opts, args = getopt.do_longs([], 'abc=1', ['abcd='], [])\r
        self.assertEqual(opts, [('--abcd', '1')])\r
        self.assertEqual(args, [])\r
\r
        opts, args = getopt.do_longs([], 'abc', ['ab', 'abc', 'abcd'], [])\r
        self.assertEqual(opts, [('--abc', '')])\r
        self.assertEqual(args, [])\r
\r
        # Much like the preceding, except with a non-alpha character ("-") in\r
        # option name that precedes "="; failed in\r
        # http://python.org/sf/126863\r
        opts, args = getopt.do_longs([], 'foo=42', ['foo-bar', 'foo=',], [])\r
        self.assertEqual(opts, [('--foo', '42')])\r
        self.assertEqual(args, [])\r
\r
        self.assertError(getopt.do_longs, [], 'abc=1', ['abc'], [])\r
        self.assertError(getopt.do_longs, [], 'abc', ['abc='], [])\r
\r
    def test_getopt(self):\r
        # note: the empty string between '-a' and '--beta' is significant:\r
        # it simulates an empty string option argument ('-a ""') on the\r
        # command line.\r
        cmdline = ['-a', '1', '-b', '--alpha=2', '--beta', '-a', '3', '-a',\r
                   '', '--beta', 'arg1', 'arg2']\r
\r
        opts, args = getopt.getopt(cmdline, 'a:b', ['alpha=', 'beta'])\r
        self.assertEqual(opts, [('-a', '1'), ('-b', ''),\r
                                ('--alpha', '2'), ('--beta', ''),\r
                                ('-a', '3'), ('-a', ''), ('--beta', '')])\r
        # Note ambiguity of ('-b', '') and ('-a', '') above. This must be\r
        # accounted for in the code that calls getopt().\r
        self.assertEqual(args, ['arg1', 'arg2'])\r
\r
        self.assertError(getopt.getopt, cmdline, 'a:b', ['alpha', 'beta'])\r
\r
    def test_gnu_getopt(self):\r
        # Test handling of GNU style scanning mode.\r
        cmdline = ['-a', 'arg1', '-b', '1', '--alpha', '--beta=2']\r
\r
        # GNU style\r
        opts, args = getopt.gnu_getopt(cmdline, 'ab:', ['alpha', 'beta='])\r
        self.assertEqual(args, ['arg1'])\r
        self.assertEqual(opts, [('-a', ''), ('-b', '1'),\r
                                ('--alpha', ''), ('--beta', '2')])\r
\r
        # recognize "-" as an argument\r
        opts, args = getopt.gnu_getopt(['-a', '-', '-b', '-'], 'ab:', [])\r
        self.assertEqual(args, ['-'])\r
        self.assertEqual(opts, [('-a', ''), ('-b', '-')])\r
\r
        # Posix style via +\r
        opts, args = getopt.gnu_getopt(cmdline, '+ab:', ['alpha', 'beta='])\r
        self.assertEqual(opts, [('-a', '')])\r
        self.assertEqual(args, ['arg1', '-b', '1', '--alpha', '--beta=2'])\r
\r
        # Posix style via POSIXLY_CORRECT\r
        self.env["POSIXLY_CORRECT"] = "1"\r
        opts, args = getopt.gnu_getopt(cmdline, 'ab:', ['alpha', 'beta='])\r
        self.assertEqual(opts, [('-a', '')])\r
        self.assertEqual(args, ['arg1', '-b', '1', '--alpha', '--beta=2'])\r
\r
    def test_libref_examples(self):\r
        s = """\r
        Examples from the Library Reference:  Doc/lib/libgetopt.tex\r
\r
        An example using only Unix style options:\r
\r
\r
        >>> import getopt\r
        >>> args = '-a -b -cfoo -d bar a1 a2'.split()\r
        >>> args\r
        ['-a', '-b', '-cfoo', '-d', 'bar', 'a1', 'a2']\r
        >>> optlist, args = getopt.getopt(args, 'abc:d:')\r
        >>> optlist\r
        [('-a', ''), ('-b', ''), ('-c', 'foo'), ('-d', 'bar')]\r
        >>> args\r
        ['a1', 'a2']\r
\r
        Using long option names is equally easy:\r
\r
\r
        >>> s = '--condition=foo --testing --output-file abc.def -x a1 a2'\r
        >>> args = s.split()\r
        >>> args\r
        ['--condition=foo', '--testing', '--output-file', 'abc.def', '-x', 'a1', 'a2']\r
        >>> optlist, args = getopt.getopt(args, 'x', [\r
        ...     'condition=', 'output-file=', 'testing'])\r
        >>> optlist\r
        [('--condition', 'foo'), ('--testing', ''), ('--output-file', 'abc.def'), ('-x', '')]\r
        >>> args\r
        ['a1', 'a2']\r
        """\r
\r
        import types\r
        m = types.ModuleType("libreftest", s)\r
        run_doctest(m, verbose)\r
\r
    def test_issue4629(self):\r
        longopts, shortopts = getopt.getopt(['--help='], '', ['help='])\r
        self.assertEqual(longopts, [('--help', '')])\r
        longopts, shortopts = getopt.getopt(['--help=x'], '', ['help='])\r
        self.assertEqual(longopts, [('--help', 'x')])\r
        self.assertRaises(getopt.GetoptError, getopt.getopt, ['--help='], '', ['help'])\r
\r
def test_main():\r
    run_unittest(GetoptTests)\r
\r
if __name__ == "__main__":\r
    test_main()\r
Commit	Line	Data
4710c53d	1	# test_getopt.py\r
	2	# David Goodger <dgoodger@bigfoot.com> 2000-08-19\r
	3	\r
	4	from test.test_support import verbose, run_doctest, run_unittest, EnvironmentVarGuard\r
	5	import unittest\r
	6	\r
	7	import getopt\r
	8	\r
	9	sentinel = object()\r
	10	\r
	11	class GetoptTests(unittest.TestCase):\r
	12	def setUp(self):\r
	13	self.env = EnvironmentVarGuard()\r
	14	if "POSIXLY_CORRECT" in self.env:\r
	15	del self.env["POSIXLY_CORRECT"]\r
	16	\r
	17	def tearDown(self):\r
	18	self.env.__exit__()\r
	19	del self.env\r
	20	\r
	21	def assertError(self, args, *kwargs):\r
	22	self.assertRaises(getopt.GetoptError, args, *kwargs)\r
	23	\r
	24	def test_short_has_arg(self):\r
	25	self.assertTrue(getopt.short_has_arg('a', 'a:'))\r
	26	self.assertFalse(getopt.short_has_arg('a', 'a'))\r
	27	self.assertError(getopt.short_has_arg, 'a', 'b')\r
	28	\r
	29	def test_long_has_args(self):\r
	30	has_arg, option = getopt.long_has_args('abc', ['abc='])\r
	31	self.assertTrue(has_arg)\r
	32	self.assertEqual(option, 'abc')\r
	33	\r
	34	has_arg, option = getopt.long_has_args('abc', ['abc'])\r
	35	self.assertFalse(has_arg)\r
	36	self.assertEqual(option, 'abc')\r
	37	\r
	38	has_arg, option = getopt.long_has_args('abc', ['abcd'])\r
	39	self.assertFalse(has_arg)\r
	40	self.assertEqual(option, 'abcd')\r
	41	\r
	42	self.assertError(getopt.long_has_args, 'abc', ['def'])\r
	43	self.assertError(getopt.long_has_args, 'abc', [])\r
	44	self.assertError(getopt.long_has_args, 'abc', ['abcd','abcde'])\r
	45	\r
	46	def test_do_shorts(self):\r
	47	opts, args = getopt.do_shorts([], 'a', 'a', [])\r
	48	self.assertEqual(opts, [('-a', '')])\r
	49	self.assertEqual(args, [])\r
	50	\r
	51	opts, args = getopt.do_shorts([], 'a1', 'a:', [])\r
	52	self.assertEqual(opts, [('-a', '1')])\r
	53	self.assertEqual(args, [])\r
	54	\r
	55	#opts, args = getopt.do_shorts([], 'a=1', 'a:', [])\r
	56	#self.assertEqual(opts, [('-a', '1')])\r
	57	#self.assertEqual(args, [])\r
	58	\r
	59	opts, args = getopt.do_shorts([], 'a', 'a:', ['1'])\r
	60	self.assertEqual(opts, [('-a', '1')])\r
	61	self.assertEqual(args, [])\r
	62	\r
	63	opts, args = getopt.do_shorts([], 'a', 'a:', ['1', '2'])\r
	64	self.assertEqual(opts, [('-a', '1')])\r
65	self.assertEqual(args, ['2'])\r
66	\r
67	self.assertError(getopt.do_shorts, [], 'a1', 'a', [])\r
68	self.assertError(getopt.do_shorts, [], 'a', 'a:', [])\r
69	\r
70	def test_do_longs(self):\r
71	opts, args = getopt.do_longs([], 'abc', ['abc'], [])\r
72	self.assertEqual(opts, [('--abc', '')])\r
73	self.assertEqual(args, [])\r
74	\r
75	opts, args = getopt.do_longs([], 'abc=1', ['abc='], [])\r
76	self.assertEqual(opts, [('--abc', '1')])\r
77	self.assertEqual(args, [])\r
78	\r
79	opts, args = getopt.do_longs([], 'abc=1', ['abcd='], [])\r
80	self.assertEqual(opts, [('--abcd', '1')])\r
81	self.assertEqual(args, [])\r
82	\r
83	opts, args = getopt.do_longs([], 'abc', ['ab', 'abc', 'abcd'], [])\r
84	self.assertEqual(opts, [('--abc', '')])\r
85	self.assertEqual(args, [])\r
86	\r
87	# Much like the preceding, except with a non-alpha character ("-") in\r
88	# option name that precedes "="; failed in\r
89	# http://python.org/sf/126863\r
90	opts, args = getopt.do_longs([], 'foo=42', ['foo-bar', 'foo=',], [])\r
91	self.assertEqual(opts, [('--foo', '42')])\r
92	self.assertEqual(args, [])\r
93	\r
94	self.assertError(getopt.do_longs, [], 'abc=1', ['abc'], [])\r
95	self.assertError(getopt.do_longs, [], 'abc', ['abc='], [])\r
96	\r
97	def test_getopt(self):\r
98	# note: the empty string between '-a' and '--beta' is significant:\r
99	# it simulates an empty string option argument ('-a ""') on the\r
100	# command line.\r
101	cmdline = ['-a', '1', '-b', '--alpha=2', '--beta', '-a', '3', '-a',\r
102	'', '--beta', 'arg1', 'arg2']\r
103	\r
104	opts, args = getopt.getopt(cmdline, 'a:b', ['alpha=', 'beta'])\r
105	self.assertEqual(opts, [('-a', '1'), ('-b', ''),\r
106	('--alpha', '2'), ('--beta', ''),\r
107	('-a', '3'), ('-a', ''), ('--beta', '')])\r
108	# Note ambiguity of ('-b', '') and ('-a', '') above. This must be\r
109	# accounted for in the code that calls getopt().\r
110	self.assertEqual(args, ['arg1', 'arg2'])\r
111	\r
112	self.assertError(getopt.getopt, cmdline, 'a:b', ['alpha', 'beta'])\r
113	\r
114	def test_gnu_getopt(self):\r
115	# Test handling of GNU style scanning mode.\r
116	cmdline = ['-a', 'arg1', '-b', '1', '--alpha', '--beta=2']\r
117	\r
118	# GNU style\r
119	opts, args = getopt.gnu_getopt(cmdline, 'ab:', ['alpha', 'beta='])\r
120	self.assertEqual(args, ['arg1'])\r
121	self.assertEqual(opts, [('-a', ''), ('-b', '1'),\r
122	('--alpha', ''), ('--beta', '2')])\r
123	\r
124	# recognize "-" as an argument\r
125	opts, args = getopt.gnu_getopt(['-a', '-', '-b', '-'], 'ab:', [])\r
126	self.assertEqual(args, ['-'])\r
127	self.assertEqual(opts, [('-a', ''), ('-b', '-')])\r
128	\r
129	# Posix style via +\r
130	opts, args = getopt.gnu_getopt(cmdline, '+ab:', ['alpha', 'beta='])\r
131	self.assertEqual(opts, [('-a', '')])\r
132	self.assertEqual(args, ['arg1', '-b', '1', '--alpha', '--beta=2'])\r
133	\r
134	# Posix style via POSIXLY_CORRECT\r
135	self.env["POSIXLY_CORRECT"] = "1"\r
136	opts, args = getopt.gnu_getopt(cmdline, 'ab:', ['alpha', 'beta='])\r
137	self.assertEqual(opts, [('-a', '')])\r
138	self.assertEqual(args, ['arg1', '-b', '1', '--alpha', '--beta=2'])\r
139	\r
140	def test_libref_examples(self):\r
141	s = """\r
142	Examples from the Library Reference: Doc/lib/libgetopt.tex\r
143	\r
144	An example using only Unix style options:\r
145	\r
146	\r
147	>>> import getopt\r
148	>>> args = '-a -b -cfoo -d bar a1 a2'.split()\r
149	>>> args\r
150	['-a', '-b', '-cfoo', '-d', 'bar', 'a1', 'a2']\r
151	>>> optlist, args = getopt.getopt(args, 'abc:d:')\r
152	>>> optlist\r
153	[('-a', ''), ('-b', ''), ('-c', 'foo'), ('-d', 'bar')]\r
154	>>> args\r
155	['a1', 'a2']\r
156	\r
157	Using long option names is equally easy:\r
158	\r
159	\r
160	>>> s = '--condition=foo --testing --output-file abc.def -x a1 a2'\r
161	>>> args = s.split()\r
162	>>> args\r
163	['--condition=foo', '--testing', '--output-file', 'abc.def', '-x', 'a1', 'a2']\r
164	>>> optlist, args = getopt.getopt(args, 'x', [\r
165	... 'condition=', 'output-file=', 'testing'])\r
166	>>> optlist\r
167	[('--condition', 'foo'), ('--testing', ''), ('--output-file', 'abc.def'), ('-x', '')]\r
168	>>> args\r
169	['a1', 'a2']\r
170	"""\r
171	\r
172	import types\r
173	m = types.ModuleType("libreftest", s)\r
174	run_doctest(m, verbose)\r
175	\r
176	def test_issue4629(self):\r
177	longopts, shortopts = getopt.getopt(['--help='], '', ['help='])\r
178	self.assertEqual(longopts, [('--help', '')])\r
179	longopts, shortopts = getopt.getopt(['--help=x'], '', ['help='])\r
180	self.assertEqual(longopts, [('--help', 'x')])\r
181	self.assertRaises(getopt.GetoptError, getopt.getopt, ['--help='], '', ['help'])\r
182	\r
183	def test_main():\r
184	run_unittest(GetoptTests)\r
185	\r
186	if __name__ == "__main__":\r
187	test_main()\r