U
    /d                     @   s   d Z ddlZddlmZ ddlZddlZddlZddlZG dd dejZ	G dd dejZ
ejdd	d
d Zejjdd Zejjejjdddd ZdS )z
Tests which scan for certain occurrences in the code, they may not find
all of these occurrences but should catch almost all. This file was adapted
from NumPy.
    N)Pathc                   @   s$   e Zd Zdd Zdd Zdd ZdS )	ParseCallc                 C   s
   g | _ d S N)ls)self r   B/tmp/pip-unpacked-wheel-9gxwnfpp/scipy/_lib/tests/test_warnings.py__init__   s    zParseCall.__init__c                 C   s    t j| | | j|j d S r   )astNodeVisitorgeneric_visitr   appendattrr   noder   r   r   visit_Attribute   s    zParseCall.visit_Attributec                 C   s   | j |j d S r   )r   r   idr   r   r   r   
visit_Name   s    zParseCall.visit_NameN)__name__
__module____qualname__r	   r   r   r   r   r   r   r      s   r   c                       s$   e Zd Z fddZdd Z  ZS )	FindFuncsc                    s    t    || _g | _g | _d S r   )superr	   _FindFuncs__filenamebad_filtersbad_stacklevels)r   filename	__class__r   r   r	      s    
zFindFuncs.__init__c                 C   s   t  }||j tj| | |jd dks<|jd dkrd|jd jdkrd| j	
d| j|j |jd dkrt|jdks|jd	 d
kr| jdkrd S t|jdkrd S dd |jD }d|kr| j
d| j|j d S )Nsimplefilterfilterwarningsr   ignorez{}:{}warn   warningsz_lib/tests/test_warnings.py   c                 S   s   h | ]
}|j qS r   )arg).0kwr   r   r   	<setcomp>8   s     z'FindFuncs.visit_Call.<locals>.<setcomp>
stacklevel)r   visitfuncr
   r   r   r   argssr   r   formatr   linenolenkeywordsr   )r   r   pr/   r   r   r   
visit_Call$   s,    
zFindFuncs.visit_Call)r   r   r   r	   r6   __classcell__r   r   r   r   r      s   r   session)Zscopec               
   C   s   t tjj} g }g }| dD ]d}tt|L}tj	|
 t|d}t|| }|| ||j ||j W 5 Q R X q||fS )Nz*.py)r   )r   scipy__file__parentrglobtokenizeopenstrr
   parsereadr   relative_tor-   extendr   r   )baser   r   pathfiletreefinderr   r   r   warning_calls>   s    
rI   c              	      s   | \}}t jddt jddt jddt jddt jddt jdd	f  fd
d|D }|r~tdd|d S )NZdatasetsz_fetchers.pyz__init__.pyoptimizez_optimize.pysparsestatsz_discrete_distns.pyz_continuous_distns.pyc                    s"   g | ]}| d d  kr|qS ):r   )split)r)   itemZallowed_filtersr   r   
<listcomp>f   s
      z.test_warning_calls_filters.<locals>.<listcomp>xwarning ignore filter should not be used, instead, use
numpy.testing.suppress_warnings (in tests only);
found in:
    {}
    )osrE   joinAssertionErrorr1   )rI   r   r   r   rP   r   test_warning_calls_filtersS   s    
rW   zstacklevels currently missing)reasonc                 C   sT   | \}}d}|r,|d d|7 }|d7 }|rD|d d|7 }|rPt|d S )N rR   rS   z

z6warnings should have an appropriate stacklevel:
    {})r1   rU   rV   )rI   r   r   msgr   r   r   test_warning_calls_stacklevelsq   s    r[   )__doc__rT   pathlibr   r
   r=   r9   Zpytestr   r   r   ZfixturerI   markZslowrW   Zxfailr[   r   r   r   r   <module>   s   !


