我有一个使用 argparse 库的 Python 模块。如何为代码库的该部分编写测试?
您应该重构代码并将解析移至函数:
def parse_args(args): parser = argparse.ArgumentParser(...) parser.add_argument... # ...Create your parser as you like... return parser.parse_args(args)
然后在您的main函数中,您应该使用以下命令调用它:
main
parser = parse_args(sys.argv[1:])
(其中sys.argv表示脚本名称的第一个元素已被删除,以便在 CLI 操作期间不将其作为附加开关发送。)
sys.argv
在您的测试中,您可以使用任何您想要测试的参数列表调用解析器函数:
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')