Beta Shell v2.0

File Manager

~ /usr/local/python-3.12/lib/python3.12/site-packages/numpy/tests

Name	Size	Perms	Modified	Actions
[ .. / .. ]
__init__.py	0 B	-rw-r--r--	2024-04-12 10:39:51	Edit Del
__pycache__	-	drwxr-xr-x	2024-04-12 10:39:55	Del
test__all__.py	221 B	-rw-r--r--	2024-04-12 10:39:51	Edit Del
test_ctypeslib.py	11.97 KB	-rw-r--r--	2024-04-12 10:39:51	Edit Del
test_lazyloading.py	1.13 KB	-rw-r--r--	2024-04-12 10:39:51	Edit Del
test_matlib.py	1.81 KB	-rw-r--r--	2024-04-12 10:39:51	Edit Del
test_numpy_config.py	1.21 KB	-rw-r--r--	2024-04-12 10:39:51	Edit Del
test_numpy_version.py	1.44 KB	-rw-r--r--	2024-04-12 10:39:51	Edit Del
test_public_api.py	17.74 KB	-rw-r--r--	2024-04-12 10:39:51	Edit Del
test_reloading.py	2.3 KB	-rw-r--r--	2024-04-12 10:39:51	Edit Del
test_scripts.py	1.61 KB	-rw-r--r--	2024-04-12 10:39:51	Edit Del
test_warnings.py	2.23 KB	-rw-r--r--	2024-04-12 10:39:51	Edit Del

Editing: test_warnings.py (2.23 KB)

"""
Tests which scan for certain occurrences in the code, they may not find
all of these occurrences but should catch almost all.
"""
import pytest

from pathlib import Path
import ast
import tokenize
import numpy

class ParseCall(ast.NodeVisitor):
    def __init__(self):
        self.ls = []

    def visit_Attribute(self, node):
        ast.NodeVisitor.generic_visit(self, node)
        self.ls.append(node.attr)

    def visit_Name(self, node):
        self.ls.append(node.id)


class FindFuncs(ast.NodeVisitor):
    def __init__(self, filename):
        super().__init__()
        self.__filename = filename

    def visit_Call(self, node):
        p = ParseCall()
        p.visit(node.func)
        ast.NodeVisitor.generic_visit(self, node)

        if p.ls[-1] == 'simplefilter' or p.ls[-1] == 'filterwarnings':
            if node.args[0].value == "ignore":
                raise AssertionError(
                    "warnings should have an appropriate stacklevel; found in "
                    "{} on line {}".format(self.__filename, node.lineno))

        if p.ls[-1] == 'warn' and (
                len(p.ls) == 1 or p.ls[-2] == 'warnings'):

            if "testing/tests/test_warnings.py" == self.__filename:
                # This file
                return

            # See if stacklevel exists:
            if len(node.args) == 3:
                return
            args = {kw.arg for kw in node.keywords}
            if "stacklevel" in args:
                return
            raise AssertionError(
                "warnings should have an appropriate stacklevel; found in "
                "{} on line {}".format(self.__filename, node.lineno))


@pytest.mark.slow
def test_warning_calls():
    # combined "ignore" and stacklevel error
    base = Path(numpy.__file__).parent

    for path in base.rglob("*.py"):
        if base / "testing" in path.parents:
            continue
        if path == base / "__init__.py":
            continue
        if path == base / "random" / "__init__.py":
            continue
        # use tokenize to auto-detect encoding on systems where no
        # default encoding is defined (e.g. LANG='C')
        with tokenize.open(str(path)) as file:
            tree = ast.parse(file.read())
            FindFuncs(path).visit(tree)