test a function called twice in python

Question

I have the following function which is called twice

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

I need to test this function whether it is called twice. Below test case test's if it called at all/many times with given arguments.

@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)

I want to write a test case where "mock_call_me.assert_called_once_with("abc")" raises an assertion error so that i can show it is called twice.

I don't know whether it is possible or not.Can anyone tell me how to do this?

Thanks

voidspace.org.uk/python/mock/mock.html#mock.Mock.call_count

coldmind
– coldmind

2014-11-14 16:28:25 +00:00
Commented Nov 14, 2014 at 16:28 — coldmind
– coldmind, Commented Nov 14, 2014 at 16:28

coldmind · Accepted Answer · 2014-11-14 16:34:13Z

123

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

answered Nov 14, 2014 at 16:34

coldmind

5,4973 gold badges24 silver badges23 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

Yaroslav Varkhol · Accepted Answer · 2018-12-03 14:39:17Z

66

You can even check parameters passed to each call:

from mock import patch, call

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

answered Dec 3, 2018 at 14:39

Yaroslav Varkhol

9878 silver badges9 bronze badges

2 Comments

Luke Over a year ago

This method will still pass even if mock_call_me has been called more than twice. If you'd like to assert parameters and number of calls, make sure to also validate call-count using self.assertEqual(mock_call_me.call_count, 2) as mentioned above.

MadMax1138 Over a year ago

I combined both methods to validate that it was called with specific parameters and that it was only called twice: mock_call_me.assert_has_calls([call(self.val1), call(self.val2)]) self.assertEqual(mock_call_me.call_count, 2)

Akshay kamath B · Accepted Answer · 2021-11-29 17:15:57Z

14

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

answered Nov 29, 2021 at 17:15

Akshay kamath B

2413 silver badges4 bronze badges

Comments

adarsh · Accepted Answer · 2014-11-14 16:29:39Z

1

I know that if you use flexmock then you can just write it this way:

flexmock(call_me).should_receive('abc').once() flexmock(call_me).should_receive('abc').twice()

Link: http://has207.github.io/flexmock/

answered Nov 14, 2014 at 16:29

adarsh

7,0084 gold badges33 silver badges52 bronze badges

Collectives™ on Stack Overflow

test a function called twice in python

4 Answers 4

Comments

2 Comments

Comments

Comments

Hot Network Questions

Collectives™ on Stack Overflow

4 Answers 4

Comments

2 Comments

Comments

Comments

Related