diff --git a/tests/unit/test_testing/test_sub_client/demo.py b/tests/unit/test_testing/test_sub_client/demo.py index 7a14dea65e..dce3de56cb 100644 --- a/tests/unit/test_testing/test_sub_client/demo.py +++ b/tests/unit/test_testing/test_sub_client/demo.py @@ -11,7 +11,7 @@ @get("/notify/{topic:str}") async def get_notified(topic: str) -> ServerSentEvent: - async def generator() -> AsyncIterator[bytes]: + async def generator() -> AsyncIterator[str]: yield topic while True: await asyncio.sleep(0.1) diff --git a/tests/unit/test_testing/test_sub_client/test_subprocess_client.py b/tests/unit/test_testing/test_sub_client/test_subprocess_client.py index ff3784fef8..4096ee1a61 100644 --- a/tests/unit/test_testing/test_sub_client/test_subprocess_client.py +++ b/tests/unit/test_testing/test_sub_client/test_subprocess_client.py @@ -43,7 +43,7 @@ async def test_subprocess_async_client(async_client: httpx.AsyncClient) -> None: break -def test_subprocess_sync_client(sync_client: httpx.AsyncClient) -> None: +def test_subprocess_sync_client(sync_client: httpx.Client) -> None: """Demonstrates functionality of the async client with an infinite SSE source that cannot be tested with the regular async test client. """