File: //lib/python3/dist-packages/numpy/testing/_private/__pycache__/noseclasses.cpython-310.pyc
o
6��a�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 rdS t �|�r|j|ju S t �|�r |j|jkS t �|�r'|j|jkS t �|�r4|j|j j
jkS t � |�durB|t � |�u S t |d�rM|j|jkS t
|t�rTdS td��)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__� isbuiltin�__name__r �isclass�ismethod�__self__� __class__� getmodule�hasattr�
isinstance�property�
ValueError)�self�module�object� r �D/usr/lib/python3/dist-packages/numpy/testing/_private/noseclasses.py� _from_module s"
z NumpyDocTestFinder._from_modulec
C sB t j�| |||||||� ddlm}m} m}
m} m} |
|�rL| j rL|j
� � D ]$\}
}|� d|
� �}||�s:| |�rK| � ||�rK| �|||||||� q'| |�r�| j r�|j
� � D ]F\}
}t
|t�rft||
�}t
|t�rqt||
�j}| |�s�| |�s�| |�s�t
|t�r�| � ||�r�|� d|
� �}
| �|||
||||� qXdS dS 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 �_recurser �itemsr! r � staticmethod�getattr� classmethod�__func__r )r �tests�obj�namer � source_lines�globs�seenr" r r# r r �valname�val�valname1r r r r' : sD �
���
��
����zNumpyDocTestFinder._findN)r r � __qualname__r! r' r r r r r s r c @ � e Zd Zdd� ZdS )�NumpyOutputCheckerc C sz t j�| |||�}|s;d|v rdS |�dd�}|�dd�}dD ]}|�d| d�}|�d| d�}q t j�| |||�}|S )Nz#randomTz'>z'<)� � z'<i%d'�int)r% �
OutputChecker� check_output�replace)r �want�got� optionflags�ret�szr r r r> l s �
�z NumpyOutputChecker.check_outputN)r r r7 r> r r r r r9 k � r9 c @ s e Zd Z ddd�ZdS )�NumpyDocTestCaser N�_c C s( || _ || _tjj| |||||d� d S )N)rB �setUp�tearDown�checker)� _result_var� _nose_objr% � DocTestCase�__init__)r �testrB rH rI rJ r/ �
result_varr r r rN � s
�zNumpyDocTestCase.__init__)r NNNNrG )r r r7 rN r r r r rF � s �rF c @ sj e Zd ZdZdZejejB ZddgZ e
Z e Z
eZejfdd�Zdd� Zd d
� Zd d � Zd
d� Zdd� ZdS )� NumpyDoctest� numpydoctesti� zgenerate_numpy_api.pyzsetup.pyc C s t �| ||� d| _d | _d S )NT)r �options�
doctest_tests�doctest_result_var�r �parser�envr r r rS � s
zNumpyDoctest.optionsc C sF t �| ||� | �� | _t�� | _| jr!dd� |jjD �|j_d S d S )Nc S s g | ] }|j d kr|�qS �r% )r0 ��.0�pr r r �
<listcomp>� s
�z*NumpyDoctest.configure.<locals>.<listcomp>) r � configure�test_finder_class�finderr% �
DocTestParserrW �enabled�plugins�r rS �configr r r r^ � s
�zNumpyDoctest.configurec C s\ t tj�|j��}tddtd�|_d|v r,|�d�}|d }t ||ji |g�|j|< dS 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 �np�scipyr$ ���N)
r �os�path�dirname�filenamerg �numpyr2 �split�
__import__)r rO �pkg_namer\ �p2r r r �set_test_context� s
�
�z NumpyDoctest.set_test_contextc c s� � | � |j�stj�d|� d S z| j�|�}W n
ty" Y d S w |s'd S |�� t |j
�}|D ] }|j s8q2|j s>||_ | �
|� | j|| j| �� | jd�V q2d S )Nz Doctest doesn't want module %s)rB rJ rP )�matchesr �npd�log�debugr` �find�AttributeError�sortr rh �examplesro ru �doctest_case_class�doctest_optflags�out_check_classrU )r r r. � module_filerO r r r �loadTestsFromModule� s2 � �
��z NumpyDoctest.loadTestsFromModulec C s t jdi t�� d S )Nr )rp �set_printoptions� print_state�r r r r � afterContext s zNumpyDoctest.afterContextc C s( t j�|�}|| jv r
dS tj�| |�S )NF)rl rm �basename�doctest_ignorerw �Doctest�wantFile)r �file�bnr r r r� s
zNumpyDoctest.wantFileN)r r r7 r0 �scorer% �NORMALIZE_WHITESPACE�ELLIPSISr r� rF r~ r9 r� r r_ rl �environrS r^ ru r� r� r� r r r r rQ � s �
* rQ 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.
� unpluggerTi� r% c C s
|| _ d S �N)� to_unplug)r r� r r r rN s
zUnplugger.__init__c C s d S r� r rV r r r rS s zUnplugger.optionsc s � fdd�|j j D �|j _ d S )Nc s g | ]
}|j � jkr|�qS r )r0 r� rZ r� r r r] s �z'Unplugger.configure.<locals>.<listcomp>)rc rd r r� r r^ s zUnplugger.configureNrY )
r r r7 �__doc__r0 rb r� rN rS r^ r r r r r�
s
r� 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.T� KNOWNFAILF)�label� isfailurec C s$ d}|j ddd|�|d�dd� d S )N�NOSE_WITHOUT_KNOWNFAILz--no-knownfail�
store_true� noKnownFailFz3Disable special handling of KnownFailure exceptions)�action�dest�default�help)�
add_option�get)r rW rX �env_optr r r rS - s
�zKnownFailurePlugin.optionsc C s. | j sd S || _t|dd�}|rd| _d S d S )Nr� F)�
can_configure�confr+ rb )r rS r� �disabler r r r^ 4 s
�z KnownFailurePlugin.configureN) r r r7 r� rb r r � knownfailrl r� rS r^ 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 sX � � }�� | � � � }||kr*ztd�||��� ty) | ��t�� � Y d S w d S )Nz6FPU mode changed from {0:#x} to {1:#x} during the test)rO �AssertionError�format�
addFailure�sys�exc_info)�result�old_mode�new_mode�r� rO r r �runH s
� ��z/FPUModeCheckPlugin.prepareTestCase.<locals>.run)� numpy.core._multiarray_testsr� )r rO r� r r� r �prepareTestCaseE s
z"FPUModeCheckPlugin.prepareTestCaseN)r r r7 r� r� r r r r r� ? s r� c @ r8 )�NumpyTestProgramc C sh | j du rtjj| jj| jj| jd�| _ | jj�| j �}|dur#|| _ | 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� verbosityre )�
testRunner�nose�core�TextTestRunnerre r� r� rc �prepareTestRunnerr� rO r� �
wasSuccessful�success)r � plug_runnerr r r �runTests[ s
� zNumpyTestProgram.runTestsN)r r r7 r� r r r r r� Z rE r� )&rl r� r% r
rp r� � nose.pluginsr rw �nose.plugins.errorclassr r �nose.plugins.baser � nose.utilr �
nosetesterr �utilsr r
r& r r= r9 rM rF �get_printoptionsr� r� rQ r� r� � KnownFailurer� r� � TestProgramr� r r r r �<module> s, O x