Python 与代码一起定义单元测试类

Question

我正在进行快速原型(prototype)设计，因此现在我想将单元测试与我正在测试的代码一起编写 - 而不是将测试放在单独的文件中。但是，我只想构建测试并在我们将包含文件中的代码作为 main 运行时调用它们。如下:

class MyClass:
    def __init(self)__:
        # do init stuff here

    def myclassFunction():
        # do something


if __name__ == '__main__':
    import unittest

    class TestMyClass(unittest.TestCase):
        def test_one(self):
            # test myclassFunction()

    suite = unittest.TestLoader().loadTestsFromTestCase(TestMyClass)
    unittest.TextTestRunner(verbosity=2).run(suite)

这当然不会运行，因为unittest无法找到或使用测试类TestMyClass。有没有办法让这种安排发挥作用，或者我是否必须将除了对单元测试的调用之外的所有内容都从 if __name__ == '__main__' 范围中取出？

谢谢!

Answer 1

如果将 TestMyClass 移至 if __name__ == '__main__' 上方你会得到你想要的:仅当文件作为 main 执行时才会运行测试

类似这样的事情

import unittest
class MyClass:
  def __init(self)__:
    # do init stuff here

  def myclassFunction():
    # do something


class TestMyClass(unittest.TestCase):
    def test_one(self):

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