def METHOD_NAME(func):
    old_sig = inspect.signature(func)
    @functools.wraps(func)
    @matplotlib.style.context(style)
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        __tracebackhide__ = True
        img = _ImageComparisonBase(
            func, tol=tol, remove_text=remove_text, savefig_kwargs=savefig_kwargs
        )
        matplotlib.testing.set_font_settings_for_testing()
        with _collect_new_figures() as figs:
            func(*args, **kwargs)
        assert len(figs) == len(
            baseline_images
        ), "Test generated {} images but there are {} baseline images".format(
            len(figs), len(baseline_images)
        )
        for fig, baseline in zip(figs, baseline_images):
            img.compare(fig, baseline, _default_extension, _lock=False)
    parameters = list(old_sig.parameters.values())
    new_sig = old_sig.replace(parameters=parameters)
    wrapper.__signature__ = new_sig
    return wrapper