如何为模拟后添加的额外功能编写测试用例

Question

from flexmock import flexmock
class Addition:
    def add(self,num1,num2):
        return num1+num2

创建模拟

mock=Addition()
flexmock(mock)
mock.should_receive('add')
assert mock.add(1,2)==3

我使用Flexmock模拟了加法课程。现在假设如果我要进一步确保添加方法仅接收整数值。我如何编写测试案例以增加要求。还是我遵循完全错误的路径？

Answer 1

，这确实取决于正确。您可以将原始功能更改为具有Exception，并且可以测试导致出现的案例，也可以执行此类操作(我使用pytest BTW(：

def test_add(num1, num2):
    with pytest.raises(TypeError):
        return num1 + num2
test_add(2, 'fizz') ---> PASSED
test_add(2, 2) ---> FAILED
Failed: DID NOT RAISE <type 'exceptions.TypeError'>