fix types

basis-company · Sep 12, 2023 · 74a386d · 74a386d
1 parent 546b8b6
commit 74a386d
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/test_queue.py b/tests/test_queue.py
@@ -12,7 +12,7 @@ class Request(NamedTuple):
     bucket: int
 
 
-context = {
+context: dict[str, float] = {
     'started': 0,
     'stopped': 0
 }
@@ -40,10 +40,10 @@ async def handle(self, *requests: Request) -> None:
 
 @mark.asyncio
 async def test_queue() -> None:
-    queue = Queue(JsonTupleSerializer(), RuntimeStorage())
+    queue: Queue = Queue(JsonTupleSerializer(), RuntimeStorage())
 
-    def get_pipe(thread: str) -> str:
-        return Tube(SyncHandler, Route(thread=thread)).pipe
+    def get_pipe(thread: int) -> str:
+        return Tube(SyncHandler, Route(thread=str(thread))).pipe
 
     await queue.register(
         SyncHandler,