TYP: annotation of __init__ return type (PEP 484) (pandas/tests) (#46278)

Author: EkaterinaKuzkina

pandas/tests/base/test_constructors.py

@@ -51,7 +51,7 @@ def bar(self, *args, **kwargs):
             pass
 
     class Delegate(PandasDelegate, PandasObject):
-        def __init__(self, obj):
+        def __init__(self, obj) -> None:
             self.obj = obj
 
     def test_invalid_delegation(self):

pandas/tests/dtypes/test_inference.py

@@ -91,7 +91,7 @@ class MockNumpyLikeArray:
     a scalar (`is_scalar(np.array(1)) == False`), but it is not list-like either.
     """
 
-    def __init__(self, values):
+    def __init__(self, values) -> None:
         self._values = values
 
     def __iter__(self):
@@ -323,7 +323,7 @@ def test_is_dict_like_fails(ll):
 @pytest.mark.parametrize("has_contains", [True, False])
 def test_is_dict_like_duck_type(has_keys, has_getitem, has_contains):
     class DictLike:
-        def __init__(self, d):
+        def __init__(self, d) -> None:
             self.d = d
 
         if has_keys:
@@ -1937,7 +1937,7 @@ def test_is_scalar_number(self):
         #  subclasses are.
 
         class Numeric(Number):
-            def __init__(self, value):
+            def __init__(self, value) -> None:
                 self.value = value
 
             def __int__(self):

pandas/tests/extension/arrow/arrays.py

@@ -179,7 +179,7 @@ def all(self, axis=0, out=None):
 
 
 class ArrowBoolArray(ArrowExtensionArray):
-    def __init__(self, values):
+    def __init__(self, values) -> None:
         if not isinstance(values, pa.ChunkedArray):
             raise ValueError
 
@@ -189,7 +189,7 @@ def __init__(self, values):
 
 
 class ArrowStringArray(ArrowExtensionArray):
-    def __init__(self, values):
+    def __init__(self, values) -> None:
         if not isinstance(values, pa.ChunkedArray):
             raise ValueError
 

pandas/tests/extension/arrow/test_timestamp.py

@@ -40,7 +40,7 @@ def construct_array_type(cls) -> type_t[ArrowTimestampUSArray]:
 
 
 class ArrowTimestampUSArray(ArrowExtensionArray):
-    def __init__(self, values):
+    def __init__(self, values) -> None:
         if not isinstance(values, pa.ChunkedArray):
             raise ValueError
 

pandas/tests/extension/decimal/array.py

@@ -41,7 +41,7 @@ class DecimalDtype(ExtensionDtype):
     na_value = decimal.Decimal("NaN")
     _metadata = ("context",)
 
-    def __init__(self, context=None):
+    def __init__(self, context=None) -> None:
         self.context = context or decimal.getcontext()
 
     def __repr__(self) -> str:
@@ -66,7 +66,7 @@ def _is_numeric(self) -> bool:
 class DecimalArray(OpsMixin, ExtensionScalarOpsMixin, ExtensionArray):
     __array_priority__ = 1000
 
-    def __init__(self, values, dtype=None, copy=False, context=None):
+    def __init__(self, values, dtype=None, copy=False, context=None) -> None:
         for i, val in enumerate(values):
             if is_float(val):
                 if np.isnan(val):

pandas/tests/extension/json/array.py

@@ -67,7 +67,7 @@ class JSONArray(ExtensionArray):
     dtype = JSONDtype()
     __array_priority__ = 1000
 
-    def __init__(self, values, dtype=None, copy=False):
+    def __init__(self, values, dtype=None, copy=False) -> None:
         for val in values:
             if not isinstance(val, self.dtype.type):
                 raise TypeError("All values must be of type " + str(self.dtype.type))

pandas/tests/extension/list/array.py

@@ -44,7 +44,7 @@ class ListArray(ExtensionArray):
     dtype = ListDtype()
     __array_priority__ = 1000
 
-    def __init__(self, values, dtype=None, copy=False):
+    def __init__(self, values, dtype=None, copy=False) -> None:
         if not isinstance(values, np.ndarray):
             raise TypeError("Need to pass a numpy array as values")
         for val in values:

pandas/tests/extension/test_common.py

@@ -14,7 +14,7 @@ class DummyDtype(dtypes.ExtensionDtype):
 
 
 class DummyArray(ExtensionArray):
-    def __init__(self, data):
+    def __init__(self, data) -> None:
         self.data = data
 
     def __array__(self, dtype):

pandas/tests/extension/test_extension.py

@@ -8,7 +8,7 @@
 
 
 class MyEA(ExtensionArray):
-    def __init__(self, values):
+    def __init__(self, values) -> None:
         self._values = values
 
 

pandas/tests/frame/constructors/test_from_records.py

@@ -194,7 +194,7 @@ def test_from_records_bad_index_column(self):
 
     def test_from_records_non_tuple(self):
         class Record:
-            def __init__(self, *args):
+            def __init__(self, *args) -> None:
                 self.args = args
 
             def __getitem__(self, i):

pandas/tests/frame/methods/test_select_dtypes.py

@@ -15,7 +15,7 @@
 class DummyDtype(ExtensionDtype):
     type = int
 
-    def __init__(self, numeric):
+    def __init__(self, numeric) -> None:
         self._numeric = numeric
 
     @property
@@ -28,7 +28,7 @@ def _is_numeric(self):
 
 
 class DummyArray(ExtensionArray):
-    def __init__(self, data, dtype):
+    def __init__(self, data, dtype) -> None:
         self.data = data
         self._dtype = dtype
 

pandas/tests/frame/methods/test_set_index.py

@@ -595,7 +595,7 @@ def test_set_index_custom_label_type(self):
         # GH#24969
 
         class Thing:
-            def __init__(self, name, color):
+            def __init__(self, name, color) -> None:
                 self.name = name
                 self.color = color
 
@@ -673,7 +673,7 @@ def test_set_index_custom_label_type_raises(self):
 
         # purposefully inherit from something unhashable
         class Thing(set):
-            def __init__(self, name, color):
+            def __init__(self, name, color) -> None:
                 self.name = name
                 self.color = color
 

pandas/tests/frame/methods/test_to_records.py

@@ -358,7 +358,7 @@ def test_to_records_dtype_mi(self, df, kwargs, expected):
     def test_to_records_dict_like(self):
         # see GH#18146
         class DictLike:
-            def __init__(self, **kwargs):
+            def __init__(self, **kwargs) -> None:
                 self.d = kwargs.copy()
 
             def __getitem__(self, key):

pandas/tests/frame/test_arithmetic.py

@@ -41,7 +41,7 @@ def switch_numexpr_min_elements(request):
 
 
 class DummyElement:
-    def __init__(self, value, dtype):
+    def __init__(self, value, dtype) -> None:
         self.value = value
         self.dtype = np.dtype(dtype)
 

pandas/tests/frame/test_constructors.py

@@ -1363,7 +1363,7 @@ def test_constructor_sequence_like(self):
         # collections.Sequence like
 
         class DummyContainer(abc.Sequence):
-            def __init__(self, lst):
+            def __init__(self, lst) -> None:
                 self._lst = lst
 
             def __getitem__(self, n):

pandas/tests/frame/test_stack_unstack.py

@@ -1861,7 +1861,7 @@ def test_unstack_number_of_levels_larger_than_int32(self, monkeypatch):
         # GH 26314: Change ValueError to PerformanceWarning
 
         class MockUnstacker(reshape_lib._Unstacker):
-            def __init__(self, *args, **kwargs):
+            def __init__(self, *args, **kwargs) -> None:
                 # __init__ will raise the warning
                 super().__init__(*args, **kwargs)
                 raise Exception("Don't compute final result.")

pandas/tests/frame/test_subclass.py

@@ -42,7 +42,7 @@ class CustomDataFrame(DataFrame):
             custom plotting functions.
             """
 
-            def __init__(self, *args, **kw):
+            def __init__(self, *args, **kw) -> None:
                 super().__init__(*args, **kw)
 
             @property

pandas/tests/groupby/test_counting.py

@@ -364,7 +364,7 @@ class RaisingObjectException(Exception):
         pass
 
     class RaisingObject:
-        def __init__(self, msg="I will raise inside Cython"):
+        def __init__(self, msg="I will raise inside Cython") -> None:
             super().__init__()
             self.msg = msg
 

pandas/tests/indexes/datetimes/test_timezones.py

@@ -40,7 +40,7 @@
 class FixedOffset(tzinfo):
     """Fixed offset in minutes east from UTC."""
 
-    def __init__(self, offset, name):
+    def __init__(self, offset, name) -> None:
         self.__offset = timedelta(minutes=offset)
         self.__name = name
 

pandas/tests/indexes/test_index_new.py

@@ -352,7 +352,7 @@ def test_constructor_ndarray_like(self, array):
         # it should be possible to convert any object that satisfies the numpy
         # ndarray interface directly into an Index
         class ArrayLike:
-            def __init__(self, array):
+            def __init__(self, array) -> None:
                 self.array = array
 
             def __array__(self, dtype=None) -> np.ndarray:

pandas/tests/indexing/test_iloc.py

@@ -1072,7 +1072,7 @@ def test_iloc_getitem_float_duplicates(self):
     def test_iloc_setitem_custom_object(self):
         # iloc with an object
         class TO:
-            def __init__(self, value):
+            def __init__(self, value) -> None:
                 self.value = value
 
             def __str__(self) -> str:

pandas/tests/io/formats/test_console.py

@@ -12,7 +12,7 @@ class MockEncoding:
     side effect should be an exception that will be raised.
     """
 
-    def __init__(self, encoding):
+    def __init__(self, encoding) -> None:
         super().__init__()
         self.val = encoding
 

pandas/tests/io/json/test_pandas.py

@@ -585,7 +585,7 @@ def test_frame_nonprintable_bytes(self):
         # GH14256: failing column caused segfaults, if it is not the last one
 
         class BinaryThing:
-            def __init__(self, hexed):
+            def __init__(self, hexed) -> None:
                 self.hexed = hexed
                 self.binary = bytes.fromhex(hexed)
 
@@ -1817,7 +1817,7 @@ def test_to_json_multiindex_escape(self):
 
     def test_to_json_series_of_objects(self):
         class _TestObject:
-            def __init__(self, a, b, _c, d):
+            def __init__(self, a, b, _c, d) -> None:
                 self.a = a
                 self.b = b
                 self._c = _c

pandas/tests/io/json/test_readlines.py

@@ -281,7 +281,7 @@ def test_chunksize_is_incremental():
     )
 
     class MyReader:
-        def __init__(self, contents):
+        def __init__(self, contents) -> None:
             self.read_count = 0
             self.stringio = StringIO(contents)
 

pandas/tests/io/json/test_ujson.py

@@ -660,7 +660,7 @@ def toDict(self):
 
     def test_default_handler(self):
         class _TestObject:
-            def __init__(self, val):
+            def __init__(self, val) -> None:
                 self.val = val
 
             @property
@@ -714,7 +714,7 @@ def my_obj_handler(_):
 
     def test_encode_object(self):
         class _TestObject:
-            def __init__(self, a, b, _c, d):
+            def __init__(self, a, b, _c, d) -> None:
                 self.a = a
                 self.b = b
                 self._c = _c

pandas/tests/io/parser/common/test_common_basic.py

@@ -41,7 +41,7 @@ def test_override_set_noconvert_columns():
     # Usecols needs to be sorted in _set_noconvert_columns based
     # on the test_usecols_with_parse_dates test from test_usecols.py
     class MyTextFileReader(TextFileReader):
-        def __init__(self):
+        def __init__(self) -> None:
             self._currow = 0
             self.squeeze = False
 

pandas/tests/io/parser/test_python_parser_only.py

@@ -319,7 +319,7 @@ def test_python_engine_file_no_next(python_parser_only):
     parser = python_parser_only
 
     class NoNextBuffer:
-        def __init__(self, csv_data):
+        def __init__(self, csv_data) -> None:
             self.data = csv_data
 
         def __iter__(self):

pandas/tests/io/parser/test_unsupported.py

@@ -114,7 +114,7 @@ def test_python_engine(self, python_engine):
     def test_python_engine_file_no_iter(self, python_engine):
         # see gh-16530
         class NoNextBuffer:
-            def __init__(self, csv_data):
+            def __init__(self, csv_data) -> None:
                 self.data = csv_data
 
             def __next__(self):

pandas/tests/io/test_common.py

@@ -28,7 +28,7 @@
 class CustomFSPath:
     """For testing fspath on unknown objects"""
 
-    def __init__(self, path):
+    def __init__(self, path) -> None:
         self.path = path
 
     def __fspath__(self):

pandas/tests/io/test_html.py

@@ -1243,7 +1243,7 @@ def test_parse_failure_rewinds(self):
         # Issue #17975
 
         class MockFile:
-            def __init__(self, data):
+            def __init__(self, data) -> None:
                 self.data = data
                 self.at_end = False
 

pandas/tests/io/test_pickle.py

@@ -436,12 +436,12 @@ def python_pickler(obj, path):
             pickle.dump(obj, fh, protocol=-1)
 
     class MockReadResponse:
-        def __init__(self, path):
+        def __init__(self, path) -> None:
             self.file = open(path, "rb")
             if "gzip" in path:
                 self.headers = {"Content-Encoding": "gzip"}
             else:
-                self.headers = {"Content-Encoding": None}
+                self.headers = {"Content-Encoding": ""}
 
         def __enter__(self):
             return self
@@ -478,7 +478,7 @@ def test_pickle_fsspec_roundtrip():
 
 
 class MyTz(datetime.tzinfo):
-    def __init__(self):
+    def __init__(self) -> None:
         pass
 
 

pandas/tests/io/test_sql.py

@@ -1538,7 +1538,7 @@ def test_con_unknown_dbapi2_class_does_not_error_without_sql_alchemy_installed(
         self,
     ):
         class MockSqliteConnection:
-            def __init__(self, *args, **kwargs):
+            def __init__(self, *args, **kwargs) -> None:
                 self.conn = sqlite3.Connection(*args, **kwargs)
 
             def __getattr__(self, name):

pandas/tests/reshape/test_pivot.py

@@ -1999,7 +1999,7 @@ def test_pivot_number_of_levels_larger_than_int32(self, monkeypatch):
         # GH 20601
         # GH 26314: Change ValueError to PerformanceWarning
         class MockUnstacker(reshape_lib._Unstacker):
-            def __init__(self, *args, **kwargs):
+            def __init__(self, *args, **kwargs) -> None:
                 # __init__ will raise the warning
                 super().__init__(*args, **kwargs)
                 raise Exception("Don't compute final result.")

pandas/tests/scalar/timedelta/test_arithmetic.py

@@ -988,7 +988,7 @@ def test_compare_custom_object(self):
         """
 
         class CustomClass:
-            def __init__(self, cmp_result=None):
+            def __init__(self, cmp_result=None) -> None:
                 self.cmp_result = cmp_result
 
             def generic_result(self):