saving dependency work

lib/python3.9/site-packages/astroid/test_utils.py

@@ -0,0 +1,79 @@
+# Licensed under the LGPL: https://www.gnu.org/licenses/old-licenses/lgpl-2.1.en.html
+# For details: https://github.com/PyCQA/astroid/blob/main/LICENSE
+# Copyright (c) https://github.com/PyCQA/astroid/blob/main/CONTRIBUTORS.txt
+
+"""Utility functions for test code that uses astroid ASTs as input."""
+
+from __future__ import annotations
+
+import contextlib
+import functools
+import sys
+import warnings
+from collections.abc import Callable
+
+import pytest
+
+from astroid import manager, nodes, transforms
+
+
+def require_version(minver: str = "0.0.0", maxver: str = "4.0.0") -> Callable:
+    """Compare version of python interpreter to the given one.
+    Skip the test if older.
+    """
+
+    def parse(python_version: str) -> tuple[int, ...]:
+        try:
+            return tuple(int(v) for v in python_version.split("."))
+        except ValueError as e:
+            msg = f"{python_version} is not a correct version : should be X.Y[.Z]."
+            raise ValueError(msg) from e
+
+    min_version = parse(minver)
+    max_version = parse(maxver)
+
+    def check_require_version(f):
+        current: tuple[int, int, int] = sys.version_info[:3]
+        if min_version < current <= max_version:
+            return f
+
+        version: str = ".".join(str(v) for v in sys.version_info)
+
+        @functools.wraps(f)
+        def new_f(*args, **kwargs):
+            if current <= min_version:
+                pytest.skip(f"Needs Python > {minver}. Current version is {version}.")
+            elif current > max_version:
+                pytest.skip(f"Needs Python <= {maxver}. Current version is {version}.")
+
+        return new_f
+
+    return check_require_version
+
+
+def get_name_node(start_from, name, index=0):
+    return [n for n in start_from.nodes_of_class(nodes.Name) if n.name == name][index]
+
+
+@contextlib.contextmanager
+def enable_warning(warning):
+    warnings.simplefilter("always", warning)
+    try:
+        yield
+    finally:
+        # Reset it to default value, so it will take
+        # into account the values from the -W flag.
+        warnings.simplefilter("default", warning)
+
+
+def brainless_manager():
+    m = manager.AstroidManager()
+    # avoid caching into the AstroidManager borg since we get problems
+    # with other tests :
+    m.__dict__ = {}
+    m._failed_import_hooks = []
+    m.astroid_cache = {}
+    m._mod_file_cache = {}
+    m._transform = transforms.TransformVisitor()
+    m.extension_package_whitelist = {}
+    return m