• Pytest学习笔记3——参数化


      前言

      在讲pytest与unittest的区别文章中,我们知道其中一个区别就是参数化,unittest框架使用的第三方库ddt来参数化的,而pytest框架就直接使用装饰器@pytest.mark.parametrize来对测试用例进行传参。这个是针对测试方法来参数化,还有一种是针对前置处理函数来传参。但往往这两种是可以结合使用。

      Params参数化(单个)

      前面讲fixture函数时,它有个参数params用来传递参数,并且与request结合使用,先看单个:

    import pytest
    
    seq = [1, 2, 3]
    
    
    @pytest.fixture(params=seq)
    def test_data(request):
        print("参数")
        return request.param
    
    
    class TestData:
        def test_1(self, test_data):
            print("用例", test_data)
    
    
    if __name__ == '__main__':
        pytest.main()
    

      

      运行结果:

    fixture_test03.py::TestData::test_1[1] 参数
    PASSED                            [ 33%]用例 1
    
    fixture_test03.py::TestData::test_1[2] 参数
    PASSED                            [ 66%]用例 2
    
    fixture_test03.py::TestData::test_1[3] 参数
    PASSED                            [100%]用例 3
    
    
    ============================== 3 passed in 0.05s ==============================
    

      

      Params参数化(多个)

      列表里有多个dict数据,request会自动化循环读取每个索引下的值,从0开始,看下面例子:

    import pytest
    
    seq = [{'a':1,'b':2},{'c':3,'d':4}]
    
    
    @pytest.fixture(params=seq)
    def test_data(request):
        print("参数:%s"%request.param)
        return request.param
    
    
    class TestData:
        def test_1(self, test_data):
            print("用例", test_data)
    
    
    
    
    if __name__ == '__main__':
        pytest.main()
    

      

      运行结果:

    ============================== 2 passed in 0.04s ==============================
    
    Process finished with exit code 0
    参数:{'a': 1, 'b': 2}
    PASSED                   [ 50%]用例 {'a': 1, 'b': 2}
    参数:{'c': 3, 'd': 4}
    PASSED                   [100%]用例 {'c': 3, 'd': 4}
    

      

      注意:params需要与request一起使用。

      装饰器@pytest.mark.parametrize参数化(单个)

      我们做接口测试时,编写测试用例需要输入一组数据,然后执行程序,得到输出数据,经过断言将实际结果与预期结果进行比较,从而得到这条用例执行的结果。但是输入的数据肯定不是一两个,这个时候需要参数化,比如登录的接口,我们可以用测试账号1登录,也可用账号2,账号3...进行登录,如果每一个参数写一条测试用例肯定是不行的,所以引入@pytest.mark.parametrize()装饰器进行参数化,这就是背景。

      格式:@pytest.mark.parametrize("参数名称","参数列表"),第一个是参数的名称,第二个是参数列表(list),也就是参数的值。

      如:@pytest.mark.parametrize("x",[1, 2, 3])

      如:@pytest.mark.parametrize('请求方式,接口地址,传参,预期结果',[('get','www.baidu.com','{"page":1}','{"code":0,"msg":"成功"})',('post','www.baidu.com','{"page":2}','{"code":0,"msg":"成功"}')])

      1、第一个参数是字符串,多个参数中间用逗号隔开

      2、第二个参数是list,多组数据用元祖类型;传三个或更多参数也是这样传。list的每个元素都是一个元组,元组里的每个元素和按参数顺序一一对应

      3、传一个参数 @pytest.mark.parametrize('参数名',list) 进行参数化

      4、传两个参数@pytest.mark.parametrize('参数名1,参数名2',[(参数1_data[0], 参数2_data[0]),(参数1_data[1], 参数2_data[1])]) 进行参数化

    seq = [1, 2, 3]
    
    def get_data(x):
        print(x)
        pass
    
    
    @pytest.mark.parametrize("x",seq)
    def test_data(x):
        a = get_data(x)
        print(a)
    
    if __name__ == '__main__':
        pytest.main()
    

      

      运行结果:

    ============================== 3 passed in 0.04s ==============================
    
    Process finished with exit code 0
    PASSED                                   [ 33%]1
    None
    PASSED                                   [ 66%]2
    None
    PASSED                                   [100%]3
    None
    

      

      备注:第一个函数输出的是参数值,类中的函数打印的是外面那个函数的返回值,因为没有return,默认返回是None。

      与request结合使用

      如果装饰器@pytest.mark.parametrize与request结合使用,如果测试方法写在类中,则@pytest.mark.parametrize的参数名称要与@pytest.fixture函数名称保持一致。并且要添加indirect=True参数,目的是把ss_data当做函数去执行,而不是参数。

      如:

      

    import pytest
    seq = [1,2,3]
    @pytest.fixture()
    def ss_data(request):
        print("参数:%s"%request.param)
        return request.param
    
    
    
    class TestData:
        @pytest.mark.parametrize("ss_data",seq,indirect=True)
        def test_1(self,ss_data):
            print("用例", ss_data)
    
    if __name__ == '__main__':
        pytest.main()
    

      

      

      运行结果:

      

    collecting ... collected 3 items
    
    fixture_test03.py::TestData::test_1[1] 
    fixture_test03.py::TestData::test_1[2] 
    fixture_test03.py::TestData::test_1[3] 
    
    ============================== 3 passed in 0.04s ==============================
    
    Process finished with exit code 0
    参数:1
    PASSED                            [ 33%]用例 1
    参数:2
    PASSED                            [ 66%]用例 2
    参数:3
    PASSED                            [100%]用例 3
    

      

      注意1:如果将@pytest.mark.parametrize参数名称改成x的执行,就会报错如:

      注意2:如果不使用fixture装饰器函数,也不使用外部函数,也是可以的,例子如下:

    import pytest
    seq = [1,2,3]
    
    class TestData:
        @pytest.mark.parametrize("x",seq,indirect=False)
        def test_1(self,x):
            print("用例", x)
    
    if __name__ == '__main__':
        pytest.main()
    

      

      运行结果:

    ============================== 3 passed in 0.04s ==============================
    
    Process finished with exit code 0
    PASSED                            [ 33%]用例 1
    PASSED                            [ 66%]用例 2
    PASSED                            [100%]用例 3
    

      可以看出,装饰器@pytest.mark.parametrize参数传递的是参数名称了,而不是函数了。根据开关indirect自由控制,默认是False.

      装饰器@pytest.mark.parametrize参数化(多个)

       多个参数和单个参数写法差不多,只不过多个参数中间用逗号隔开,列表中用元组来分组,举个例子(不带类的测试方法和没有使用request参数):

    import pytest
    seq = [(1,2),(3,4)]
    
    def get_data(x,y):
        print(x)
        print(y)
        pass
    
    
    @pytest.mark.parametrize("x,y",seq)
    def test_data(x,y):
        get_data(x,y)
        # print(a)
    
    if __name__ == '__main__':
        pytest.main()
    

      

      运行结果:

    ============================== 2 passed in 0.04s ==============================
    
    Process finished with exit code 0
    PASSED                                 [ 50%]1
    2
    PASSED                                 [100%]3
    4
    

      这就是把参数列表list里面的元组中两个值,看作一对参数(x,y)。

      与request结合使用

      前面介绍过单个使用的场景,现在是多个参数的时候,如何使用呢?

      其实差不多,但是需要注意一些问题,先看个例子:

    import pytest
    seq = [(1,2),(3,4)]
    
    @pytest.fixture()
    def get_data(request):
        x = request.param[0]
        y = request.param[1]
        print(x,111)
        print(y,222)
        # print(request.param)
        pass
    
    # indirect=True声明x是个函数
    @pytest.mark.parametrize("get_data",seq)
    def test_data(get_data):
        print(get_data)
    
    if __name__ == '__main__':
        pytest.main()
    

      

      运行结果:

    plugins: allure-pytest-2.8.6, rerunfailures-5.0
    collecting ... collected 2 items
    
    fixture_005.py::test_data[get_data0] PASSED                              [ 50%](1, 2)
    
    fixture_005.py::test_data[get_data1] PASSED                              [100%](3, 4)
    
    
    ============================== 2 passed in 0.02s ==============================
    
    Process finished with exit code 0
    

      我们可以看出来fixture装饰器函数根本没有执行,这是为什么呢?

      因为使用@pytest.mark.parametrize装饰器参数化时,默认是以参数而不是函数,这里的@pytest.mark.parametrize("get_data",seq)中get_data是参数名称,不是函数,如果要声明为函数,需要添加:indirect=True。

      我们将其改之,再运行:

    import pytest
    seq = [(1,2),(3,4)]
    
    @pytest.fixture()
    def get_data(request):
        x = request.param[0]
        y = request.param[1]
        print(x,111)
        print(y,222)
        # print(request.param)
        pass
    
    # indirect=True声明x是个函数
    @pytest.mark.parametrize("get_data",seq,indirect=True)
    def test_data(get_data):
        print(get_data)
    
    if __name__ == '__main__':
        pytest.main()
    

      

      运行结果:

    fixture_005.py::test_data[get_data0] 1 111
    2 222
    PASSED                              [ 50%]None
    
    fixture_005.py::test_data[get_data1] 3 111
    4 222
    PASSED                              [100%]None
    
    
    ============================== 2 passed in 0.02s ==============================
    
    Process finished with exit code 0
    

      

      多个fixture与@pytest.mark.parametrize组合

      用例前面可以有多个条件,参数化装饰器也可以叠加,使用parametrize装饰器叠加时,用例组合是2个参数个数相乘

      

    import pytest
    
    seq1 = [1,2,3]
    seq2 = [4,5,6]
    
    @pytest.fixture()
    def get_seq1(request):
        seq1 = request.param
        print("seq1:",seq1)
        return seq1
    
    @pytest.fixture()
    def get_seq2(request):
        seq2 = request.param
        print("seq2:", seq2)
        return seq2
    
    @pytest.mark.parametrize("get_seq1",seq1,indirect=True)
    @pytest.mark.parametrize("get_seq2",seq2,indirect=True)
    def test_1(get_seq1,get_seq2):
        print(get_seq1,11)
        print(get_seq2,22)
    
    if __name__ == '__main__':
        pytest.main()
    

      

      运行结果:

    collecting ... collected 9 items
    
    fixture_test04.py::test_1[4-1] 
    fixture_test04.py::test_1[4-2] 
    fixture_test04.py::test_1[4-3] 
    fixture_test04.py::test_1[5-1] 
    fixture_test04.py::test_1[5-2] 
    fixture_test04.py::test_1[5-3] 
    fixture_test04.py::test_1[6-1] 
    fixture_test04.py::test_1[6-2] 
    fixture_test04.py::test_1[6-3] 
    
    ============================== 9 passed in 0.07s ==============================
    
    Process finished with exit code 0
    seq1: 1
    seq2: 4
    PASSED                                    [ 11%]1 11
    4 22
    seq1: 2
    seq2: 4
    PASSED                                    [ 22%]2 11
    4 22
    seq1: 3
    seq2: 4
    PASSED                                    [ 33%]3 11
    4 22
    seq1: 1
    seq2: 5
    PASSED                                    [ 44%]1 11
    5 22
    seq1: 2
    seq2: 5
    PASSED                                    [ 55%]2 11
    5 22
    seq1: 3
    seq2: 5
    PASSED                                    [ 66%]3 11
    5 22
    seq1: 1
    seq2: 6
    PASSED                                    [ 77%]1 11
    6 22
    seq1: 2
    seq2: 6
    PASSED                                    [ 88%]2 11
    6 22
    seq1: 3
    seq2: 6
    PASSED                                    [100%]3 11
    6 22
    

      

      如果使用类,也是一样的,如图:

       seq1和seq2参数都是3个,所以总的用例数就是9个。

      单独使用@pytest.mark.parametrize参数组合

      总结

      pytest与unittest的区别之一参数化已经讲完,希望可以帮助你学习pytest框架。

      如果对你有帮助或喜欢自动化测试开发的朋友,可以加入右下方QQ交流群学习与探索,更多干货与你分享。

  • 相关阅读:
    [SHOI2014]信号增幅仪
    [SDOI2016]征途
    Luogu P3226 [HNOI2012]集合选数
    Comet OJ C1076 [Contest #4]求和
    Luogu P2657 [SCOI2009]windy数
    Luogu P1864 [NOI2009]二叉查找树
    UVA10559 Blocks
    Luogu P1880 [NOI1995]石子合并
    简单DP
    CF1097F Alex and a TV Show
  • 原文地址:https://www.cnblogs.com/liudinglong/p/12901716.html
Copyright © 2020-2023  润新知