save

Author: herobank110

src/pytestqt/qtbot.py

@@ -1,7 +1,7 @@
 import contextlib
 import weakref
 import warnings
-from typing import TYPE_CHECKING, Callable, Optional, Any
+from typing import TYPE_CHECKING, Callable, Optional, Any, cast
 
 from pytestqt.exceptions import TimeoutError, ScreenshotError
 from pytestqt.qt_compat import qt_api
@@ -19,14 +19,14 @@
     # versions possibly using 'qtpy' library.
     QWidget = Any
 
-BeforeCloseFunc = Callable[[QWidget], None]
+BeforeCloseFunc = Callable[["QWidget"], None]
 
 
-def _parse_ini_boolean(value):
+def _parse_ini_boolean(value: bool | str) -> bool:
     if value in (True, False):
-        return value
+        return cast("bool", value)
     try:
-        return {"true": True, "false": False}[value.lower()]
+        return {"true": True, "false": False}[cast("str", value).lower()]
     except KeyError:
         raise ValueError("unknown string for bool: %r" % value)
 
@@ -178,9 +178,7 @@ def _should_raise(self, raising_arg):
         else:
             return True
 
-    def addWidget(
-        self, widget: "QWidget", *, before_close_func: Optional[BeforeCloseFunc] = None
-    ):
+    def addWidget(self, widget, *, before_close_func: Optional[BeforeCloseFunc] = None):
         """
         Adds a widget to be tracked by this bot. This is not required, but will ensure that the
         widget gets closed by the end of the test, so it is highly recommended.
@@ -198,7 +196,7 @@ def addWidget(
             raise TypeError(f"Need to pass a QWidget to addWidget: {widget!r}")
         _add_widget(self._request.node, widget, before_close_func=before_close_func)
 
-    def waitActive(self, widget, *, timeout=5000):
+    def waitActive(self, widget, *, timeout: int = 5000):
         """
         Context manager that waits for ``timeout`` milliseconds or until the window is active.
         If window is not exposed within ``timeout`` milliseconds, raise
@@ -743,7 +741,7 @@ def mouseRelease(*args, **kwargs):
 
 def _add_widget(
     item,
-    widget: "QWidget",
+    widget,
     *,
     before_close_func: Optional[BeforeCloseFunc] = None,
 ):