fix mypy

Author: karpetrosyan

tests/_core/_async/test_sqlite_storage.py

@@ -1,7 +1,7 @@
 import uuid
 from dataclasses import replace
 from datetime import datetime
-from typing import Any
+from typing import Any, AsyncIterator
 from unittest.mock import AsyncMock
 from zoneinfo import ZoneInfo
 
@@ -391,6 +391,7 @@ async def test_incomplete_entries() -> None:
     )
 
     # read only part of the stream
+    assert isinstance(entry.response.stream, AsyncIterator)
     await entry.response.stream.__anext__()
 
     # Verify the entry was created but is incomplete, so get_entries should skip it

tests/_core/_sync/test_sqlite_storage.py

@@ -1,7 +1,7 @@
 import uuid
 from dataclasses import replace
 from datetime import datetime
-from typing import Any
+from typing import Any, Iterator
 from unittest.mock import MagicMock
 from zoneinfo import ZoneInfo
 
@@ -391,6 +391,7 @@ def test_incomplete_entries() -> None:
     )
 
     # read only part of the stream
+    assert isinstance(entry.response.stream, Iterator)
     entry.response.stream.__next__()
 
     # Verify the entry was created but is incomplete, so get_entries should skip it