Function make_call_asserter

tests/test_datastructures.py:931–955

Utility to assert a certain number of function calls. :param func: Additional callback for each function call. .. code-block:: python assert_calls, func = make_call_asserter() with assert_calls(2): func() func()

(func=None)

def make_call_asserter(func=None):
    """Utility to assert a certain number of function calls.

    :param func: Additional callback for each function call.

    .. code-block:: python
        assert_calls, func = make_call_asserter()
        with assert_calls(2):
            func()
            func()
    """
    calls = [0]

    @contextmanager
    def asserter(count, msg=None):
        calls[0] = 0
        yield
        assert calls[0] == count

    def wrapped(*args, **kwargs):
        calls[0] += 1
        if func is not None:
            return func(*args, **kwargs)

    return asserter, wrapped


class TestCallbackDict: