python2中的Monkeypatching类方法

Question

我有一个名为square.py的模块

import math

class Square(object):
    def __init__(radius):
        self.radius = radius

    def calculate_area(self):
        return math.sqrt(self.radius) * math.pi 

我已经使用py.test编写了测试

from square import Square


def test_mocking_class_methods(monkeypatch):
    monkeypatch.setattr('test_class_pytest.Square.calculate_area', lambda: 1)
    assert Square.calculate_area() ==  1

在python 2中运行这个测试会给出以下输出：

>       assert Square.calculate_area() ==  1
E       TypeError: unbound method <lambda>() must be called with Square instance as first argument (got nothing instead)

但是在python3中也通过了同样的测试。你知道这是为什么吗?我该如何修复这个测试，使其与python 2一起工作？

Answer 1

您需要在实例上调用calculate_area()，但您在Square类上调用了它。您从未创建一个正方形来计算其面积。

Answer 2

Python2确保调用类上的方法时，第一个参数是该类的实例(通常称为self)。

所以，它的预期是这样的：

Square().calculate_area()
# Whih is equivalent to
Square.calculate_area(Square())

但是这也会抛出一个错误，因为它是一个意外的参数(TypeError: <lambda>() takes no arguments (1 given))

要防止检查self参数，请使用staticmethod装饰器：

monkeypatch.setattr('test_class_pytest.Square.calculate_area', staticmethod(lambda: 1))