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)