如何在python中使用unittest與多文件模塊

Question

我想創建測試用例來檢查許多文件中的相同函數。每個文件具有相同的函數名稱但算法不同。

我試圖創建一個循環來測試單元測試中的每個文件，但它沒有奏效。

Answer 1

__import__('name')將返回一個模塊。所以在這裏我們可以找到一個解決方案：

lib1.py，lib2.py，lib3.py：

def func(): 
    return 123 

test.py：

import unittest 

class TestFoo(unittest.TestCase): 
    def test_bar(self): 
     for name in ['lib1','lib2','lib3']: 
      result=__import__(name).func() 
      self.assertEqual(result, 123) 

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