Add __lenght_hint__ magic method return type of int (#43)

sobolevn · web-flow · commit f90292139165 · 2022-10-13T11:55:51.000-07:00
diff --git a/README.md b/README.md
@@ -43,6 +43,7 @@ more transformations. The following are supported:
     - `__str__` returns `str`
     - `__repr__` returns `str`
     - `__len__` returns `int`
+    - `__length_hint__` returns `int`
     - `__init__` returns `None`
     - `__del__` returns `None`
     - `__bool__` returns `bool`
diff --git a/autotyping/autotyping.py b/autotyping/autotyping.py
@@ -67,6 +67,7 @@ class State:
     "__str__": "str",
     "__repr__": "str",
     "__len__": "int",
+    "__length_hint__": "int",
     "__init__": "None",
     "__del__": "None",
     "__bool__": "bool",
diff --git a/tests/test_codemod.py b/tests/test_codemod.py
@@ -281,6 +281,23 @@ def __not_str__():
         """
         self.assertCodemod(before, after, annotate_magics=True)
 
+    def test_annotate_magics_len(self) -> None:
+        before = """
+            def __len__():
+                pass
+
+            def __length_hint__():
+                pass
+        """
+        after = """
+            def __len__() -> int:
+                pass
+
+            def __length_hint__() -> int:
+                pass
+        """
+        self.assertCodemod(before, after, annotate_magics=True)
+
     def test_exit(self) -> None:
         before = """
             def __exit__(self, typ, value, tb):
