Add assertions for handlers to ensure mypy is happy

sourcebots · Aug 13, 2023 · a363153 · a363153
1 parent 325a2f8
commit a363153
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/sbot/timeout.py b/sbot/timeout.py
@@ -27,6 +27,8 @@ def timeout_handler(signal_type: int, stack_frame: Optional[FrameType]) -> None:
     :param signal_type: The sginal that triggered this handler
     :param stack_frame: The stack frame at the time of the signal
     """
+    assert sys.platform != 'win32'
+
     logger.info(TIMEOUT_MESSAGE)
 
     # If the process doesn't terminate in a given time, exit less cleanly
@@ -46,6 +48,8 @@ def win_timeout_handler() -> None:
 
     NOTE: This function is only called on Windows.
     """
+    assert sys.platform == 'win32'
+
     logger.info(TIMEOUT_MESSAGE)
     os.kill(os.getpid(), signal.SIGTERM)