fix tests

Author: tobymao

tests/test_queue.py

@@ -332,7 +332,7 @@ async def test_map(self) -> None:
         task.cancel()
 
     async def test_map_timeout(self) -> None:
-        with self.assertRaises(TimeoutError):
+        with self.assertRaises((TimeoutError, asyncio.CancelledError)):
             await self.queue.map("echo", [{}], timeout=0.1)
 
     async def test_poll_interval(self) -> None: