使用unittest.TestCase Subclass Python的方法调用测试方法

时间:2016-02-10 17:09:13

标签: python testing python-unittest

我一直在为unittest模块阅读the documentation,但是如果我能做出我所做的事情,我就无法破译。我希望这样做。具体来说,我希望能够做类似下面的课程:

class TestMyCode(unittest.TestCase):

    def test_code(self):
        self.assertTrue(True)

    # Not sure how to do implement this method below
    def run_tests(self):
        # This would run all of the tests in my class
        self.run()

我想这样做的原因是因为我想从另一个Python文件中运行我的测试。例如,我说我在另一个文件中编写代码:

from test_my_code import TestMyCode
tests = TestMyCode()
# Runs all of the tests from the TestMyCode subclass
tests.run_tests()

这可能吗?

由于

1 个答案:

答案 0 :(得分:1)

您尝试执行的操作的替代方法是运行下面显示的main模块。如果您希望在其中运行测试,请确保test模块可用于导入。希望这能帮到你!

模块: test

import unittest


class TestMyCode(unittest.TestCase):

    def test_code(self):
        self.assertTrue(True)

模块: main

import test


def run_tests(module):
    module.unittest.main(module.__name__)


if __name__ == '__main__':
    run_tests(test)

附录:要执行您要求的操作,请添加一个方法,复制unittest.main完成的一些内部工作以运行测试。以下两个模块有助于演示如何修改TestMyCode类,以便它可以在没有外界帮助的情况下自行运行测试。

模块: test_module

import unittest


class TestMyCode(unittest.TestCase):

    def test_code(self):
        self.assertTrue(True)

    @classmethod
    def run_tests(cls):
        test = unittest.defaultTestLoader.loadTestsFromTestCase(cls)
        unittest.TextTestRunner().run(test)

模块: main

import test_module


def main():
    test_module.TestMyCode.run_tests()


if __name__ == '__main__':
    main()