python - 从被测模块运行单元测试

Question

我有一堆模块，对于每个模块，我都有一个基于 unittest 的测试。我想在每个模块中定义 main 来运行它的测试，但是由于导入循环导致导入错误(特别是当我在测试中使用 from mymodule import myclass 时)。我怀疑这是一个已解决的问题，所以 - 我应该在我的模块的 main 中放入什么来运行其相应的测试？

Answer 1

如果我没理解错的话，你有一个如下所示的文件(我们称之为 mymodule.py):

import unittest
from mymoduletests import MyModuleTests

class myclass(object):
    def somefunction(self, x):
        return x*x

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

和一个看起来像这样的单独文件(我们称之为 mymoduletests.py):

import unittest
from mymodule import myclass

class MyModuleTests(unittest.TestCase):
    def test_somefunction(self):
        m = myclass()
        self.assertEqual(4, m.somefunction(2))

如果您运行 mymodule.py，您会得到以下结果:

Traceback (most recent call last):
  File "mymodule.py", line 2, in <module>
    from mymoduletests import MyModuleTests
  File "/Users/srgerg/Desktop/p/mymoduletests.py", line 2, in <module>
    from mymodule import myclass
  File "/Users/srgerg/Desktop/p/mymodule.py", line 2, in <module>
    from mymoduletests import MyModuleTests
ImportError: cannot import name MyModuleTests

但是，如果您将 mymodule.py 更改为:

class myclass(object):
    def somefunction(self, x):
        return x*x

if __name__ == '__main__':
    import unittest
    from mymoduletests import MyModuleTests
    unittest.main()

然后运行它，你会得到:

.
----------------------------------------------------------------------
Ran 1 test in 0.000s

OK

我的理解正确吗？