Function _check_warning

dask/utils_test.py:155–164

Conditionally check if a warning is raised

(condition: bool, category: type[Warning], message: str)

@contextlib.contextmanager
def _check_warning(condition: bool, category: type[Warning], message: str):
    """Conditionally check if a warning is raised"""
    if condition:
        import pytest

        with pytest.warns(category, match=message) as ctx:
            yield ctx
    else:
        with contextlib.nullcontext() as ctx:
            yield ctx