pytest学习实践-day45：单元测试- 针对方法进行单元测试举例：1.写测试驱动-即写测试类 -TDD方式 2.写测试函数直接测试

def is_prime(number):
    """如果是质数，返回true"""
    for i in range(2, number):
        if number % i == 0:
            return False
    return True

--------------测试驱动：一句话解释：写一个测试类，测试一个函数------------

"""TDD：测试驱动：写一个测试类，测试一个函数"""
import pytest
from is_prime import is_prime


class TestIsPrime(object):
    """测试 is_prime"""

    @pytest.mark.parametrize("a1", [3, 4, 5])
    def test_is_prime(self, a1):
        assert is_prime(a1) is True

----  单元测试：简单的做法，写测试函数单独测试被测函数  ----- 

import pytest
from is_prime import is_prime


@pytest.mark.parametrize("a1", [3, 4, 5, 6, 7, 8, 9])
def test_is_prime(a1):
    assert is_prime(a1) is True