Fix linter errors

dosisod · Dec 27, 2023 · ac56b62 · ac56b62
1 parent 6103901
commit ac56b62
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/test/conftest.py b/test/conftest.py
@@ -1,8 +1,11 @@
-from unittest.mock import patch
+from collections.abc import Generator
+from unittest.mock import Mock, patch
+
 import pytest
 
+
 @pytest.fixture(autouse=True)
-def fake_tty():
+def fake_tty() -> Generator[Mock, None, None]:
     # Pytest doesnt run in a TTY, so the new TTY detection code is causing a lot of color related
     # tests to fail. This hack makes it so color is always enabled, like it would in a normal TTY.
     with patch("sys.stdout.isatty") as p: