Typecheck tests

Author: CoolCat467

pyproject.toml

@@ -71,7 +71,7 @@ idlemypyextension = ["py.typed"]
 package = true
 
 [tool.mypy]
-files = ["src/idlemypyextension/",]
+files = ["src/idlemypyextension/", "tests"]
 mypy_path = "stubs"
 enable_error_code = ["truthy-bool", "mutable-override"]
 show_column_numbers = true

tests/test_annotate.py

@@ -13,11 +13,11 @@
 from idlemypyextension import annotate
 
 if TYPE_CHECKING:
-    from collections.abc import Collection, Sequence
+    from collections.abc import Collection, Generator, Sequence
 
 
 @pytest.fixture(autouse=True)
-def mock_extension_log() -> MagicMock:
+def mock_extension_log() -> Generator[MagicMock, None, None]:
     """Fixture to override extension_log with an empty function."""
     with patch(
         "idlemypyextension.utils.extension_log",

tests/test_tktrio.py

@@ -12,15 +12,15 @@
 
 if TYPE_CHECKING:
     import tkinter as tk
-    from collections.abc import Callable
+    from collections.abc import Callable, Generator
 
     from typing_extensions import TypeVarTuple, Unpack
 
     PosArgT = TypeVarTuple("PosArgT")
 
 
 @pytest.fixture(autouse=True)
-def mock_extension_log() -> MagicMock:
+def mock_extension_log() -> Generator[MagicMock, None, None]:
     """Fixture to override extension_log with an empty function."""
     with patch(
         "idlemypyextension.utils.extension_log",
@@ -116,7 +116,7 @@ def test_invalid_initialization() -> None:
         ValueError,
         match=r"^Must be subclass of both tk\.Misc and tk\.Wm$",
     ):
-        TkTrioRunner(None, None)
+        TkTrioRunner(None, None)  # type: ignore[arg-type]
 
 
 def test_schedule_task_not_threadsafe(trio_runner: TkTrioRunner) -> None:
@@ -149,7 +149,7 @@ async def test() -> None:
 
     assert trio_runner.run_status == RunStatus.TRIO_RUNNING_CANCELED
 
-    while trio_runner.run_status != RunStatus.NO_TASK:
+    while str(trio_runner.run_status) != str(RunStatus.NO_TASK):
         trio_runner.root.update()
 
     assert nursery.cancel_scope.cancel_called