# 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 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
|