Numpy 用模拟对象测试代码

模拟对象(mock)是真实对象的替代物,用来测试真实对象的部分行为。如果你看过电影《外星人入侵》(Body Snatchers),应该已经对模拟对象有基本的了解了。一般来说,只有在不方便创建真实对象(例如数据库连接)时,或者对真实对象的测试会产生不希望有的副作用时,才需要用到模拟对象。例如,我们可能不希望在测试过程中,把数据写入到文件系统或数据库中。

本章我们将测试一个核反应堆——当然不是真的,它只是一个类。这个核反应堆类可以做阶乘的计算。假设求阶乘会引起一系列的反应,导致发生核灾难的后果。我们将使用mock包,用一个模拟对象来模拟阶乘计算。

具体步骤

首先我们将安装mock包,然后创建一个模拟对象并用它来测试一段代码。

  1. 安装mock。

为了安装mock包,请执行如下的命令。

sudo easy_install mock
  1. 创建一个模拟对象。

核反应堆类有一个do_work方法,该方法会调用危险的factorial方法。我们希望用模拟对象替代factorial方法。用如下方式创建模拟对象。

reactor.factorial = MagicMock(return_value=6)

这将确保模拟对象被调用后,其返回值是6。

  1. 断言行为。

我们可以用若干种方法,检查模拟对象的行为,进而测试真实对象的行为。例如,我们可以断言,我们用正确的参数调用了潜在的有爆炸危险的factorial方法,具体如下所示。

reactor.factorial.assert_called_with(3, "mocked")

使用了模拟对象的完整的测试代码如下。

from mock import MagicMock
import numpy
import unittest

class NuclearReactor():
    def __init__(self, n):
        self.n = n

    def do_work(self, msg):
            print "Working"

            return self.factorial(self.n, msg)

    def factorial(self, n, msg):
            print msg 

    if n == 0:
            return 1

        if n < 0:
                raise ValueError, "Core meltdown"

        return numpy.arange(1, n+1).cumprod()

class NuclearReactorTest(unittest.TestCase):
    def test_called(self):
        reactor = NuclearReactor(3)
        reactor.factorial = MagicMock(return_value=6)
        result = reactor.do_work("mocked")
        self.assertEqual(6, result)
        reactor.factorial.assert_called_with(3, "mocked")

     def test_unmocked(self):
             reactor = NuclearReactor(3)
             reactor.factorial(3, "unmocked")
             numpy.testing.assert_rais es(ValueError)

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

我们向factorial方法传递了一个字符串,目的是说明模拟对象并不执行真实对象中的代码。单元测试的作用和上一攻略中介绍的相同。第二个测试没有对核反应堆类做任何测试,只是为了演示不使用模拟对象而直接执行真实代码的情况。

测试后的输出结果如下。

Working
.unmocked
.
----------------------------------------------------------------------
Ran 2 tests in 0.000s

OK

攻略小结

模拟对象不具有任何真实的行为。他们就像是伪装成人类的外星人,但还达不到外星人的智慧程度。正如外星人不能说出他所替换的真实人的生日一样,我们需要对模拟对象进行设置,使其能以适当的方式作出反应。例如,本例中的模拟对象返回6。我们可以记录模拟对象的调用情况——被调用了多少次,用的什么参数。

Python教程

Java教程

Web教程

数据库教程

图形图像教程

大数据教程

开发工具教程

计算机教程