Merge pull request #8 from karimkhaleel/Fix-annotating-__init__.py-files-by-stripping-__init__-from-modpath

fix annotating __init__.py files by stripping __init__ from modpath

Author: asottile-sentry

auto_type_annotate.py

@@ -36,6 +36,7 @@ def _to_mod(fname: str, roots: tuple[str, ...]) -> str:
             relative.removesuffix('.py')
             .replace('/', '.')
             .replace('\\', '.')
+            .removesuffix('.__init__')
         )
     else:
         raise AssertionError(f'{fname=} not found in {roots=}')

tests/auto_type_annotate_test.py

@@ -35,6 +35,12 @@ def test_to_mod_src_layout():
     assert _to_mod('bar.py', ('.', 'src')) == 'bar'
 
 
+def test_to_mod_init_files():
+    assert _to_mod('a/__init__.py', ('.',)) == 'a'
+    assert _to_mod('a/__init__b.py', ('.',)) == 'a.__init__b'
+    assert _to_mod('a/__init__b/c.py', ('.',)) == 'a.__init__b.c'
+
+
 def _find_untyped(s):
     visitor = FindUntyped()
     visitor.visit_module(_MOD, ast.parse(s))