OwlCyberSecurity - MANAGER
Edit File: test_overrides.cpython-38.pyc
U �p�]9 � @ sF d dl mZmZmZ d dlZd dlZd 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� de �Z G dd� de �Z eG dd� de ��Z eG dd� de ��Z!dd� Z"eG d d � d e ��Z#G d d � d e �Z$G d d!� d!e �Z%dS )"� )�division�absolute_import�print_functionN)�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 �A/usr/lib/python3/dist-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_subclasses8 s �z/TestGetImplementingArgs.test_ndarray_subclassesc C s` G dd� dt �}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 �OtherP s r6 r% )�objectr& r r r r' )r r6 r �otherr r r r �test_ndarray_and_duck_arrayN s z3TestGetImplementingArgs.test_ndarray_and_duck_arrayc C sz G dd� dt j�}G dd� dt�}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 a s r6 r% )r& r1 r7 r r2 r r )r r. r6 r Zsubarrayr8 r r r �$test_ndarray_subclass_and_duck_array\ s ��z<TestGetImplementingArgs.test_ndarray_subclass_and_duck_arrayc C s G dd� dt �}G dd� d|�}G dd� d|�}G dd� dt �}|� }|� }|� }|� }ttd g�g � tt|g�|g� tt|d g�|g� tt|||g�|g� tt|||g�||g� tt||g�||g� tt||g�||g� tt|||g�|||g� tt|||g�|||g� d S ) Nc @ s e Zd ZeZdS )z8TestGetImplementingArgs.test_many_duck_arrays.<locals>.ANr) r r r r �Ao s r; c @ s e Zd ZeZdS )z8TestGetImplementingArgs.test_many_duck_arrays.<locals>.BNr) r r r r �Br s r< c @ s e Zd ZeZdS )z8TestGetImplementingArgs.test_many_duck_arrays.<locals>.CNr) r r r r �Cu s r= c @ s e Zd ZeZdS )z8TestGetImplementingArgs.test_many_duck_arrays.<locals>.DNr) r r r r �Dx s r> r% )r7 r r ) r r; r< r= r> �a�b�c�dr r r �test_many_duck_arraysm s" z-TestGetImplementingArgs.test_many_duck_arraysc sr t td�� � fdd�td�D �}dd� |D �}t|d d� �}t||d d� � ttd�� t|� W 5 Q R X d S )N)r- c s" g | ]}t d t|� tf� ��qS )r; )�type�strr7 )�.0�i�� namespacer r � <listcomp>� s zETestGetImplementingArgs.test_too_many_duck_arrays.<locals>.<listcomp>�! c S s g | ] }|� �qS r r �rF �tr r r rJ � s � zdistinct argument types)�dictr �ranger r r � TypeError)r �typesZ relevant_argsZactualr rH r �test_too_many_duck_arrays� s z1TestGetImplementingArgs.test_too_many_duck_arraysN) r* r+ r, r( r5 r9 r: rC rS r r r r r$ ' s r$ c @ s e Zd Zedd� �Zdd� ZdS )�TestNDArrayArrayFunctionc C sd G dd� dt �}G dd� dtj�}G dd� dtj�}t�dg�}|� }|�|�}|�|�}|jttjf|dfi d �}t|d � |jttj|f||fi d �}t|t k� |jttj|f||fi d �}t|d � |jttj|f||fi d �}t|d � t t d �� t� ||f� W 5 Q R X 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) r7 r& r1 r r2 r- r# r r r r rQ Z concatenate) r r6 r0 r. r r8 r4 r3 �resultZexpectedr r r � test_method� sL � � � � z$TestNDArrayArrayFunction.test_methodc C sD t �d�}dd� }ttd�� |j|t jf|fi d� W 5 Q R X d S )Nr% c S s | S r r ��xr r r r � r z:TestNDArrayArrayFunction.test_no_wrapper.<locals>.<lambda>Z_implementationrU )r& r r �AttributeErrorr- r1 )r r rV r r r �test_no_wrapper� s �z(TestNDArrayArrayFunction.test_no_wrapperN)r* r+ r, �requires_array_functionrY r] r r r r rT � s /rT c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestArrayFunctionDispatchc C s: t dtjd �D ]$}t�tjt|d��}t|tk� 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! r` r )r r! r* �sys�flags�optimize�__doc__�r r r r �test_name_and_docstring� s z1TestArrayFunctionDispatch.test_name_and_docstringc C sl G dd� dt �}|� }t|�\}}}}}t||k� t|tk� 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 rV 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 rm )r7 r! r r �set)r rm r �objrV rR r r r r r �test_interface� s z(TestArrayFunctionDispatch.test_interfacec C s8 G dd� dt �}|� }ttd�� t|� W 5 Q R X d S )Nc @ s e Zd Zdd� ZdS )z?TestArrayFunctionDispatch.test_not_implemented.<locals>.MyArrayc S s t S r r rk r r r r- � s zRTestArrayFunctionDispatch.test_not_implemented.<locals>.MyArray.__array_function__Nrl r r r r rm � s rm rW )r7 r rQ r! �r rm r r r r �test_not_implemented� s z.TestArrayFunctionDispatch.test_not_implementedN)r* r+ r, rd rj rp rr r r r r r_ � s r_ c @ s e Zd Zdd� Zdd� ZdS )� TestVerifyMatchingSignaturesc C s� t dd� dd� � t d dd�ddd�� t ddd�ddd�� tt�� t dd� dd� � W 5 Q R X tt�� t dd� ddd�� W 5 Q R X tt�� t dd d�dd d�� W 5 Q R X tt�� t dd d�dd d�� W 5 Q R X d S )Nc S s dS �Nr r rZ r r r r r zNTestVerifyMatchingSignatures.test_verify_matching_signatures.<locals>.<lambda>c S s dS rt r rZ r r r r r r% c S s dS rt r rZ r r r r r c S s dS rt r )r? r r r r r c S s dS rt r )r@ r r r r r c S s dS rt r rZ r r r r r c S s dS rt r rZ r r r r r c S s dS rt r ��yr r r r r c S s dS rt r rZ r r r r r c S s dS rt r ru r r r r r )N)N)r% )N)N)N)N)r% )r% )r r � RuntimeErrorri r r r � test_verify_matching_signatures� s z<TestVerifyMatchingSignatures.test_verify_matching_signaturesc C sD t t�� tdd� �dd� �}W 5 Q R X tdd� dd�dd� �}d S ) Nc S s | fS r r rZ r r r r r zKTestVerifyMatchingSignatures.test_array_function_dispatch.<locals>.<lambda>c S s d S r r ru r r r �f s zDTestVerifyMatchingSignatures.test_array_function_dispatch.<locals>.fc S s | fS r r rZ r r r r r F)Zverifyc S s d S r r ru r r r ry s )r rw r )r ry r r r � test_array_function_dispatch s z9TestVerifyMatchingSignatures.test_array_function_dispatchN)r* r+ r, rx rz r r r r rs � s rs c s. i � G � �fdd�dt ��� 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 s4 |� kr t S t�fdd�|D ��s&t S � | ||�S )Nc 3 s | ]}t |� �V qd S r )� issubclassrL )rm r r � <genexpr>" s zT_new_duck_type_and_implements.<locals>.MyArray.__array_function__.<locals>.<genexpr>)r �allrk ��HANDLED_FUNCTIONSrm r r r- s zA_new_duck_type_and_implements.<locals>.MyArray.__array_function__Nrl r r~ r r rm s rm c s �� fdd�}|S )z/Register an __array_function__ implementations.c s | � �< | S r r )rV )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>.implements)r7 )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 �_6 s z7TestArrayFunctionImplementation.test_one_arg.<locals>._r% r r� )r� r! r �r rm r� r� r r r � test_one_arg3 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 5 Q R X ||� d d �}t|d � tt�� ||� dd � W 5 Q R X d S )Nc S s | fS r r �r �optionr r r r @ r zDTestArrayFunctionImplementation.test_optional_args.<locals>.<lambda>�defaultc S s |S r r r� r r r �func_with_option@ s zLTestArrayFunctionImplementation.test_optional_args.<locals>.func_with_optionr� c S s |S r r )r � new_optionr r r �my_array_func_with_optionD s zUTestArrayFunctionImplementation.test_optional_args.<locals>.my_array_func_with_optionr% Zextra)r� �yes)r� Zno)N)r� )r� )r� r r r rQ )r rm r� r� r� rX r r r �test_optional_args= s z2TestArrayFunctionImplementation.test_optional_argsc C sl t � \}}tdd� dd�dd� �}t�d�}t||�|k� t|jd� ttd�� ||� � W 5 Q R X d S ) Nc S s | fS r r r r r r r Y r zFTestArrayFunctionImplementation.test_not_implemented.<locals>.<lambda>Zmy)�modulec S s | S r r r r r r rV Y s zBTestArrayFunctionImplementation.test_not_implemented.<locals>.funcr% z%no implementation found for 'my.func') r� r r& r r r r+ r rQ )r rm r� rV r r r r rr V s �z4TestArrayFunctionImplementation.test_not_implementedN)r* r+ r, r� r� rr r r r r r� 0 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- m s z@TestNDArrayMethods.test_repr.<locals>.MyArray.__array_function__Nrl r r r r rm l s rm r% z MyArray(1)�1)r& r1 r r2 r �reprrE rq r r r � test_reprh s z TestNDArrayMethods.test_reprN)r* r+ r, r� r r r r r� f 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+ �charZequalZfftZlinalgZsolveri r r r �test_set_modulew s z"TestNumPyFunctions.test_set_modulec C s t �tj�}td|jk� d S )N�axis)�inspect� signaturer& r� r Z 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||�S r )r� r- r r r r r- � s zPTestNumPyFunctions.test_sum_on_mock_array.<locals>.ArrayProxy.__array_function__c _ s | j j||�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� �proxyrX 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- rk �� __class__r r r- � s zYTestNumPyFunctions.test_sum_forwarding_implementation.<locals>.MyArray.__array_function__)r* r+ r, r� r- � __classcell__r r r� r rm � s rm r% r� )r& r1 r r2 r r� rq 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� u s r� )&Z __future__r r r r� re Zunittestr r� r& Z numpy.testingr r r r Znumpy.core.overridesr r r r Z numpy.compatr ZpytestZmarkZskipifr^ r r! r# r7 r$ rT r_ rs r� r� r� r� r r r r �<module> s8 � o=' 5