How do I use assertions with the unittest mock library?

How do I use assertions with the unittest mock library?

You can use assertions in the unittest mock library to verify that your mock objects are being called as expected. For example, you can use assert_called_with() to check if a mock was called with specific arguments.

Can you give me an example of that?

Sure! Here's a simple example:

from unittest.mock import Mock

mock_function = Mock()
mock_function(1, 2, 3)

mock_function.assert_called_with(1, 2, 3)

This checks that mock_function was called with the arguments 1, 2, and 3.

What if I want to check how many times it was called?

You can use assert_called_once() to check if it was called exactly once, or assert_called() to check if it was called at least once. For example:

mock_function.assert_called_once()