1. 程式人生 > 其它 >python進階11-unittest引數化

python進階11-unittest引數化

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

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

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

pip install nose-parameterized

下面是程式碼

import
unittest from nose_parameterized import parameterized #匯入這個模組 class My(unittest.TestCase):
@parameterized.expand(
    [
        (</span>1,2,3<span style="color: rgba(0, 0, 0, 1)">),
        (</span>1,2,3<span style="color: rgba(0, 0, 0, 1)">),
        (</span>1,2,3<span style="color: rgba(0, 0, 0, 1)">),
        (</span>1,2,4<span style="color: rgba(0, 0, 0, 1)">)
     ]
)</span><span style="color: rgba(0, 128, 0, 1)">#</span><span style="color: rgba(0, 128, 0, 1)">使用它提供的裝飾器裝飾這個函式,把咱們寫的這4個數據放到這個list裡面</span>
<span style="color: rgba(0, 0, 255, 1)">def</span><span style="color: rgba(0, 0, 0, 1)"> test1(self,a,b,c):
    self.assertEqual(a</span>+<span style="color: rgba(0, 0, 0, 1)">b,c)

if name=='main':
unittest.main()

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