Update test_cancel.py

Author: BobTheBuidler

tests/test_cancel.py

@@ -6,22 +6,29 @@
 @pytest.mark.parametrize("num_to_cancel", [0, 1, 2, 3])
 async def test_cancel(num_to_cancel: int) -> None:
     cache_item_task_finished = False
-  
+
     @alru_cache
     async def coro(val: int) -> int:
+        # I am a long running coro function
         nonlocal cache_item_task_finished
         await asyncio.sleep(2)
         cache_item_task_finished = True
         return val
 
-    tasks = [asyncio.create_task(coro()) for _ in range(3)]
+    # create 3 tasks for the cached function using the same key
+    tasks = [asyncio.create_task(coro(1)) for _ in range(3)]
 
     # force the event loop to run once so the tasks can begin
     await asyncio.sleep(0)
 
+    # maybe cancel some tasks
     for i in range(num_to_cancel):
         tasks[i].cancel()
 
+    # allow enough time for the non-cancelled tasks to complete
     await asyncio.sleep(3)
 
+    # check state
     assert cache_item_task_finished is num_to_cancel < 3
+    assert all(task.cancelled() for task in tasks[:num_to_cancel])
+    assert all(task.finished() for task in tasks[num_to_cancel:])