o US`c@s8ddlmZddlmZeddZddZdd Zd S) )contextmanager)_coreccsd}t}|j}|j}z(dV|r!|j|ks|j|kr!tdW|s2|j|ks.|j|kr4tddSdS|sF|j|ksB|j|krFtdw)NTz'assert_checkpoints block did not yield!z$assert_no_checkpoints block yielded!)r current_taskZ_cancel_pointsZ_schedule_pointsAssertionError)expected__tracebackhide__taskZ orig_cancelZ orig_scheduler D/usr/local/lib/python3.10/dist-packages/trio/testing/_checkpoints.py_assert_yields_or_nots4  r cCs d}tdS)aUse as a context manager to check that the code inside the ``with`` block either exits with an exception or executes at least one :ref:`checkpoint `. Raises: AssertionError: if no checkpoint was executed. Example: Check that :func:`trio.sleep` is a checkpoint, even if it doesn't block:: with trio.testing.assert_checkpoints(): await trio.sleep(0) Tr rr r r assert_checkpointsrcCs d}tdS)aUse as a context manager to check that the code inside the ``with`` block does not execute any :ref:`checkpoints `. Raises: AssertionError: if a checkpoint was executed. Example: Synchronous code never contains any checkpoints, but we can double-check that:: send_channel, receive_channel = trio.open_memory_channel(10) with trio.testing.assert_no_checkpoints(): send_channel.send_nowait(None) TFr rr r r assert_no_checkpoints-rrN) contextlibrrr rrr r r r s