def parse_args(args):
parser = argparse.ArgumentParser(...)
parser.add_argument...
# ...Create your parser as you like...
return parser.parse_args(args)
parser = parse_args(sys.argv[1:])
(其中sys.argv
代表脚本名称的第一个元素被删除,以使其在CLI操作期间不作为附加开关发送。)
def test_parser(self):
parser = parse_args(['-l', '-m'])
self.assertTrue(parser.long)
# ...and so on.
这样,您就不必只为了测试解析器而执行应用程序的代码。
如果稍后需要在应用程序中更改和/或向解析器添加选项,请创建一个工厂方法:
def create_parser():
parser = argparse.ArgumentParser(...)
parser.add_argument...
# ...Create your parser as you like...
return parser
以后,您可以根据需要对其进行操作,然后进行如下测试:
class ParserTest(unittest.TestCase):
def setUp(self):
self.parser = create_parser()
def test_something(self):
parsed = self.parser.parse_args(['--something', 'test'])
self.assertEqual(parsed.something, 'test')