Record simple lazy modules as well

Author: johnslavik

Lib/test/test_lazy_import/__init__.py

@@ -969,6 +969,21 @@ def test_lazy_modules_is_per_interpreter(self):
         # Basic test that sys.lazy_modules exists and is a set
         self.assertIsInstance(sys.lazy_modules, dict)
 
+    def test_lazy_module_without_children_is_tracked(self):
+        code = textwrap.dedent("""
+            import sys
+            lazy import json
+            assert "json" in sys.lazy_modules, (
+                f"expected 'json' in sys.lazy_modules, got {set(sys.lazy_modules)}"
+            )
+            assert sys.lazy_modules["json"] == set(), (
+                f"expected empty set for sys.lazy_modules['json'], "
+                f"got {sys.lazy_modules['json']!r}"
+            )
+            print("OK")
+        """)
+        assert_python_ok("-c", code)
+
 
 @support.requires_subprocess()
 class CommandLineAndEnvVarTests(unittest.TestCase):

Python/import.c

@@ -4377,6 +4377,12 @@ register_lazy_on_parent(PyThreadState *tstate, PyObject *name,
         Py_ssize_t dot = PyUnicode_FindChar(name, '.', 0,
                                             PyUnicode_GET_LENGTH(name), -1);
         if (dot < 0) {
+            PyObject *lazy_submodules = ensure_lazy_submodules(
+                (PyDictObject *)lazy_modules, name);
+            if (lazy_submodules == NULL) {
+                goto done;
+            }
+            Py_DECREF(lazy_submodules);
             ret = 0;
             goto done;
         }