Selenium实战（八）——pytest单元测试（2）

一、断言

　　不像在unittest中的断言方法：assertEqual()、assertIn()、assertTrue()、assertIs()等。pytest单元测试框架并没有提供专门的断言方法，而是直接使用Python的assert进行断言。

　　创建test_assert.py文件

# 功能：用于计算a与b相加的和
def add(a, b):
    return a + b


# 功能：判断素数
def is_prime(n):
    if n <= 1:
        return False
    for i in range(2, n):
        if n % i == 0:
            return False
    return True


# 测试相等
def test_add_1():
    assert add(3, 10) == 13


# 测试不相等
def test_add_2():
    assert add(4, 8) != 98


# 测试大于或等于
def test_add_3():
    assert add(9, 3) >= 11


# 小于或等于
def test_add_4():
    assert add(4, 7) <= 17


# 测试包含
def test_in():
    a = "hello"
    b = "ell"
    assert b in a


# 测试不包含
def test_not_in():
    a = "print"
    b = "OOO"
    assert b not in a


# 判断是否为True
def test_true_1():
     assert is_prime(13)


# 判断是否为True
def test_true_2():
    assert is_prime(7) is True


# 判断是否不为True
def test_true_3():
    assert not is_prime(4)


# 判断是否不为True
def test_true_4():
    assert is_prime(6) is not True


# 判断是否为False
def test_false_1():
    assert is_prime(8) is False

 二、Fixture

　　Fixture通常用来对测试方法、测试函数、测试类和整个测试文件进行初始化或还原测试环境。创建test_fixtures_01.py文件：

# 功能函数
def multiply(a, b):
    return a * b


# ============Fixture===========
def setup_module(module):
    print("setup_module=============>")

def teardown_module(module):
    print("teardown_module=================>")


def setup_function(function):
    print("setup_function------------------>")

def teardown_function(function):
    print("teardown_function----------->")

def setup():
    print("setup...............>>")

def teardown():
    print("teardown...........>>")

# ==========测试用例==========
def test_multiply_3_4():
    print('test_number_3_4')
    assert multiply(3, 4) == 12

def test_multiply_a_3():
    print('test_strings_a_3')
    assert multiply('a', 3) == 'aaa'

 　　这里主要用到模块级别和函数级别的Fixture

• setup_module/teardown_moudle:在当前文件中，在所有测试用例执行之前与之后执行。
• setup_function/teardown_function:在每个测试函数之前与之后执行
• setup/teardown:在每个测试函数之前鱼之后执行。这两个方法同样可以作用于类方法。

　　ytest是支持使用测试类的，同样必须以“Test”开头，在引入测试类的情况下，Fixture的用法如下。创建test_fixtures_02.py文件

# 功能函数
def multiply(a, b):
    return a * b

class TestMultiply:
    # ========Fixture===========
    @classmethod
    def setup_class(cls):
        print("setup_class======================>")

    @classmethod
    def teardown_class(cls):
        print("teardown_class========================>")

    def setup_method(self, method):
        print("setup_method------------------>")

    def teardown_method(self, method):
        print("teardown_method------------------->")

    def setup(self):
        print("setup.....................>")

    def cleardown(self):
        print("cleardown...................>")

# ====================测试用例==================
    def test_numbers_5_6(self):
        print('test_numbers_5_6')
        assert multiply(5, 6) == 30

    def test_strings_b_2(self):
        print('test_strings_b_2')
        assert multiply('b', 2) == 'bb'

 　　这里主要用到类级别和方法级别的Fixture

• setup_class/teardown_class:在当前测试类的开始与结束时执行
• setup_method/teardown_method:在每个测试方法开始与结束时执行
• setup/teardown:在每个测试方法开始与结束时执行，同样可以作用于测试函数。