- fixed missing __init__.py in test;

- added always including full __new__ method, if present, as it
  seems important for the LLM to know;

src/coverup/codeinfo.py

@@ -105,7 +105,8 @@ def _summarize(path: T.List[ast.AST]) -> ast.AST:
         path[-1] = copy.deepcopy(path[-1])
         for c in ast.iter_child_nodes(path[-1]):
             if (isinstance(c, ast.ClassDef) or
-                (isinstance(c, (ast.FunctionDef, ast.AsyncFunctionDef)) and c.name != "__init__")):
+                (isinstance(c, (ast.FunctionDef, ast.AsyncFunctionDef)) and \
+                 c.name not in ("__init__", "__new__"))):
                 # Leave "__init__" unmodified as it's likely to contain important member information
                 c.body = [ast.Expr(ast.Constant(value=ast.literal_eval("...")))]
 

tests/test_codeinfo.py

@@ -173,6 +173,13 @@ def bar(self):
 
         def func(x: C):
             x.foo()
+
+        class D:
+            def __new__(cls, *args, **kwargs):
+                return super().__new__(cls)
+
+            def d(self):
+                pass
         """
     )
 
@@ -197,6 +204,16 @@ def bar(self):
                 ..."""
     )
 
+    assert codeinfo.get_info(tree, 'D') == textwrap.dedent('''\
+        class D:
+
+            def __new__(cls, *args, **kwargs):
+                return super().__new__(cls)
+
+            def d(self):
+                ...'''
+    )
+
     assert codeinfo.get_info(tree, 'C.foo') == textwrap.dedent("""\
         class C:
             ...
@@ -331,6 +348,7 @@ def test_get_info_name_includes_module_fqn(import_fixture):
     tmp_path = import_fixture
 
     (tmp_path / "foo").mkdir()
+    (tmp_path / "foo" / "__init__.py").write_text("")
     (tmp_path / "foo" / "bar.py").write_text(textwrap.dedent("""\
         class C:
             pass