pytest框架(1)

安装

pip install -U pytest

测试demo

# content of test_sample.py
def func(x):
    return x + 1


def test_answer():
    assert func(3) == 5

执行结果

$ pytest
=========================== test session starts ============================
platform linux -- Python 3.x.y, pytest-6.x.y, py-1.x.y, pluggy-0.x.y
cachedir: $PYTHON_PREFIX/.pytest_cache
rootdir: $REGENDOC_TMPDIR
collected 1 item

test_sample.py F                                                     [100%]

================================= FAILURES =================================
_______________________________ test_answer ________________________________

    def test_answer():
>       assert func(3) == 5
E       assert 4 == 5
E        +  where 4 = func(3)

test_sample.py:6: AssertionError
========================= short test summary info ==========================
FAILED test_sample.py::test_answer - assert 4 == 5
============================ 1 failed in 0.12s =============================

Python测试发现的约定

• 如果未指定参数，则集合从开始 testpaths （如果配置）或当前目录。或者，命令行参数可以用于目录、文件名或节点ID的任意组合。递归到目录中，除非它们匹配 norecursedirs .
• 在这些目录中，搜索 test_*.py 或 *_test.py 文件，由其导入 test package name .
• 从这些文件中收集测试项：
  • test 在类之外加前缀的测试函数或方法
  • test 内置前缀测试函数或方法 Test 带前缀的测试类（不带 init 方法）

例如以下

tests/
|-- example
|   |-- test_example_01.py
|   |-- test_example_02.py
|   '-- test_example_03.py
|-- foobar
|   |-- test_foobar_01.py
|   |-- test_foobar_02.py
|   '-- test_foobar_03.py
'-- hello
    '-- world
        |-- test_world_01.py
        |-- test_world_02.py
        '-- test_world_03.py

执行结果

=========================== test session starts ============================
platform linux -- Python 3.x.y, pytest-5.x.y, py-1.x.y, pluggy-0.x.y
rootdir: $REGENDOC_TMPDIR, inifile:
collected 5 items

tests/example/test_example_01.py .                                   [ 20%]
tests/example/test_example_02.py .                                   [ 40%]
tests/example/test_example_03.py .                                   [ 60%]
tests/foobar/test_foobar_01.py .                                     [ 80%]
tests/foobar/test_foobar_02.py .                                     [100%]

========================= 5 passed in 0.02 seconds =========================

运行方式

1.Pycharm 界面运行，设置默认使用pytest运行

（1）在菜单栏点击【File】>【Settings】

（2）【Settings】>【Tools】>【Python Integrated Tools】>【Default test runner】下拉菜单中勾选【py.test】，点击【apply】应用设置，点击【ok】关闭窗口

（3）右键，如下图，pytest已成为PyCharm脚本的默认执行方式

2.命令行执行

pytest
pytest test_add.py
pytest test_add.py::test_func

pytest 常用的命令行参数

常用参数
pytest --collect-only 只收集用例
pytest -k “add ” 匹配所有名称中包含add的用例（‘add or div’ ‘TestClass’）
pytest -m mark标签名 标记
pytest - - junitxml=./result.xml 生成执行结果文件
pytest --setup-show 回溯fixture的执行过程
更多的用法使用pytest —help查看帮助文档

pytest 框架结构

● 类似的setup,teardown同样更灵活，
● 模块级(setup_module/teardown_module)模块始末，全局的（优先最高）
● 函数级(setup_function/teardown_function)只对函数用例生效(不在类中)
● 类级**(setup_class/teardown_class)**只在类中前后运行一次（在类中）
● 方法级(setup_method/teardown_methond)开始于方法始末（在类中）
● **类里面的（setup/teardown）**运行在调用方法的前后

参数化

@pytest.mark.parametrize进⾏参数化

import pytest
def add(a,b):
    return a+b
    
class TestAdd:
    def setup(self):
        print("开始计算".center(30,"*"))
        
    def teardown(self):
        print("结束计算".center(30,"*"))
        
    @pytest.mark.parametrize('a,b,expect',[[1,2,3],[2,3,5]])
    def test_add(self, a,b,expect):
        print("{}+{}={}".format(str(a),str(b),str(expect)))
        assert expect == add(a, b)

使用YAML参数化

安装

python3.X使用以下命令安装

pip install pyyaml

python2.X 使用以下命令安装

pip install yaml

语法

大小写敏感
使用缩进表示层级关系
缩进不允许使用tab，只允许空格
缩进的空格数不重要，只要相同层级的元素左对齐即可
'#'表示注释

数组
“-” 后面要有空格

- A
- B
- C

字典
"："冒号后面要有空格

child-key: value

混合

companies:
    -
        id: 1
        name: company1
        price: 200W
    -
        id: 2
        name: company2
        price: 500W

相当于

companies: [{id: 1,name: company1,price: 200W},{id: 2,name: company2,price: 500W}]

YAML读取

cases_datas.yaml

ids:
  - 正数和正数相加
  - 0和0相加
datas:
#  -
#    - a
#    - b
#    - expect
  -
    - 1
    - 2
    - 3
  -
    - 0
    - 0
    - 0

读取yaml文件内容

#do_yaml.py
# pip install pyyaml
import yaml
class Do_YAML:

    def read_yaml(self,file_path):
        with open(file_path,encoding='utf-8') as f:
            datas = yaml.safe_load(f)
        return datas


if __name__ == '__main__':
    t = Do_YAML()
    datas = t.read_yaml(file_path='./cases_datas.yaml')
    print(datas)

打印结果

{'ids': ['正数和正数相加', '0和0相加'], 'datas': [[1, 2, 3], [0, 0, 0]]}

pytest demo

Calculator.py

#Calculator.py
class Calculator:

    def add(self,a,b):
        return a+b

    def div(self,a,b):
        return a/b

    def multi(self,a,b):
        return a*b

    def sub(self,a,b):
        return a-b

test_add.py

#test_add.py
# pip install pytest
import pytest
# pip install pyyaml
from do_yaml import Do_YAML
from Calculator import Calculator

calc = Calculator()

do_yaml = Do_YAML()
case_datas = do_yaml.read_yaml(file_path="./add_cases.yaml")

class TestAdd:
    def setup(self):
        print("开始计算".center(30,"*"))

    def teardown(self):
        print("结束计算".center(30,"*"))


    @pytest.mark.parametrize('a,b,expect',case_datas["datas"], ids=case_datas["ids"])
    def test_add(self, a,b,expect):
        print("{}+{}={}".format(str(a),str(b),str(expect)))
        assert expect == calc.add(a, b)

参考资料：

英文文档
https://docs.pytest.org/en/stable/getting-started.html#create-your-first-test
中文文档：
https://www.osgeo.cn/pytest/contents.html
yaml文档
https://pyyaml.org/wiki/PyYAMLDocumentation