"""

Tests which scan for certain occurrences in the code, they may not find

all of these occurrences but should catch almost all.

"""

import ast

import tokenize

from pathlib import Path

import pytest

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 and getattr(node.args[0], "value", None) == "ignore":

if not self.__filename.name.startswith("test_"):

raise AssertionError(

"ignore filters should only be used in tests; "

f"found in {self.__filename} on line {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 or skip_file_prefixes exists:

if len(node.args) == 3:

return

args = {kw.arg for kw in node.keywords}

if "stacklevel" in args:

return

if "skip_file_prefixes" in args:

return

raise AssertionError(

"warnings should have an appropriate stacklevel or skip_file_prefixes; "

f"found in {self.__filename} on line {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

if path == base / "conftest.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)