Function _has_unittest_import

scripts/update_lib/patch_spec.py:321–328

Check if 'import unittest' is already present in the file.

(tree: ast.Module)

def _has_unittest_import(tree: ast.Module) -> bool:
    """Check if 'import unittest' is already present in the file."""
    for node in tree.body:
        if isinstance(node, ast.Import):
            for alias in node.names:
                if alias.name == UT and alias.asname is None:
                    return True
    return False


def _find_import_insert_line(tree: ast.Module) -> int: