Fixing test failures.

Author: advait-dixit

mypyc/irbuild/function.py

@@ -823,7 +823,11 @@ def load_type(builder: IRBuilder, typ: TypeInfo, unbounded_type: Type | None, li
         builtin_addr_type, src = builtin_names[typ.fullname]
         class_obj = builder.add(LoadAddress(builtin_addr_type, src, line))
     # This elif-condition finds the longest import that matches the load_attr_path.
-    elif module_name := max((i for i in builder.imports if load_attr_path.startswith(i)), key=len):
+    elif module_name := max(
+        (i for i in builder.imports if load_attr_path == i or load_attr_path.startswith(f"{i}.")),
+        default="",
+        key=len,
+    ):
         # Load the imported module.
         loaded_module = builder.load_module(module_name)
         # Recursively load attributes of the imported module. These may be submodules, classes or any other object.

mypyc/test-data/commandline.test

@@ -261,3 +261,31 @@ print("imported foo")
 importing...
 imported foo
 done
+
+[case testImportFromInitPy]
+# cmd: foo.py
+import foo
+
+[file pkg2/__init__.py]
+
+[file pkg2/mod2.py]
+class A:
+    class B:
+        pass
+
+[file pkg1/__init__.py]
+from pkg2.mod2 import A
+
+[file foo.py]
+import pkg1
+from typing import TypedDict
+
+class Eggs(TypedDict):
+    obj1: pkg1.A.B
+
+print(type(Eggs(obj1=pkg1.A.B())["obj1"]).__name__)
+print(type(Eggs(obj1=pkg1.A.B())["obj1"]).__module__)
+
+[out]
+B
+pkg2.mod2