Current File : //lib64/python3.9/site-packages/numpy/testing/_private/__pycache__/noseclasses.cpython-39.pyc
a
z[yc�8 � @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dl m
Z
m Z d dl m
Z
d dlmZ ddlmZ ddlmZmZ G d d
� d
ej�ZG d d � d ej�ZG d
d� dej�Ze�� Z G dd� dej �Z G dd� d�Z G dd� de �Z e Z!G dd� de
�Z"G dd� dej#j$�Z%dS )� N)�doctests)�
ErrorClass�ErrorClassPlugin)�Plugin)�src� )�get_package_name)�KnownFailureException�KnownFailureTestc @ s e Zd Zdd� Zdd� ZdS )�NumpyDocTestFinderc C s� |du r dS t �|�r"|j|ju S t �|�r8|j|jkS t �|�rN|j|jkS t �|�rh|j|j j
jkS t � |�dur�|t � |�u S t |d�r�|j|jkS t
|t�r�dS td��dS )zY
Return true if the given object is defined in the given
module.
NT�
__module__z"object must be a class or function)�inspect�
isfunction�__dict__� __globals__Z isbuiltin�__name__r �isclass�ismethod�__self__� __class__Z getmodule�hasattr�
isinstance�property�
ValueError)�self�module�object� r �H/usr/lib64/python3.9/site-packages/numpy/testing/_private/noseclasses.py� _from_module s"
z NumpyDocTestFinder._from_modulec
C sD t j�| |||||||� ddlm}m} m}
m} m} |
|�r�| j r�|j
� � D ]H\}
}|� d|
� �}||�st| |�rN| � ||�rN| �|||||||� qN| |��r@| j �r@|j
� � D ]�\}
}t
|t�r�t||
�}t
|t�r�t||
�j}| |��s| |��s| |��st
|t�r�| � ||�r�|� d|
� �}
| �|||
||||� q�dS )zm
Find tests for the given object and any contained objects, and
add them to `tests`.
r )� isroutiner �ismoduler r �.N)�doctest�
DocTestFinder�_findr
r r r! r r Z_recurser �itemsr r � staticmethod�getattr� classmethod�__func__r )r �tests�obj�namer Z source_lines�globs�seenr r r! r r Zvalname�valZvalname1r r r r% : s< �
��
��
��zNumpyDocTestFinder._findN)r r � __qualname__r r% r r r r r s r c @ s e Zd Zdd� ZdS )�NumpyOutputCheckerc C sz t j�| |||�}|svd|v r"dS |�dd�}|�dd�}dD ]$}|�d| d�}|�d| d�}q>t j�| |||�}|S )Nz#randomTz'>z'<)� � z'<i%d'�int)r# �
OutputChecker� check_output�replace)r ZwantZgot� optionflags�retZszr r r r7 l s �
�z NumpyOutputChecker.check_outputN)r r r1 r7 r r r r r2 k s r2 c @ s e Zd Zddd�ZdS )�NumpyDocTestCaser N�_c C s( || _ || _tjj| |||||d� d S )N)r9 �setUp�tearDown�checker)Z _result_varZ _nose_objr# � DocTestCase�__init__)r �testr9 r= r> r? r, �
result_varr r r rA � s
�zNumpyDocTestCase.__init__)r NNNNr< )r r r1 rA r r r r r; � s �r; c @ sj e Zd ZdZdZejejB ZddgZ e
Z e Z
eZejfdd�Zdd� Zd d
� Zd d � Zd
d� Zdd� ZdS )� NumpyDoctestZ numpydoctesti� zgenerate_numpy_api.pyzsetup.pyc C s t �| ||� d| _d | _d S )NT)r �optionsZ
doctest_tests�doctest_result_var�r �parser�envr r r rE � s zNumpyDoctest.optionsc C sB t �| ||� | �� | _t�� | _| jr>dd� |jjD �|j_d S )Nc S s g | ]}|j d kr|�qS )r# )r- ��.0�pr r r �
<listcomp>� s
�z*NumpyDoctest.configure.<locals>.<listcomp>) r � configure�test_finder_class�finderr# Z
DocTestParserrH �enabled�plugins�r rE �configr r r rN � s
zNumpyDoctest.configurec C sX t tj�|j��}tddtd�|_d|v rT|�d�}|d }t ||ji |g�|j|< dS )a` Configure `test` object to set test context
We set the numpy / scipy standard doctest namespace
Parameters
----------
test : test object
with ``globs`` dictionary defining namespace
Returns
-------
None
Notes
-----
`test` object modified in place
�__main__)� __builtins__�__file__r ZnpZscipyr" ���N)
r �os�path�dirname�filenamerV �numpyr. �split�
__import__)r rB �pkg_namerL Zp2r r r �set_test_context� s
�
z NumpyDoctest.set_test_contextc c s� | � |j�s tj�d|� d S z| j�|�}W n tyB Y d S 0 |sLd S |�� t |j
�}|D ]>}|j snqb|j sz||_ | �
|� | j|| j| �� | jd�V qbd S )Nz Doctest doesn't want module %s)r9 r? rC )�matchesr �npd�log�debugrP �find�AttributeError�sortr rW Zexamplesr\ ra �doctest_case_class�doctest_optflags�out_check_classrF )r r r+ Z module_filerB r r r �loadTestsFromModule� s,
�z NumpyDoctest.loadTestsFromModulec C s t jf i t�� d S �N)r] Zset_printoptions� print_state�r r r r � afterContext s zNumpyDoctest.afterContextc C s( t j�|�}|| jv rdS tj�| |�S )NF)rY rZ �basename�doctest_ignorerc �Doctest�wantFile)r �fileZbnr r r rt s
zNumpyDoctest.wantFileN)r r r1 r- �scorer# ZNORMALIZE_WHITESPACE�ELLIPSISrj rr r; ri r2 rk r rO rY �environrE rN ra rl rp rt r r r r rD � s �
* rD c @ s6 e Zd ZdZdZdZdZd
dd�Zdd � Zd
d � Z d S )� Unpluggerzi Nose plugin to remove named plugin late in loading
By default it removes the "doctest" plugin.
Z unpluggerTi� r# c C s
|| _ d S rm )� to_unplug)r rz r r r rA s zUnplugger.__init__c C s d S rm r rG r r r rE s zUnplugger.optionsc s � fdd�|j j D �|j _ d S )Nc s g | ]}|j � jkr|�qS r )r- rz rJ ro r r rM s �z'Unplugger.configure.<locals>.<listcomp>)rR rS r ro r rN s zUnplugger.configureN)r# )
r r r1 �__doc__r- rQ rv rA rE rN r r r r ry
s
ry c @ s8 e Zd ZdZdZeeddd�Zej fdd�Z
dd � Z d
S ) �KnownFailurePlugina1 Plugin that installs a KNOWNFAIL error class for the
KnownFailureClass exception. When KnownFailure is raised,
the exception will be logged in the knownfail attribute of the
result, 'K' or 'KNOWNFAIL' (verbose) will be output, and the
exception will not be counted as an error or failure.TZ KNOWNFAILF)ZlabelZ isfailurec C s$ d}|j ddd|�|d�dd� d S )NZNOSE_WITHOUT_KNOWNFAILz--no-knownfail�
store_true� noKnownFailFz3Disable special handling of KnownFailure exceptions)�action�dest�default�help)Z
add_option�get)r rH rI Zenv_optr r r rE - s
�zKnownFailurePlugin.optionsc C s* | j s
d S || _t|dd�}|r&d| _d S )Nr~ F)Z
can_configure�confr( rQ )r rE r� �disabler r r rN 4 s z KnownFailurePlugin.configureN) r r r1 r{ rQ r r Z knownfailrY rx rE rN r r r r r| " s �r| c @ s e Zd ZdZdd� ZdS )�FPUModeCheckPluginzx
Plugin that checks the FPU mode before and after each test,
raising failures if the test changed the mode.
c s ddl m� � �fdd�}|S )Nr )� get_fpu_modec sZ � � }�� | � � � }||krVztd�||���W n" tyT | ��t�� � Y n0 d S )Nz6FPU mode changed from {0:#x} to {1:#x} during the test)rB �AssertionError�formatZ
addFailure�sys�exc_info)�resultZold_modeZnew_mode�r� rB r r �runH s
�� z/FPUModeCheckPlugin.prepareTestCase.<locals>.run)Z numpy.core._multiarray_testsr� )r rB r� r r� r �prepareTestCaseE s
z"FPUModeCheckPlugin.prepareTestCaseN)r r r1 r{ r� r r r r r� ? s r� c @ s e Zd Zdd� ZdS )�NumpyTestProgramc C sh | j du r(tjj| jj| jj| jd�| _ | jj�| j �}|durF|| _ | j � | j
�| _ | j � � | _
| j
S )a Run Tests. Returns true on success, false on failure, and
sets self.success to the same value.
Because nose currently discards the test result object, but we need
to return it to the user, override TestProgram.runTests to retain
the result
N)�stream� verbosityrT )Z
testRunner�nose�coreZTextTestRunnerrT r� r� rR ZprepareTestRunnerr� rB r� Z
wasSuccessful�success)r Z plug_runnerr r r �runTests[ s
� zNumpyTestProgram.runTestsN)r r r1 r� r r r r r� Z s r� )&rY r� r# r
r] r� Z nose.pluginsr rc Znose.plugins.errorclassr r Znose.plugins.baser Z nose.utilr Z
nosetesterr Zutilsr r
r$ r r6 r2 r@ r; Zget_printoptionsrn rs rD ry r| Z KnownFailurer� r� Z TestProgramr� r r r r �<module> s* O x
Mini Shell