Python unittest單元測試框架實現參數化
當我們在使用TestNG時,發現它有一個非常好用的參數化功能。當你的測試用例有固定的參數和斷言結果時,它可以相似用例的節省用例的個數。
例子如下:
import static org.testng.Assert.assertEquals;import org.testng.annotations.DataProvider;import org.testng.annotations.Test;/** * Created by fnngj on 2017/3/19. */public class Demo { // 定義測試數據 @DataProvider(name = 'data') public Object[][] Users() { return new Object[][] {{ 1, 1, 2},{ 2, 2, 5},{ 3, 3, 6}, }; } @Test(dataProvider='data') public void testAdd(int a,int b,int c) { assertEquals(a + b, c); }}
相對而言,Python下面單元測試框架要弱上少,尤其是Python自帶的unittest測試框架,不支持參數化,不支持多線程執行用例,不支持HTML測試報告的生成...。好再,部分不足我們可以通過unittest擴展來滿足需求。比如現在要介紹一個參數化的擴展。
在沒有參數化功能的情況下,我們的用例需要這樣編寫。
import unittestclass TestAdd(unittest.TestCase): def test_add_01(self): self.assertEqual(1 + 2, 3) def test_add_02(self): self.assertEqual(2 + 2, 5) def test_add_03(self): self.assertEqual(3 + 3, 6)if __name__ == ’__main__’: unittest.main()
nose-parameterized是一個針對Python單元測試框架實現參數化的擴展。同時支持不同的單元測試框架。
GitHub地址:https://github.com/wolever/nose-parameterized
然后,unittest就可以像TestNG一樣寫用例了。
import unittestfrom nose_parameterized import parameterizedclass TestAdd(unittest.TestCase): @parameterized.expand([ ('01',1, 1, 2), ('02',2, 2, 5), ('03',3, 3, 6), ]) def test_add(self, name, a, b, c): self.assertEqual(a + b, c)if __name__ == ’__main__’: unittest.main(verbosity=2)
執行結果:
test_add_0_01 (__main__.TestAdd) ... ok
test_add_1_02 (__main__.TestAdd) ... FAIL
test_add_2_03 (__main__.TestAdd) ... ok
當相同入參和斷言結果的用例越多,這種寫法用起來越爽!
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支持好吧啦網。
相關文章: