python - 测试在python中调用两次的函数

Question

我有以下函数被调用了两次

def func():
    i=2
    while i
       call_me("abc")
       i-=1

我需要测试这个函数是否被调用了两次。下面的测试用例测试它是否使用给定参数调用了所有/多次。

@patch('call_me')
def test_func(self,mock_call_me):
    self.val="abc"
    self.assertEqual(func(),None)
    mock_call_me.assert_called_with(self.val)

我想编写一个测试用例，其中“mock_call_me.assert_called_once_with("abc")”引发断言错误，以便我可以显示它被调用了两次。

我不知道这是否可行。谁能告诉我该怎么做？

谢谢

Answer 1

@patch('call_me')
def test_func(self,mock_call_me):
  self.assertEqual(func(),None)
  self.assertEqual(mock_call_me.call_count, 2)