78 lines
2.4 KiB
Python
78 lines
2.4 KiB
Python
|
# 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
|