Python测试整个脚本

Question

我有一个Python脚本。我使用unittest进行测试，但是如何测试整个脚本呢？

我的想法是这样的：

def test_script(self):
    output=runScript('test.py --a 5 --b 3')
    self.assertEqual(output, '8') 

test.py接受参数a和b，并输出a+b，在本例中为8

Answer 1

您可以使用subprocess库来调用脚本并捕获输出。

import subprocess

p = subprocess.Popen(
    ['./test.py', '--a', ...],
    stdout=subprocess.PIPE,
    stderr=subprocess.STDOUT
)

print p.stdout.read()

Answer 2

不确定这是否是您想要的，请使用python unittest来结束黑盒测试。

import unittest # install and import

将测试包装在TestCase中

class ScriptTest(unittest.TestCase):
    def test_script(self):
        output=runScript('test.py --a 5 --b 3')
        self.assertEqual(output, '8')

将TestCase添加到unittest

if __name__ == '__main__':
    suite = unittest.TestLoader().loadTestsFromTestCase(ScriptTest)
    unittest.TextTestRunner(verbosity=2).run(suite)