chore: add missing type hints

add missing type hints to code

Author: guotin

pytest_rts/pytest/fake_item.py

@@ -1,8 +1,9 @@
 """This module contains a fake pytest item class"""
+from _pytest.config import Config
 
 
 class FakeItem: # pylint: disable=too-few-public-methods
  """Fake class"""
 
- def __init__(self, config) -> None:
+ def __init__(self, config: Config) -> None:
  self.config = config

pytest_rts/pytest/runner_plugin.py

@@ -1,6 +1,6 @@
 """This module contains code for running a specific test set"""
 # pylint: disable=too-few-public-methods
-from typing import List
+from typing import List, Set
 
 from _pytest.main import Session
 from _pytest.nodes import Item
@@ -12,7 +12,7 @@
 class RunnerPlugin:
  """Plugin class for pytest"""
 
- def __init__(self, existing_tests) -> None:
+ def __init__(self, existing_tests: Set[str]) -> None:
  """Set existing tests"""
  self.existing_tests = existing_tests
 

pytest_rts/tests/test_common.py

@@ -1,5 +1,8 @@
 """Tests for common utility functions"""
+from typing import cast
+
 import pytest
+from _pytest.nodes import Item
 
 from pytest_rts.utils.common import filter_pytest_items, strip_pytest_cov_testname
 
@@ -21,12 +24,12 @@
  ),
  ],
 )
-def test_strip_pytest_cov_testname(testname, expected):
+def test_strip_pytest_cov_testname(testname: str, expected: str) -> None:
  """Test pytest-cov testname stripping to actual pytest item.nodeid strings"""
  assert strip_pytest_cov_testname(testname) == expected
 
 
-def test_filter_pytest_items():
+def test_filter_pytest_items() -> None:
  """Test for filtering the test set
  based on given existing tests
  """
@@ -49,16 +52,16 @@ def get_closest_marker(self, markername: str) -> bool:
  return False
 
  collected_items = [
- FakePytestItem("test1", "test_func_1", ""),
- FakePytestItem("test2", "test_func_2", "skip"),
- FakePytestItem("test3", "test_func_3", ""),
- FakePytestItem("test4", "test_func_4", "skipif"),
- FakePytestItem("test5", "test_func_5", ""),
+ cast(Item, FakePytestItem("test1", "test_func_1", "")),
+ cast(Item, FakePytestItem("test2", "test_func_2", "skip")),
+ cast(Item, FakePytestItem("test3", "test_func_3", "")),
+ cast(Item, FakePytestItem("test4", "test_func_4", "skipif")),
+ cast(Item, FakePytestItem("test5", "test_func_5", "")),
  ]
- existing_tests = [
+ existing_tests = {
  "test_func_1",
  "test_func_3",
- ]
+ }
 
  filtered_items = filter_pytest_items(collected_items, existing_tests)
 

pytest_rts/tests/test_e2e.py

@@ -1,8 +1,10 @@
 """Integration tests for pytest-rts"""
 import os
 
+from _pytest.pytester import Testdir
 
-def test_only_new_functions_are_ran(testdir):
+
+def test_only_new_functions_are_ran(testdir: Testdir) -> None:
  """Test case for running pytest-rts when new tests are added
  and changes are not committed.
  """

pytest_rts/utils/common.py

@@ -5,7 +5,9 @@
 from _pytest.nodes import Item
 
 
-def filter_pytest_items(pytest_items: List[Item], existing_tests) -> List[Item]:
+def filter_pytest_items(
+ pytest_items: List[Item], existing_tests: Set[str]
+) -> List[Item]:
  """Select pytest items if they are new and not marked as skipped"""
  return list(
  filter(