()
| 32 | |
| 33 | |
| 34 | def test_broken_callback(): |
| 35 | from dask.callbacks import Callback |
| 36 | |
| 37 | def _f_ok(*args, **kwargs): |
| 38 | pass |
| 39 | |
| 40 | def _f_broken(*args, **kwargs): |
| 41 | raise ValueError("my_exception") |
| 42 | |
| 43 | dsk = {"x": 1} |
| 44 | |
| 45 | with Callback(start=_f_broken, finish=_f_ok): |
| 46 | with Callback(start=_f_ok, finish=_f_ok): |
| 47 | with pytest.raises(ValueError, match="my_exception"): |
| 48 | get(dsk, "x") |
| 49 | |
| 50 | |
| 51 | def bad(x): |