有没有办法在TestCase之外使用Python单元测试断言？

Question

我需要创建一个伪的帮助程序类以用于单元测试（注入到测试的类中）。有没有办法在此类中使用TestCase断言？

我想将这些断言用于Fake类执行的一些常见检查。就像是：

class FakeFoo(object):

  def do_foo(self, a, b):
    assertNotNull(a)
    ...

Answer 1

您可以创建的实例unittest.TestCase()并在其上调用方法。

import unittest

tc = unittest.TestCase()
tc.assertIsNotNone(a)

在较旧的Python版本（Python 2.7和更早的版本3.0、3.1）上，您需要在类类中传递 现有
方法的名称TestCase（通常在子类上传递测试方法的名称）。__init__在这种情况下将：

tc = unittest.TestCase('__init__')
tc.assertIsNotNone(a)

但是，您可能正在寻找一个好的Mock库。mock将是一个不错的选择。

另一种选择是使用pytest，它可以增强assert语句以提供与unittest.TestCase()断言方法相同或更多的上下文；你只需写信assert a is not None。