Function test__check_warning

dask/tests/test_utils_test.py:39–50

()

def test__check_warning():
    class MyWarning(Warning):
        pass

    with warnings.catch_warnings():
        warnings.simplefilter("error")
        with _check_warning(True, MyWarning, "foo"):
            warnings.warn("foo", MyWarning)

    with pytest.warns(MyWarning, match="foo"):
        with _check_warning(False, MyWarning, "foo"):
            warnings.warn("foo", MyWarning)