Attempt at type checking

Author: atharva-2001

src/syrupy/extensions/base.py

@@ -15,6 +15,7 @@
     Optional,
     Set,
     Tuple,
+    Any
 )
 
 from syrupy.constants import (
@@ -67,7 +68,7 @@ def serialize(
         exclude: Optional["PropertyFilter"] = None,
         include: Optional["PropertyFilter"] = None,
         matcher: Optional["PropertyMatcher"] = None,
-        **kwargs
+        **kwargs: Dict[Any, Any],
     ) -> "SerializedData":
         """
         Serializes a python object / data structure into a string
@@ -109,7 +110,7 @@ def is_snapshot_location(self, *, location: str) -> bool:
         return location.endswith(self._file_extension)
 
     def discover_snapshots(
-        self, *, test_location: "PyTestLocation", **kwargs
+        self, *, test_location: "PyTestLocation", **kwargs: Dict[Any, Any]
     ) -> "SnapshotCollections":
         """
         Returns all snapshot collections in test site
@@ -217,7 +218,7 @@ def delete_snapshots(
 
     @abstractmethod
     def _read_snapshot_collection(
-        self, *, snapshot_location: str, **kwargs
+        self, *, snapshot_location: str, **kwargs: Dict[Any, Any]
     ) -> "SnapshotCollection":
         """
         Read the snapshot location and construct a snapshot collection object
@@ -236,15 +237,15 @@ def _read_snapshot_data_from_location(
     @classmethod
     @abstractmethod
     def _write_snapshot_collection(
-        cls, *, snapshot_collection: "SnapshotCollection", **kwargs
+        cls, *, snapshot_collection: "SnapshotCollection", **kwargs: Dict[Any, Any]
     ) -> None:
         """
         Adds the snapshot data to the snapshots in collection location
         """
         raise NotImplementedError
 
     @classmethod
-    def dirname(cls, *, test_location: "PyTestLocation", **kwargs) -> str:
+    def dirname(cls, *, test_location: "PyTestLocation", **kwargs: Dict[Any, Any]) -> str:
         test_dir = Path(test_location.filepath).parent
         return str(test_dir.joinpath(SNAPSHOT_DIRNAME))
 
@@ -260,15 +261,21 @@ class SnapshotReporter(ABC):
     _context_line_count = 1
 
     def diff_snapshots(
-        self, serialized_data: "SerializedData", snapshot_data: "SerializedData", **kwargs
+        self,
+        serialized_data: "SerializedData",
+        snapshot_data: "SerializedData",
+        **kwargs: Dict[Any, Any],
     ) -> "SerializedData":
         env = {DISABLE_COLOR_ENV_VAR: "true"}
         attrs = {"_context_line_count": 0}
         with env_context(**env), obj_attrs(self, attrs):
             return "\n".join(self.diff_lines(serialized_data, snapshot_data))
 
     def diff_lines(
-        self, serialized_data: "SerializedData", snapshot_data: "SerializedData", **kwargs
+        self,
+        serialized_data: "SerializedData",
+        snapshot_data: "SerializedData",
+        **kwargs: Dict[Any, Any],
     ) -> Iterator[str]:
         for line in self.__diff_lines(str(snapshot_data), str(serialized_data)):
             yield reset(line)
@@ -408,7 +415,7 @@ def matches(
         *,
         serialized_data: "SerializableData",
         snapshot_data: "SerializableData",
-        **kwargs
+        **kwargs: Dict[Any, Any],
     ) -> bool:
         """
         Compares serialized data and snapshot data and returns