013- unittest单元测试框架

Posted 2021-03-26 kkkhycz

unittest单元测试框架

重要的概念

1. TestCase
TestCase 是最小的测试单元，用于检查特定输入集合的特定返回值。unittest提供了TestCase基类，我们创建的测试类需要继承该基类，它可以用来创建新的测试用例
2. TestSuite
测试套件是测试用例、测试套件或两者的集合，用于组装一组要运行的测试。unittest提供了TestSuite类来创建测试套件
3. Test Runner
Test Runner是一个组件，用于协调测试的执行并向用户提供结果。unittest提供了TextTestRunner类运行测试用例
4. TestFixture
TestFixture代表执行一个或多个测试所需的环境准备，以及关联的清理动作。uinttest提供了setUp( )/tearDown( )、setUpClass( )/tearDownClass( )

TestSuite

调用TestSuite类下面的addTest( )来添加测试用例。因为一次只能添加一条用例，所以需要指定测试类及测试方法

suit = unittest.TestSuite()
suit.addTest(Calculator("test_add"))
suit.addTest(Calculator("test_sub"))

1. 通过测试套件添加顺序可以控制测试用例的执行顺序
2. 当一个测试文件中有很多测试用例时，并不是每次都要执行所有的测试用例，因而可以通过测试套件和运行器可以灵活的控制要执行的测试用例

discover方法

unittest提供了defaultTestLoader类中的discover( )方法，可以从多个文件中查找测试用例

discover(start_dir,pattern="test*.py",top_level_dir=None )

1. start_dir：待测试的模块名或测试用例目录
2. pattern="test*.py"：测试用例文件名的匹配原则。此处匹配文件名以"test"开头的“.py”类型的文件，星号“*”表示任意多个字符
3. top_level_dir=None：测试模块的顶层目录，如果没有顶层目录，则默认为None

import unittest
# 定义测试用例的目录为当前目录中的test_case目录
test_dir = r"./test_case"
suit = unittest.defaultTestLoader.discover(test_dir,pattern="test*.py")

if __name__ == "__main__":
runner = unittest.TextTestRunner()
runner.run(suit)

如果将discover( )方法中的start_dir 参数定义为“./test_case目录，那么只能加载test_a.py文件中的测试用例。如果想让unittest查找test_case/下子目录中的测试文件，需要在每个子目录下放一个__init__.py文件。__init__.py文件的作用是将一个目标标记成一个标准的Python模块

跳过测试和预期失败

在运行测试时，有时需要直接跳过某些测试用例，或则当测试用例符合某个条件时跳过测试，又或者直接将测试用例设置为失败。unittest提供了实现这些需求的装饰器

unittest.skip(reason)
无条件的跳过装饰的测试，需要说明跳过测试的原因
unittest.skipIf(condition,reason)
如果条件为真，则跳过装饰的测试
unittest.skipUnless(condition,reason)
当条件为真时，执行装饰的测试
unittest.expectedFailure()
不管执行结果是否失败，都将测试标记为失败

import unittest

class Mytest(unittest.TestCase):
@unittest.skip("直接跳过测试")
def test_skip(self):
print("aaa")

@unittest.skipIf(3 > 2 ,"条件为真，跳过测试")
def test_skipIf(self):
print("bbb")

@unittest.skipUnless(3 > 2,"条件为真，执行用例")
def test_skipUnless(self):
print("ccc")

@unittest.expectedFailure
def test_expectedFailure(self):
self.assertEqual(2,3)


if "__name__" == "__main__":
unittest.main()

执行结果

Skipped: 直接跳过测试

Skipped: 条件为真，跳过测试
ccc


Ran 4 tests in 0.006s

OK (skipped=2, expected failures=1)

 

单元测试部分代码：

1. 创建一个calculator.py文件

class Calculator:

    def __init__(self,a,b):
        self.a = int(a)
        self.b = int(b)

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

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

    def mul(self):
        return self.a * self.b

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

2. 通过unittest单元测试框架编写测试用例

import unittest
from Calculator import Calculator

class TestCalculator(unittest.TestCase):
    def test_add(self):
        c = Calculator(3,5)     # type(c):<class ‘Calculator.Calculator‘>
        result = c.add()        # type(result):<class ‘int‘>
        self.assertEqual(result,8)

    def test_sub(self):
        c = Calculator(13,5)
        result = c.sub()
        self.assertEqual(result, 8)

    def test_mul(self):
        c = Calculator(2,3)
        result = c.mul()
        self.assertEqual(result,5)

    def test_div(self):
        c = Calculator(6,3)
        result = c.div()
        self.assertEqual(result,2)


if __name__ == "__main__":
    unittest.main()

3. 创建测试套件跟测试运行器

import unittest
from Calculator import Calculator

class TestCalculator(unittest.TestCase):
    # 测试用例前置动作
    def setUp(self):
        print("test start:")
    # 测试用例后置动作
    def tearDown(self):
        print("test end:")

    def test_add(self):
        c = Calculator(3,5)     # type(c):<class ‘Calculator.Calculator‘>
        result = c.add()        # type(result):<class ‘int‘>
        self.assertEqual(result,8)

    def test_sub(self):
        c = Calculator(13,5)
        result = c.sub()
        self.assertEqual(result, 8)

    def test_mul(self):
        c = Calculator(2,3)
        result = c.mul()
        self.assertEqual(result,5)

    def test_div(self):
        c = Calculator(6,3)
        result = c.div()
        self.assertEqual(result,2)


if __name__ == "__main__":
    # 创建测试套件
    suit = unittest.TestSuite()
    suit.addTest(Calculator("test_add"))
    suit.addTest(Calculator("test_sub"))
    suit.addTest(Calculator("test_mul"))
    suit.addTest(Calculator("test_div"))
    # 创建测试运行器
    runner = unittest.TextTestRunner()
    runner.run(suit)