U œËúgWã@sxddlmZddlmZmZddlmZddlmZerDddl m Z eddd œd d „ƒZ d d œdd„Z d d œdd„Z dS)é)Ú annotations)ÚAbstractContextManagerÚcontextmanager)Ú TYPE_CHECKINGé)Ú_core)Ú GeneratorÚboolzGenerator[(None, None, None)])ÚexpectedÚreturnccsjd}t ¡}|j}|j}z*dV|r@|j|ks8|j|kr@tdƒ‚W5|sd|j|ks\|j|krdtdƒ‚XdS)z5Check if checkpoints are executed in a block of code.Tz$assert_no_checkpoints block yielded!Nz'assert_checkpoints block did not yield!)rZ current_taskZ_cancel_pointsZ_schedule_pointsÚAssertionError)r Ú__tracebackhide__ZtaskZ orig_cancelZ orig_schedule©rú=/tmp/pip-unpacked-wheel-ks04xdmi/trio/testing/_checkpoints.pyÚ_assert_yields_or_not s$ÿÿ ÿÿrzAbstractContextManager[None])r cCs d}tdƒS)a©Use 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) T©r©r rrrÚassert_checkpoints srcCs d}tdƒS)aÜUse 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) TFrrrrrÚassert_no_checkpoints4srN)Ú __future__rÚ contextlibrrÚtypingrÚrÚcollections.abcrrrrrrrrÚs