子类化 python unittest.Testcase，调用相同的 main

Question

我想创建一个名为BasicTest 的python 的unittest.Testcase 子类。我希望 BasicTest 的每个子类在 main 中运行相同的例程。我怎样才能做到这一点？

例子：

in basic_test.py:

class BasicTest(unittest.TestCase):

    ...


if __name__ == '__main__':
    # Do optparse stuff
    unittest.main()



in some_basic_test.py:

class SomeBasicTest(BasicTest):
    ...

if __name__ == '__main__':
    #call the main in basic_test.py

Answer 1

# basic_test.py
class BasicTest(unittest.TestCase):

  @staticmethod
  def main():
     # Do optparse stuff
     unittest.main()

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



# some_basic_test.py
class SomeBasicTest(BasicTest):
   ...

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

Answer 2

您无法（重新）导入模块作为新的 ma​​in，因此 if __name__=="__main__" 代码是无法访问的。

多尔的建议或类似的建议似乎是最合理的。
但是，如果您无法访问相关模块，您可以考虑查看 runpy .run_module() 将模块作为 main 执行。

Answer 3

我希望 BasicTest 的每个子类在 main 中运行相同的例程

我想您想要的是在从任何测试用例运行测试之前运行一些设置/初始化代码。在这种情况下，您可能对 setUpClass 类方法。

testA.py

import unittest


class BasicTest(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        print 'Preparing to run tests'


class TestA(BasicTest):

    def test1(self):
        print 'testA: test1'

    def test2(self):
        print 'testA: test2'


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

testB.py

import unittest

from testA import BasicTest


class TestB(BasicTest):

    def test1(self):
        print 'testB: test1'

    def test2(self):
        print 'testB: test2'


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

testA.py 的输出：

Preparing to run tests
testA: test1
testA: test2
..
----------------------------------------------------------------------
Ran 2 tests in 0.000s

OK

testB.py 的输出：

Preparing to run tests
testB: test1
testB: test2
..
----------------------------------------------------------------------
Ran 2 tests in 0.000s

OK