File tree Expand file tree Collapse file tree 2 files changed +1
-10
lines changed
Expand file tree Collapse file tree 2 files changed +1
-10
lines changed Original file line number Diff line number Diff line change @@ -43,8 +43,7 @@ def put(self, item: T) -> None:
4343 async def get (self ) -> T :
4444 """Remove and return an item from the queue, waiting if necessary."""
4545 if not self .queue :
46- if self .get_waiter is not None :
47- raise ConcurrencyError ("get is already running" )
46+ assert self .get_waiter is None , "cannot call get() concurrently"
4847 self .get_waiter = self .loop .create_future ()
4948 try :
5049 await self .get_waiter
Original file line number Diff line number Diff line change @@ -37,14 +37,6 @@ async def test_get_then_put(self):
3737 item = await getter_task
3838 self .assertEqual (item , 42 )
3939
40- async def test_get_concurrently (self ):
41- """get cannot be called concurrently."""
42- getter_task = asyncio .create_task (self .queue .get ())
43- await asyncio .sleep (0 ) # let the task start
44- with self .assertRaises (ConcurrencyError ):
45- await self .queue .get ()
46- getter_task .cancel ()
47-
4840 async def test_reset (self ):
4941 """reset sets the content of the queue."""
5042 self .queue .reset ([42 ])
You can’t perform that action at this time.
0 commit comments