| # Licensed under the LGPL: https://www.gnu.org/licenses/old-licenses/lgpl-2.1.en.html |
| # For details: https://github.com/pylint-dev/astroid/blob/main/LICENSE |
| # Copyright (c) https://github.com/pylint-dev/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 and skips 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 = set() |
| return m |