1. 程式人生 > >unittest引數化

unittest引數化

我們在寫case的時候,如果用例的操作是一樣的,就是引數不同,比如說要測一個登陸的介面,要測正常登陸的、黑名單使用者登陸的、賬號密碼錯誤的等等,在unittest裡面就要寫多個case來測試。

這樣的情況只是呼叫介面的時候引數不一樣而已,再寫多個case的話就有點多餘了,那怎麼辦呢,就得把這些引數都寫到一個list裡面, 然後迴圈去執行這個case。這樣就可以省去寫多個case了。

當然有個第三方模組就直接有這樣的功能,不需要咱們再自己寫迴圈了。那就是nose-parameterized,直接pip安裝即可。

 
1 pip install nose-parameterized

下面是程式碼

 
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 import unittest from nose_parameterized import parameterized #匯入這個模組 class My(unittest.TestCase):       @parameterized.expand(
        [             (1,2,3),             (1,2,3),             (1,2,3),             (1,2,4)          ]     )#使用它提供的裝飾器裝飾這個函式,把咱們寫的這4個數據放到這個list裡面     def test1(self,a,b,c):         self.assertEqual(a+b,c)   if __name__=='__main__':     unittest.main()

下面是執行結果,咱們上面只運行了一次,可以發現,它自動幫咱們運行了4次。