3
���h� � @ sh d Z ddlZddlmZ ddlZddlZddlZG dd� dej�ZG dd� dej�Z ej
j dd � �Z dS )
z
Tests which scan for certain occurrences in the code, they may not find
all of these occurrences but should catch almost all.
� N)�Pathc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� ParseCallc C s
g | _ d S )N)�ls)�self� r �:/tmp/pip-build-5_djhm0z/numpy/numpy/tests/test_warnings.py�__init__
s zParseCall.__init__c C s t jj| |� | jj|j� d S )N)�ast� NodeVisitor�
generic_visitr �append�attr)r �noder r r �visit_Attribute s zParseCall.visit_Attributec C s | j j|j� d S )N)r r �id)r r r r r �
visit_Name s zParseCall.visit_NameN)�__name__�
__module__� __qualname__r r r r r r r r s r c s$ e Zd Z� fdd�Zdd� Z� ZS )� FindFuncsc s t � j� || _d S )N)�superr �_FindFuncs__filename)r �filename)� __class__r r r s
zFindFuncs.__init__c C s� t � }|j|j� tjj| |� |jd dks<|jd dkr`|jd jdkr`t dj
| j |j ���|jd dkr�t
|j�dks�|jd d kr�d
| j kr�d S t
|j�d kr�d S d d
� |jD �}d|kr�d S t dj
| j |j ���d S )N� � simplefilter�filterwarningsr �ignorezFwarnings should have an appropriate stacklevel; found in {} on line {}�warn� �warningsz testing/tests/test_warnings.py� c S s h | ]
}|j �qS r )�arg)�.0�kwr r r � <setcomp>2 s z'FindFuncs.visit_Call.<locals>.<setcomp>�
stacklevel���r' r' ���)r �visit�funcr r
r r �args�s�AssertionError�formatr �lineno�len�keywords)r r �pr+ r r r �
visit_Call s(
zFindFuncs.visit_Call)r r r r r3 �
__classcell__r r )r r r s r c
C s� t tj�j} xx| jd�D ]j}| d |jkr,q|| d kr:q|| d d krLqtjt|���"}t j
|j � �}t |�j
|� W d Q R X qW d S )Nz*.pyZtestingz __init__.py�random)r �numpy�__file__�parentZrglob�parents�tokenize�open�strr �parse�readr r) )�base�path�file�treer r r �test_warning_calls: s rC )
�__doc__ZpytestZpathlibr r r: r6 r
r r �markZslowrC r r r r �<module> s " |