从 python 调用任务运行 unittest.main()

Question

我正在尝试通过Python Invoke库运行一些单元测试，但是我对Python的了解不足使我无法这样做。

这是我拥有的示例代码：

my_tests.py

import unittest
class TestStringMethods(unittest.TestCase):
def test_upper(self):
self.assertEqual('foo'.upper(), 'FOO')
def test_isupper(self):
self.assertTrue('FOO'.isupper())
self.assertFalse('Foo'.isupper())
def test_split(self):
s = 'hello world'
self.assertEqual(s.split(), ['hello', 'world'])
# check that s.split fails when the separator is not a string
with self.assertRaises(TypeError):
s.split(2)
def main():
unittest.main()
if __name__ == '__main__':
main()

tasks.py

from invoke import task
@task
def tests(ctx):
main()
@task
def other_task(ctx):
print("This is fine")
def main():
import my_tests
import unittest
unittest.main(module='my_tests')
if __name__ == '__main__':
main()

这就是我得到的：

C:ittle_projectsinvoke_unittest>python my_tests.py
...
----------------------------------------------------------------------
Ran 3 tests in 0.002s
OK
C:ittle_projectsinvoke_unittest>python tasks.py
...
----------------------------------------------------------------------
Ran 3 tests in 0.001s
OK
C:ittle_projectsinvoke_unittest>inv tests
E
======================================================================
ERROR: tests (unittest.loader._FailedTest)
----------------------------------------------------------------------
AttributeError: module 'my_tests' has no attribute 'tests'
----------------------------------------------------------------------
Ran 1 test in 0.001s
FAILED (errors=1)

测试从my_tests.py和 tasks.py 运行良好，但是当我使用调用内容时，会中断。 我怎样才能让它工作，或者我下一步应该去哪里看？

Answer 1

您遇到的问题是unittest.main()使用调用程序的命令行参数来确定要运行的测试。由于你的程序是按照inv tests执行的，程序的第一个参数是tests，所以unittest试图运行一个不存在的模块名称tests的测试。

您可以通过从系统参数列表中弹出最后一个参数 (tests( 来解决此问题：

import sys
from invoke import task
@task
def tests(ctx):
# Pop "tests" off the end of the system arguments
sys.argv.pop()
main()
@task
def other_task(ctx):
print("This is fine")
def main():
import my_tests
import unittest
unittest.main(module='my_tests')
if __name__ == '__main__':
main()