async def METHOD_NAME(tx, i=1):
    nonlocal work_cancelled
    assert not work_cancelled  # no retries after cancellation!
    try:
        result = await tx.run(f"RETURN {i}")
        try:
            for _ in range(3):
                await asyncio.sleep(0)
        except asyncio.CancelledError as e:
            e.during_sleep = True
            raise
        records = [record async for record in result]
        summary = await result.consume()
        assert isinstance(summary, neo4j.ResultSummary)
        assert len(records) == 1
        assert list(records[0]) == [i]
    except asyncio.CancelledError:
        work_cancelled = True
        raise