types

Author: hugovk

tests/test_add_to_pydotorg.py

@@ -7,6 +7,10 @@
 
 import add_to_pydotorg
 
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from pyfakefs.fake_filesystem import FakeFilesystem
+
 
 @pytest.mark.parametrize(
     ["release", "expected"],
@@ -80,7 +84,7 @@ def test_minor_version_tuple(release: str, expected: tuple[int, int]) -> None:
     assert add_to_pydotorg.minor_version_tuple(release) == expected
 
 
-def test_list_files(fs) -> None:
+def test_list_files(fs: FakeFilesystem) -> None:
     # Arrange
     fs.add_real_file("tests/ftp-file-list.txt")
     fake_files = Path("tests/ftp-file-list.txt").read_text().splitlines()