Function find_dependent_tests_tree

scripts/update_lib/deps.py:1599–1704

Find dependent tests in a tree structure. Args: module_name: Module to search for (e.g., "ftplib") lib_prefix: RustPython Lib directory max_depth: Maximum depth to recurse (default 1 = show direct + 1 level of Lib deps) Returns: Dict with structure:

(
    module_name: str,
    lib_prefix: str,
    max_depth: int = 1,
    _depth: int = 0,
    _visited_tests: set[str] | None = None,
    _visited_modules: set[str] | None = None,
)

def find_dependent_tests_tree(
    module_name: str,
    lib_prefix: str,
    max_depth: int = 1,
    _depth: int = 0,
    _visited_tests: set[str] | None = None,
    _visited_modules: set[str] | None = None,
) -> dict:
    """Find dependent tests in a tree structure.

    Args:
        module_name: Module to search for (e.g., "ftplib")
        lib_prefix: RustPython Lib directory
        max_depth: Maximum depth to recurse (default 1 = show direct + 1 level of Lib deps)

    Returns:
        Dict with structure:
        {
            "module": "ftplib",
            "tests": ["test_ftplib", "test_urllib2"],  # Direct importers
            "children": [
                {"module": "urllib.request", "tests": [...], "children": []},
                ...
            ]
        }
    """
    lib_dir = pathlib.Path(lib_prefix)
    test_dir = lib_dir / "test"

    if _visited_tests is None:
        _visited_tests = set()
    if _visited_modules is None:
        _visited_modules = set()

    # Build graphs
    test_import_graph, test_lib_imports = _build_test_import_graph(test_dir)
    lib_import_graph = _build_lib_import_graph(lib_prefix)

    # Find tests that directly import this module
    target_top = module_name.split(".")[0]
    direct_tests: set[str] = set()
    for file_key, imports in test_lib_imports.items():
        if file_key in _visited_tests:
            continue
        # Match exact module OR any child submodule
        # e.g., "xml" matches imports of "xml", "xml.parsers", "xml.etree.ElementTree"
        # but "collections._defaultdict" only matches "collections._defaultdict" (no children)
        matches = any(
            imp == module_name or imp.startswith(module_name + ".") for imp in imports
        )
        if matches:
            # Check if it's a test file
            if pathlib.Path(file_key).name.startswith("test_"):
                direct_tests.add(file_key)
                _visited_tests.add(file_key)

    # Consolidate test names (test_sqlite3/test_dbapi -> test_sqlite3)
    consolidated_tests = {_consolidate_file_key(t) for t in direct_tests}