- changed to always convert relative imports to absolute, as GPT

  sometimes repeats relative includes seen in modules verbatim in
  tests, where they aren't valid;

- fixed some broken tests;

src/coverup/codeinfo.py

@@ -274,6 +274,12 @@ def get_imports(n: ast.AST):
                 if alias_name in names:
                     if not new_imp:
                         new_imp = copy.copy(imp)
+                        if isinstance(new_imp, ast.ImportFrom):
+                            # make all imports absolute, as GPT, seeing relative imports in
+                            # modules, may want to repeat them in tests as well
+                            new_imp.module = _resolve_from_import(module.path, new_imp)
+                            new_imp.level = 0
+
                         new_imp.names = []
                         imports.append(new_imp)
                     new_imp.names.append(alias)

src/coverup/segment.py

@@ -2,7 +2,7 @@
 from pathlib import Path
 from .utils import *
 import ast
-from .codeinfo import get_global_imports
+from .codeinfo import get_global_imports, parse_file
 
 
 class CodeSegment:
@@ -89,8 +89,7 @@ def find_enclosing(root, line):
                     return (node, begin, node.end_lineno+1) # +1 for range() style
 
     for fname, fcov in coverage['files'].items():
-        with open(fname, "r") as src:
-            tree = ast.parse(src.read(), fname)
+        tree = parse_file(Path(fname))
 
         missing_lines = set(fcov['missing_lines'])
         executed_lines = set(fcov['executed_lines'])

tests/test_codeinfo.py

@@ -1012,8 +1012,46 @@ def func(x: C):
     )
 
 
-def test_get_global_imports():
-    code = textwrap.dedent("""\
+def test_get_info_included_imports_always_absolute(import_fixture):
+    tmp_path = import_fixture
+    code = tmp_path / "code.py"
+    code.write_text(textwrap.dedent("""\
+        import foo
+        """
+    ))
+
+    (tmp_path / "foo").mkdir()
+    (tmp_path / "foo" / "__init__.py").write_text(textwrap.dedent("""\
+        import sys
+        from .bar import Bar
+
+        class Foo(Bar):
+            pass
+        """
+    ))
+
+    (tmp_path / "foo" / "bar.py").write_text(textwrap.dedent("""\
+        class Bar:
+            pass
+        """
+    ))
+
+    tree = codeinfo.parse_file(code)
+    assert codeinfo.get_info(tree, 'foo.Foo') == textwrap.dedent("""\
+        in foo/__init__.py:
+        ```python
+        from foo.bar import Bar
+
+        class Foo(Bar):
+            pass
+        ```"""
+    )
+
+
+def test_get_global_imports(import_fixture):
+    tmp_path = import_fixture
+    code = tmp_path / "code.py"
+    code.write_text(textwrap.dedent("""\
         import a, b
         from c import d as e
         import os
@@ -1029,7 +1067,7 @@ class Foo:
             def f():
                 if os.path.exists("foo"):
                     sha1(a.x, b.x, e.x)
-    """)
+        """))
 
     print(code)
 
@@ -1038,7 +1076,7 @@ def find_node(tree, name):
             if isinstance(n, (ast.ClassDef, ast.FunctionDef, ast.AsyncFunctionDef)) and n.name == name:
                 return n
 
-    tree = ast.parse(code)
+    tree = codeinfo.parse_file(code)
 #    print(ast.dump(tree, indent=2))
 
     f = find_node(tree, 'f')