?????????? ????????? - ??????????????? - /usr/lib64/python3.9/site-packages/numpy/core/tests/__pycache__/test_overrides.cpython-39.pyc
???????
a z[yc�N � @ sN d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dlZ d dl m Z m Z m Z mZ d dlmZmZmZmZ d dlmZ d dlZejje dd�Zd d � Zed d � �d d� �Zedd � �dd� �Z G dd� d�Z G dd� d�Z eG dd� d��Z eG dd� d��Z dd� Z!eG d d � d ��Z"G d d � d �Z#G d d!� d!�Z$G d"d#� d#�Z%dS )$� N��StringIO)�mock)�assert_� assert_equal� assert_raises�assert_raises_regex)�_get_implementing_args�array_function_dispatch�verify_matching_signatures�ARRAY_FUNCTION_ENABLED)�picklez(__array_function__ dispatch not enabled.)�reasonc O s t S �N��NotImplemented��self�args�kwargs� r �E/usr/lib64/python3.9/site-packages/numpy/core/tests/test_overrides.py�_return_not_implemented s r c C s | fS r r ��arrayr r r �<lambda> � r c C s dS �� Docstring.�originalr r r r r �dispatched_one_arg s r c C s | |fS r r �Zarray1Zarray2r r r r " r c C s dS r r r! r r r �dispatched_two_arg" s r" c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )�TestGetImplementingArgsc C s| t �d�}t|g�}tt|�|g� t||g�}tt|�|g� t|dg�}tt|�|g� td|g�}tt|�|g� d S )N� ��npr r r �list)r r r r r r � test_ndarray* s z$TestGetImplementingArgs.test_ndarrayc C s� G dd� dt j�}G dd� dt j�}t �d��t j�}t �d��|�}t �d��|�}t||g�}tt|�||g� t||g�}tt|�||g� t||g�}tt|�||g� d S )Nc @ s e Zd ZeZdS )zDTestGetImplementingArgs.test_ndarray_subclasses.<locals>.OverrideSubN��__name__� __module__� __qualname__r �__array_function__r r r r � OverrideSub; s r. c @ s e Zd ZdS )zFTestGetImplementingArgs.test_ndarray_subclasses.<locals>.NoOverrideSubN�r* r+ r, r r r r � NoOverrideSub> s r0 r$ )r& �ndarrayr �viewr r r' )r r. r0 r � override_sub�no_override_subr r r r �test_ndarray_subclasses9 s �z/TestGetImplementingArgs.test_ndarray_subclassesc C s^ G dd� d�}t �d�}|� }t||g�}tt|�||g� t||g�}tt|�||g� d S )Nc @ s e Zd ZeZdS )zBTestGetImplementingArgs.test_ndarray_and_duck_array.<locals>.OtherNr) r r r r �OtherQ s r6 r$ r% )r r6 r �otherr r r r �test_ndarray_and_duck_arrayO s z3TestGetImplementingArgs.test_ndarray_and_duck_arrayc C sx G dd� dt j�}G dd� d�}t �d�}t �d��|�}|� }tt|||g�|||g� tt|||g�|||g� d S )Nc @ s e Zd ZeZdS )zQTestGetImplementingArgs.test_ndarray_subclass_and_duck_array.<locals>.OverrideSubNr) r r r r r. _ s r. c @ s e Zd ZeZdS )zKTestGetImplementingArgs.test_ndarray_subclass_and_duck_array.<locals>.OtherNr) r r r r r6 b s r6 r$ )r& r1 r r2 r r )r r. r6 r Zsubarrayr7 r r r �$test_ndarray_subclass_and_duck_array] s ��z<TestGetImplementingArgs.test_ndarray_subclass_and_duck_arrayc C s G dd� d�}G dd� d|�}G dd� d|�}G dd� d�}|� }|� }|� }|� }t td g�g � t t|g�|g� t t|d g�|g� t t|||g�|g� t t|||g�||g� t t||g�||g� t t||g�||g� t t|||g�|||g� t t|||g�|||g� d S ) Nc @ s e Zd ZeZdS )z8TestGetImplementingArgs.test_many_duck_arrays.<locals>.ANr) r r r r �Ap s r: c @ s e Zd ZeZdS )z8TestGetImplementingArgs.test_many_duck_arrays.<locals>.BNr) r r r r �Bs s r; c @ s e Zd ZeZdS )z8TestGetImplementingArgs.test_many_duck_arrays.<locals>.CNr) r r r r �Cv s r< c @ s e Zd ZeZdS )z8TestGetImplementingArgs.test_many_duck_arrays.<locals>.DNr) r r r r �Dy s r= r$ )r r ) r r: r; r< r= �a�b�c�dr r r �test_many_duck_arraysn s" z-TestGetImplementingArgs.test_many_duck_arraysc s� t td�� � fdd�td�D �}dd� |D �}t|d d� �}t||d d� � ttd�� t|� W d � n1 sx0 Y d S )N)r- c s" g | ]}t d t|� tf� ��qS )r: )�type�str�object)�.0�i�� namespacer r � <listcomp>� r zETestGetImplementingArgs.test_too_many_duck_arrays.<locals>.<listcomp>�! c S s g | ] }|� �qS r r �rF �tr r r rJ � r � zdistinct argument types)�dictr �ranger r r � TypeError)r �typesZ relevant_args�actualr rH r �test_too_many_duck_arrays� s z1TestGetImplementingArgs.test_too_many_duck_arraysN) r* r+ r, r( r5 r8 r9 rB rT r r r r r# ( s r# c @ s e Zd Zedd� �Zdd� ZdS )�TestNDArrayArrayFunctionc C sx G dd� d�}G dd� dt j�}G dd� dt j�}t �dg�}|� }|�|�}|�|�}|jtt jf|dfi d �}t|d � |jtt j|f||fi d �}t|tu � |jtt j|f||fi d �}t|d � |jtt j|f||fi d �}t|d � t t d �� t � ||f� W d � n1 �s 0 Y t � ||f�} t � ||f�}t|| �|�� t � ||f�}t|| �|�� d S ) Nc @ s e Zd ZeZdS )z3TestNDArrayArrayFunction.test_method.<locals>.OtherNr) r r r r r6 � s r6 c @ s e Zd ZdS )z;TestNDArrayArrayFunction.test_method.<locals>.NoOverrideSubNr/ r r r r r0 � s r0 c @ s e Zd ZeZdS )z9TestNDArrayArrayFunction.test_method.<locals>.OverrideSubNr) r r r r r. � s r. r$ g �?��funcrR r r r �no implementation found) r& r1 r r2 r- r" r r r r rQ Z concatenate) r r6 r0 r. r r7 r4 r3 �resultZexpectedr r r � test_method� sH � � � � .z$TestNDArrayArrayFunction.test_methodc C sX t �d�}dd� }ttd��( |j|t jf|fi d� W d � n1 sJ0 Y d S )Nr$ c S s | S r r ��xr r r r � r z:TestNDArrayArrayFunction.test_no_wrapper.<locals>.<lambda>Z_implementationrV )r& r r �AttributeErrorr- r1 )r r rW r r r �test_no_wrapper� s �z(TestNDArrayArrayFunction.test_no_wrapperN)r* r+ r, �requires_array_functionrZ r^ r r r r rU � s /rU c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestArrayFunctionDispatchc C s: t dtjd �D ]$}t�tjt|d��}t|tu � qd S )N� r$ )Zprotocol)rP r ZHIGHEST_PROTOCOL�loads�dumpsr r )r �protoZ roundtrippedr r r � test_pickle� s �z%TestArrayFunctionDispatch.test_picklec C s( t tjd� tjjdk r$t tjd� d S )Nr ra r )r r r* �sys�flags�optimize�__doc__�r r r r �test_name_and_docstring� s z1TestArrayFunctionDispatch.test_name_and_docstringc C sj G dd� d�}|� }t |�\}}}}}t||u � t|t u � tt|�|h� t||fk� t|i � d S )Nc @ s e Zd Zdd� ZdS )z9TestArrayFunctionDispatch.test_interface.<locals>.MyArrayc S s | ||||fS r r �r rW rR r r r r r r- � s zLTestArrayFunctionDispatch.test_interface.<locals>.MyArray.__array_function__N�r* r+ r, r- r r r r �MyArray� s rn )r r r �set)r rn r �objrW rR r r r r r �test_interface� s z(TestArrayFunctionDispatch.test_interfacec C sJ G dd� d�}|� }t td�� t|� W d � n1 s<0 Y d S )Nc @ s e Zd Zdd� ZdS )z?TestArrayFunctionDispatch.test_not_implemented.<locals>.MyArrayc S s t S r r rl r r r r- � s zRTestArrayFunctionDispatch.test_not_implemented.<locals>.MyArray.__array_function__Nrm r r r r rn � s rn rX )r rQ r �r rn r r r r �test_not_implemented� s z.TestArrayFunctionDispatch.test_not_implementedN)r* r+ r, re rk rq rs r r r r r` � s r` c @ s e Zd Zdd� Zdd� ZdS )� TestVerifyMatchingSignaturesc C s6 t dd� dd� � t d dd�ddd�� t ddd�ddd�� tt��" t dd� dd� � W d � n1 sn0 Y tt��$ t dd� ddd�� W d � n1 s�0 Y tt��& t dd d�dd d�� W d � n1 s�0 Y tt��& t dd d�dd d�� W d � n1 �s(0 Y d S )Nc S s dS �Nr r r[ r r r r r zNTestVerifyMatchingSignatures.test_verify_matching_signatures.<locals>.<lambda>c S s dS ru r r[ r r r r r r$ c S s dS ru r r[ r r r r r c S s dS ru r )r> r r r r r c S s dS ru r )r? r r r r r c S s dS ru r r[ r r r r r c S s dS ru r r[ r r r r r c S s dS ru r ��yr r r r r c S s dS ru r r[ r r r r r c S s dS ru r rv r r r r r )N)N)r$ )N)N)N)N)r$ )r$ )r r � RuntimeErrorrj r r r � test_verify_matching_signatures� s 0 2 4 z<TestVerifyMatchingSignatures.test_verify_matching_signaturesc C sX t t��$ tdd� �dd� �}W d � n1 s20 Y tdd� dd�dd� �}d S ) Nc S s | fS r r r[ r r r r r zKTestVerifyMatchingSignatures.test_array_function_dispatch.<locals>.<lambda>c S s d S r r rv r r r �f s zDTestVerifyMatchingSignatures.test_array_function_dispatch.<locals>.fc S s | fS r r r[ r r r r r F)Zverifyc S s d S r r rv r r r rz s )r rx r )r rz r r r � test_array_function_dispatch s (z9TestVerifyMatchingSignatures.test_array_function_dispatchN)r* r+ r, ry r{ r r r r rt � s rt c s, i � G � �fdd�d��� fdd�} �| fS )z2Create a duck array type and implements functions.c s e Zd Z� �fdd�ZdS )z._new_duck_type_and_implements.<locals>.MyArrayc s8 |� vr t S t�fdd�|D ��s&t S � | |i |��S )Nc 3 s | ]}t |� �V qd S r )� issubclassrL )rn r r � <genexpr># r zT_new_duck_type_and_implements.<locals>.MyArray.__array_function__.<locals>.<genexpr>)r �allrl ��HANDLED_FUNCTIONSrn r r r- s zA_new_duck_type_and_implements.<locals>.MyArray.__array_function__Nrm r r r r rn s rn c s �� fdd�}|S )z/Register an __array_function__ implementations.c s | � �< | S r r )rW )r� �numpy_functionr r � decorator) s zD_new_duck_type_and_implements.<locals>.implements.<locals>.decoratorr )r� r� )r� )r� r � implements' s z1_new_duck_type_and_implements.<locals>.implementsr )r� r r r � _new_duck_type_and_implements s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� TestArrayFunctionImplementationc C s< t � \}}|t�dd� �}ttd�d� tt|� �d� d S )Nc S s dS )N�myarrayr r r r r �_7 s z7TestArrayFunctionImplementation.test_one_arg.<locals>._r$ r r� )r� r r �r rn r� r� r r r � test_one_arg4 s z,TestArrayFunctionImplementation.test_one_argc C s� t � \}}tddd��ddd��}||�ddd��}t|d �d� t|d d d �d � t||� �d� tt�� ||� d d � W d � n1 s�0 Y ||� d d �}t|d � tt�� ||� dd � W d � n1 s�0 Y d S )Nc S s | fS r r �r �optionr r r r A r zDTestArrayFunctionImplementation.test_optional_args.<locals>.<lambda>�defaultc S s |S r r r� r r r �func_with_optionA s zLTestArrayFunctionImplementation.test_optional_args.<locals>.func_with_optionr� c S s |S r r )r � new_optionr r r �my_array_func_with_optionE s zUTestArrayFunctionImplementation.test_optional_args.<locals>.my_array_func_with_optionr$ Zextra)r� �yes)r� �no)N)r� )r� )r� r r r rQ )r rn r� r� r� rY r r r �test_optional_args> s , z2TestArrayFunctionImplementation.test_optional_argsc C s� t � \}}tdd� dd�dd� �}t�d�}t||�|u � t|jd� ttd�� ||� � W d � n1 sr0 Y d S ) Nc S s | fS r r r r r r r Z r zFTestArrayFunctionImplementation.test_not_implemented.<locals>.<lambda>Zmy)�modulec S s | S r r r r r r rW Z s zBTestArrayFunctionImplementation.test_not_implemented.<locals>.funcr$ z%no implementation found for 'my.func') r� r r& r r r r+ r rQ )r rn r� rW r r r r rs W s �z4TestArrayFunctionImplementation.test_not_implementedN)r* r+ r, r� r� rs r r r r r� 1 s r� c @ s e Zd Zdd� ZdS )�TestNDArrayMethodsc C sB G dd� dt j�}t �d��|�}tt|�d� tt|�d� d S )Nc @ s e Zd Zdd� ZdS )z-TestNDArrayMethods.test_repr.<locals>.MyArrayc _ s t S r r �r r r r r r- n s z@TestNDArrayMethods.test_repr.<locals>.MyArray.__array_function__Nrm r r r r rn m s rn r$ z MyArray(1)�1)r& r1 r r2 r �reprrD rr r r r � test_repri s z TestNDArrayMethods.test_reprN)r* r+ r, r� r r r r r� g s r� c @ s@ e Zd Zdd� Zdd� Zedd� �Zedd� �Zed d � �Zd S ) �TestNumPyFunctionsc C sB t tjjd� t tjjjd� t tjjjd� t tjjjd� d S )N�numpyz numpy.charz numpy.fftz numpy.linalg) r r& �sumr+ �charZequalZfftZlinalgZsolverj r r r �test_set_modulex s z"TestNumPyFunctions.test_set_modulec C s t �tj�}td|jv � d S )N�axis)�inspect� signaturer& r� r � parameters)r r� r r r �test_inspect_sum~ s z#TestNumPyFunctions.test_inspect_sumc C s2 t � \}}|tj�dd� �}tt�|� �d� d S )Nc S s dS )Nr� r r r r r r� � s z/TestNumPyFunctions.test_override_sum.<locals>._r� )r� r& r� r r� r r r �test_override_sum� s z$TestNumPyFunctions.test_override_sumc C sf G dd� d�}|t j|d��}d|jj_t�|�}t|d� |jj�tj|f|fi � |jj � � d S )Nc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )z=TestNumPyFunctions.test_sum_on_mock_array.<locals>.ArrayProxyc S s || _ d S r )�value)r r� r r r �__init__� s zFTestNumPyFunctions.test_sum_on_mock_array.<locals>.ArrayProxy.__init__c _ s | j j|i |��S r )r� r- r r r r r- � s zPTestNumPyFunctions.test_sum_on_mock_array.<locals>.ArrayProxy.__array_function__c _ s | j j|i |��S r )r� � __array__r r r r r� � s zGTestNumPyFunctions.test_sum_on_mock_array.<locals>.ArrayProxy.__array__N)r* r+ r, r� r- r� r r r r � ArrayProxy� s r� )�specr$ ) r ZMockr� r- Z return_valuer& r� r Zassert_called_once_withr� Zassert_not_called)r r� �proxyrY r r r �test_sum_on_mock_array� s �z)TestNumPyFunctions.test_sum_on_mock_arrayc C s6 G dd� dt j�}t �d��|�}tt �|�d� d S )Nc s$ e Zd Zdd� Z� fdd�Z� ZS )zFTestNumPyFunctions.test_sum_forwarding_implementation.<locals>.MyArrayc S s dS )N�summedr )r r� �outr r r r� � s zJTestNumPyFunctions.test_sum_forwarding_implementation.<locals>.MyArray.sumc s t � �||||�S r )�superr- rl �� __class__r r r- � s zYTestNumPyFunctions.test_sum_forwarding_implementation.<locals>.MyArray.__array_function__)r* r+ r, r� r- � __classcell__r r r� r rn � s rn r$ r� )r& r1 r r2 r r� rr r r r �"test_sum_forwarding_implementation� s z5TestNumPyFunctions.test_sum_forwarding_implementationN) r* r+ r, r� r� r_ r� r� r� r r r r r� v s r� c @ s� e Zd Zdd� Zd<dd�Zdd� Zedd � �Zd ged ��R d ged ��R d ged ��R dged��R dged��R dgee� d�� dd�fddgd��R dged ��R dged d��R dged ��R dged ��R d ged��R d ged e d ��R d gee d�e d ��R d!ged"e d#d$��R d%ged&d'� ��R d(ged)d'� d*d+gd#d,��R gZ ej�d-e �ej�d.d/dg�ed0d1� ���Zej�d-e �ej�d2d3d3gd4g�ed5d6� ���Zej�d.d/dg�d7d8� �Zed9d:� �Zd;S )=� TestArrayLikec C s, G dd� d�}|| _ G dd� d�}|| _d S )Nc @ s e Zd Zddd�Zdd� ZdS )z+TestArrayLike.setup_method.<locals>.MyArrayNc S s || _ d S r ��function�r r� r r r r� � s z4TestArrayLike.setup_method.<locals>.MyArray.__init__c S sJ |t t|j�u sJ �zt | |j�}W n ty: t Y S 0 ||i |��S r )�getattrr& r* r] r )r rW rR r r �my_funcr r r r- � s z>TestArrayLike.setup_method.<locals>.MyArray.__array_function__)N)r* r+ r, r� r- r r r r rn � s rn c @ s e Zd Zddd�ZdS )z:TestArrayLike.setup_method.<locals>.MyNoArrayFunctionArrayNc S s || _ d S r r� r� r r r r� � s zCTestArrayLike.setup_method.<locals>.MyNoArrayFunctionArray.__init__)N)r* r+ r, r� r r r r �MyNoArrayFunctionArray� s r� )rn r� )r rn r� r r r � setup_method� s zTestArrayLike.setup_methodFc s � ��fdd�}t � �|� d S )Nc s* d|vs J ��r d|v r t �� t� ���S )N�like� value_error)� ValueErrorr� r� �� arr_class�enable_value_error�namer r � _definition� s z-TestArrayLike.add_method.<locals>._definition)�setattr)r r� r� r� r� r r� r � add_method� s zTestArrayLike.add_methodc O s | |fS r r r� r r r � func_args� s zTestArrayLike.func_argsc C sT | � d| j� | j�� }ttd�� tjd|d�}W d � n1 sF0 Y d S )Nr rX r$ �r� )r� rn r r rQ r& �asarray)r �ref� array_liker r r � test_array_like_not_implemented� s z-TestArrayLike.test_array_like_not_implementedr )r$ r� Z asanyarrayZascontiguousarray)ra � ZasfortranarrayZrequire� ra r� r: �F)Z requirements�emptyZfullZonesZzeros�arangeZ frombuffers )�dtypeZfromiterZ fromstringz1,2�,)r� �sepZloadtxtc C s t d�S )Nz0 1 2 3r r r r r r � r zTestArrayLike.<lambda>Z genfromtxtc C s t d�S )Nz1,2.1r r r r r r � r )�intZi8)�floatZf8)r� Z delimiterzfunction, args, kwargs� numpy_refTc C s� | � d| j� | � || j� tt|�}t| j|�}|du rFt�d�}n | j�� }tdd� |D ��}||i |�d|i��} |du r�t| �tju s�J �tdd� |D ��} || i |��} |dkr�| �d� | �d� t | | � n t| �| ju s�J �| j |u s�J �d S ) Nr Tr$ c s s | ]}t |�r|� n|V qd S r ��callable�rF r> r r r r} r z0TestArrayLike.test_array_like.<locals>.<genexpr>r� c s s | ]}t |�r|� n|V qd S r r� r� r r r r} r r� ) r� rn r� r& r �tuplerC r1 �fillr r� ) r r� r r r� �np_funcr� r� � like_argsr� Znp_argsZnp_arrr r r �test_array_like� s& z TestArrayLike.test_array_liker� r$ r� c C s� | � d| j� | � || j� tt|�}|dkr8| j�� }tdd� |D ��}ttd��& ||i |�d|i�� W d � n1 s�0 Y d S )Nr r� c s s | ]}t |�r|� n|V qd S r r� r� r r r r} ! r z<TestArrayLike.test_no_array_function_like.<locals>.<genexpr>z9The `like` argument must be an array-like that implementsr� )r� r� r� r& r r� r rQ )r r� r r r� r� r� r r r �test_no_array_function_like s �z)TestArrayLike.test_no_array_function_likec C s� | � d| j� | � d| j� |du r0t�d�}n | j�� }tj�d�}t�� ��}tj� |d�}|� |� tj ||d�}|du r�t |�tj u s�J �tj ||d�}t||� t||� n$t |�| ju s�J �|j| jj u s�J �W d � n1 s�0 Y d S )Nr �fromfileTr$ � Ztestfiler� )r� rn r& r Zrandom�tempfileZTemporaryDirectory�os�path�joinZtofiler� rC r1 r r� )r r� r� �dataZtmpdir�fnamer� Znp_resr r r �test_array_like_fromfile' s" z&TestArrayLike.test_array_like_fromfilec C sX | j d| jdd� | j�� }tt�� tjdd|d� W d � n1 sJ0 Y d S )Nr T)r� r$ )r� r� )r� rn r r rQ r& )r r� r r r �test_exception_handlingA s z%TestArrayLike.test_exception_handlingN)F)r* r+ r, r� r� r� r_ r� r& r� Zreshaper� rP Z _array_tests�pytest�markZ parametrizer� r� r� r� r r r r r� � sN � �� r� )&r� rf r� r� �ior Zunittestr r� r& Z numpy.testingr r r r Znumpy.core.overridesr r r r Z numpy.compatr r� r� Zskipifr_ r r r" r# rU r` rt r� r� r� r� r� r r r r �<module> s> � o=' 5;
| ver. 1.6 |
Github
|
.
| PHP 8.2.30 | ??????????? ?????????: 0 |
proxy
|
phpinfo
|
???????????