Simpler definition of input_ in tests of generator coroutines (#730)

dycw · Sep 12, 2024 · 255c1f9 · 255c1f9
1 parent bbb9828
commit 255c1f9
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/src/tests/test_asyncio.py b/src/tests/test_asyncio.py
@@ -272,8 +272,8 @@ async def test_main(self, *, capsys: CaptureFixture) -> None:
         async def func() -> AsyncGenerator[int | None, float | None]:
             print("Initial")  # noqa: T201
             while True:
-                input_ = yield
-                output = round(ensure_not_none(input_))
+                input_ = ensure_not_none((yield))
+                output = round(input_)
                 if output >= 0:
                     print(f"Received {input_}, yielding {output}")  # noqa: T201
                     yield output

diff --git a/src/tests/test_functions.py b/src/tests/test_functions.py
@@ -167,7 +167,7 @@ def test_main(self, *, capsys: CaptureFixture) -> None:
         def func() -> Generator[int | None, float | None, str]:
             print("Initial")  # noqa: T201
             while True:
-                input_ = yield
+                input_ = ensure_not_none((yield))
                 output = round(ensure_not_none(input_))
                 if output >= 0:
                     print(f"Received {input_}, yielding {output}")  # noqa: T201

diff --git a/src/utilities/__init__.py b/src/utilities/__init__.py
@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.54.0"
+__version__ = "0.54.1"