python - sys.argv 的 IndexError 和测试命令行参数

Question

我正在按照这些思路编写各种单元测试测试用例:

import unittest
import program

class test(unittest.TestCase):
    a = 'acgt'
    b = 'c'
    sys.argv[1] = a
    sys.argv[2] = b
    text, patterns = program.parse()
    assertEqual(text, a)
    assertEqual(patterns, [b])

if __name__ == '__main__':
    unittest.main()

不幸的是，它们都返回索引超出范围的IndexError。这是有道理的，因为命令行上没有传递任何内容。如何从 sys.argv 中填充 sys.argv 参数，以便测试通过解析器获取它们的方法？

Answer 1

您只需更改 argv 变量即可

class MyTest(unittest.TestCase):
    def test(self):
        import sys
        sys.argv = ['0', '1', '2']
        program()


def program():
    import sys
    print(sys.argv[1])
    print(sys.argv[2])

if __name__ == '__main__':
    unittest.main()

但这不是推荐的方式。而是更改程序的签名，使其接受一个列表，以便您可以轻松控制它的调用方式。

program(sys.argv)  # From command line
program(['param1', 'param2'])  # From Tests