单元测试框架之unittest(四)

一、摘要

假设我们有一组测试方法差别非常小，比如仅仅是所需要的参数有少许变化时，我们的自动化测试如何进行？unittest框架为这种场景提供了一种方式，它允许我们用subTest()上下文管理器在一个测试方法内识别这种细小的不同，这么说可能有点晕

实际上我觉得还不如直接看代码更能清晰的表达它为我们的测试带来了什么

二、代码实例

# coding:utf-8
import unittest

class NumbersTest(unittest.TestCase):

    def test_even(self):
        """
        使用subTest上下文管理器，区分细小的变化
        取模运算，返回除法的余数，但是参数是0到5的整数，没必要单独写方法
        """
        for i in range(0, 6):
            with self.subTest(i=i):
                self.assertEqual(i % 2, 0)

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

执行这段代码的结果会是：

SubTest failure: Traceback (most recent call last):
  File "C:\Python37\lib\unittest\case.py", line , in testPartExecutor
    yield
  File "C:\Python37\lib\unittest\case.py", line , in subTest
    yield
  File "D:\Programs\Python\Demo\unittest4\subtestDemo.py", line , in test_even
    self.assertEqual(i % , )
  File "C:\Program Files\JetBrains\PyCharm Community Edition 2018.3.1\helpers\pycharm\teamcity\diff_tools.py", line , in _patched_equals
    raise native_error
  File "C:\Program Files\JetBrains\PyCharm Community Edition 2018.3.1\helpers\pycharm\teamcity\diff_tools.py", line , in _patched_equals
    old(self, first, second, msg)
  File "C:\Python37\lib\unittest\case.py", line , in assertEqual
    assertion_func(first, second, msg=msg)
  File "C:\Python37\lib\unittest\case.py", line , in _baseAssertEqual
    raise self.failureException(msg)
AssertionError:  != 

SubTest failure: Traceback (most recent call last):
  File "C:\Python37\lib\unittest\case.py", line , in testPartExecutor
    yield
  File "C:\Python37\lib\unittest\case.py", line , in subTest
    yield
  File "D:\Programs\Python\Demo\unittest4\subtestDemo.py", line , in test_even
    self.assertEqual(i % , )
  File "C:\Program Files\JetBrains\PyCharm Community Edition 2018.3.1\helpers\pycharm\teamcity\diff_tools.py", line , in _patched_equals
    raise native_error
  File "C:\Program Files\JetBrains\PyCharm Community Edition 2018.3.1\helpers\pycharm\teamcity\diff_tools.py", line , in _patched_equals
    old(self, first, second, msg)
  File "C:\Python37\lib\unittest\case.py", line , in assertEqual
    assertion_func(first, second, msg=msg)
  File "C:\Python37\lib\unittest\case.py", line , in _baseAssertEqual
    raise self.failureException(msg)
AssertionError:  != 

SubTest failure: Traceback (most recent call last):
  File "C:\Python37\lib\unittest\case.py", line , in testPartExecutor
    yield
  File "C:\Python37\lib\unittest\case.py", line , in subTest
    yield
  File "D:\Programs\Python\Demo\unittest4\subtestDemo.py", line , in test_even
    self.assertEqual(i % , )
  File "C:\Program Files\JetBrains\PyCharm Community Edition 2018.3.1\helpers\pycharm\teamcity\diff_tools.py", line , in _patched_equals
    raise native_error
  File "C:\Program Files\JetBrains\PyCharm Community Edition 2018.3.1\helpers\pycharm\teamcity\diff_tools.py", line , in _patched_equals
    old(self, first, second, msg)
  File "C:\Python37\lib\unittest\case.py", line , in assertEqual
    assertion_func(first, second, msg=msg)
  File "C:\Python37\lib\unittest\case.py", line , in _baseAssertEqual
    raise self.failureException(msg)
AssertionError:  != 

One or more subtests failed
Failed subtests list: (i=), (i=), (i=)

Ran  test in .020s

FAILED (failures=)

Process finished with exit code 

而如果我们不使用subTest(), 只是写个简单的循环去断言，当程序执行到第一个断言失败时就会终止了，后边可能还有断言能够成功的也就不会被执行了

# coding:utf-8
import unittest

class NumbersTest(unittest.TestCase):

    def test_even(self):
        for i in range(0, 6):
            # with self.subTest(i=i):
            print("当前参数是：%d" % i)
            self.assertEqual(i % 2, 0)

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

执行结果会是：

当前参数是：
当前参数是：

Ran  test in .010s

FAILED (failures=)

 != 

Expected :
Actual   :
 <Click to see difference>

Traceback (most recent call last):
  File "C:\Program Files\JetBrains\PyCharm Community Edition 2018.3.1\helpers\pycharm\teamcity\diff_tools.py", line , in _patched_equals
    old(self, first, second, msg)
  File "C:\Python37\lib\unittest\case.py", line , in assertEqual
    assertion_func(first, second, msg=msg)
  File "C:\Python37\lib\unittest\case.py", line , in _baseAssertEqual
    raise self.failureException(msg)
AssertionError:  != 

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "C:\Python37\lib\unittest\case.py", line , in testPartExecutor
    yield
  File "C:\Python37\lib\unittest\case.py", line , in run
    testMethod()
  File "D:\Programs\Python\Demo\unittest4\subtestDemo.py", line , in test_even
    self.assertEqual(i % , )

Process finished with exit code