?????????? ????????? - ??????????????? - /usr/lib64/python3.9/site-packages/numpy/core/tests/__pycache__/test_indexing.cpython-39.pyc
???????
a z[yc�� � @ s& d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dl m Z d dl m Z m Z mZmZmZmZmZ G dd� d�ZG dd� d�ZG d d � d �ZG d d � d �ZG d d� d�ZG dd� d�ZG dd� d�ZG dd� d�ZG dd� d�ZG dd� d�Z G dd� d�Z G dd � d �Z G d d � d �Z dS ) � N)�array_indexing)�product)�assert_� assert_equal� assert_raises�assert_raises_regex�assert_array_equal� assert_warns� HAS_REFCOUNTc @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z d7d8� Z d9d:� Z d;d<� Z d=d>� Z!d?d@� Z"dAdB� Z#dCdD� Z$dEdF� Z%dGdH� Z&dIdJ� Z'dKdL� Z(e)j*�+dMdNdOe,�-dPg�g�e)j*�+dQdRdSg�e)j*�+dTdUdRg�dVdW� ���Z.dXdY� Z/dZd[� Z0d\S )]� TestIndexingc s� t �dggg�� tt� fdd�� tt� fdd�� tt� fdd�� tt� fdd�� tt� fdd�� tt� fdd�� tt� fd d�� tt� fd d�� tt� fd d�� tt� fd d�� tt� fd d�� tt� fdd�� tt� fdd�� tt� fdd�� tt� fdd�� tt� fdd�� tt� fdd�� tt� fdd�� tt� fdd�� tt� fdd�� tt� fdd�� tt� fdd�� d S )N� c s � d S �N� � r ��ar �D/usr/lib64/python3.9/site-packages/numpy/core/tests/test_indexing.py�<lambda> � z3TestIndexing.test_index_no_floats.<locals>.<lambda>c s � d S )N)r r r r r r r r r c s � d S )N)r r r r r r r r r c s � dd d �f S r r r r r r r r c s � d d �df S r r r r r r r r c s � d d �dd d �f S r r r r r r r r c s � dd d �d d �f S r r r r r r r r c s � d S )N)r r r r r r r r r r c s � d S )N)r r r r r r r r r r c s � d S )N)r r r r r r r r r r c s � d S �N�ffffff��r r r r r r r c s � d S )N)r r r r r r r r r c s � d S )N)r r r r r r r r ! r c s � dd d �f S r r r r r r r " r c s � d d �df S r r r r r r r # r c s � d d �dd d �f S r r r r r r r $ r c s � dd d �d d �f S r r r r r r r % r c s � d S )N)r r r r r r r r r & r c s � d S )N)r r r r r r r r r ' r c s � d S )N)r r r r r r r r r ( r c s � dd �df S r r r r r r r ) r c s � dd �dd d �f S r r r r r r r * r )�np�arrayr � IndexError��selfr r r �test_index_no_floats s. z!TestIndexing.test_index_no_floatsc s� t �dgg�� tt� fdd�� tt� fdd�� tt� fdd�� tt� fdd�� tt� fdd�� tt� fdd�� tt� fd d�� tt� fd d�� tt� fd d�� tt� fd d�� tt� fd d�� tt� fdd�� tt� fdd�� tt� fdd�� tt� fdd�� tt� fdd�� tt� fdd�� tt� fdd�� tt� fdd�� tt� fdd�� tt� fdd�� tt� fdd�� d S )Nr c s � dd � S r r r r r r r 0 r z5TestIndexing.test_slicing_no_floats.<locals>.<lambda>c s � dd �dd�f S )Nr r � r r r r r r 1 r c s � dd d�d d�f S �Nr r r r r r r r r 2 r c s � ddd�d d �f S )Nr � r r r r r r r 3 r c s � d d �dd �f S r r r r r r r 4 r c s � d d� S r r r r r r r 6 r c s � d d�dd�f S )Nr r � @r r r r r r 7 r c s � d dd�d d�f S r r r r r r r 8 r c s � d d�d d �f S r r r r r r r 9 r c s � d d �ddd�f S )Nr � @r r r r r r r : r c s � d d d� S )N� �?r r r r r r < r c s � dd �d dd�f S )Nr r r r r r r r r = r c s � dd d�d d�f S )Nr r! r r r r r r r > r c s � d d d�d d �f S )N� @r r r r r r ? r c s � d d �ddd�f S )Nr � r r r r r r r @ r c s � ddd� S )Nr" r r r r r r r r B r c s � dd d� S )Nr" r r r r r r r C r c s � dd �d dd�f S )Nr r r r r r r r D r c s � ddd�d d�f S )Nr" r r! r r r r r r r E r c s � ddd�d d �f S )Nr" r# r r r r r r F r c s � d d �ddd�f S )Ng�������?r! r r r r r r r G r c s � d d d� S r r r r r r r I r �r r r � TypeErrorr r r r �test_slicing_no_floats, s. z#TestIndexing.test_slicing_no_floatsc s& t �dggg�� tt� fdd�� d S )Nr c s � � � � � S �Nr r r r r r O r z;TestIndexing.test_index_no_array_to_index.<locals>.<lambda>r% r r r r � test_index_no_array_to_indexK s z)TestIndexing.test_index_no_array_to_indexc C s< t �g d��}t|d |t j � t|d j|jd � d S )N�r r � r )r r r Znewaxis�ndim�r r r r r �test_none_indexQ s z TestIndexing.test_none_indexc C sP t �g d��}t|d |� t|d j|u � t �d�}tt|d t j�� d S )Nr* r r )r r r r �base� isinstance�int_r- r r r �test_empty_tuple_indexW s z#TestIndexing.test_empty_tuple_indexc C sF t jddd�}t|d j|j� t|d |� tt|d �t j� d S )Nr ZV4��dtype.)r �zerosr r4 �type�ndarray)r �sr r r � test_void_scalar_empty_tuple_ s z)TestIndexing.test_void_scalar_empty_tuplec C s� t �d�}|�t j�}t �d�}t|| || � t �d�||< t|t �d�� t �d��dd�}t|| || � t �d�d d �d f ||< t|t �d�d d �d f jddd�� t �d��dd�}t|||f |||f � d S )Nr � r r �Zaxis� )r �arange�astypeZuintpr �reshape�repeat)r �indexZu_index�arrr r r � test_same_kind_index_castinge s &z)TestIndexing.test_same_kind_index_castingc C sz t �g d��}t|g g � t|g j|j� t jg t jd�}t|g g � t|g j|j� t �g �}tt|j|� d S )Nr* r3 )r r r r4 �intpr r � __getitem__�r r �br r r �test_empty_fancy_indexy s z#TestIndexing.test_empty_fancy_indexc C s� t �g d�g d�g d�g�}t|d |u� t|d |� t|d j|u � t|d |d � t|d |dd d �f � t|d |d d �df � t|d t �d �� t �d �}d |tf< t|d � d S ) Nr* �r$ r � �� � � .)r .r �.r )r .r r r )r r r r r/ �EllipsisrF r r r �test_ellipsis_index� s � z TestIndexing.test_ellipsis_indexc C s` t �g d�g d�g d�g�}t|d g d�� t|d g d�� tt|jd� tt|jd� d S )Nr* rI rK r ���i @l )r r r r r rE r- r r r �test_single_int_index� s �z"TestIndexing.test_single_int_indexc C sX t �g d�g d�g d�g�}t|t �d� |d � t|t �d� |d dd� � d S )Nr* rI rK TFr �r r r r- r r r �test_single_bool_index� s �z#TestIndexing.test_single_bool_indexc C sx t �d�}t �dg�}tt|j|� t �dgd �}tt|j|� t jdtd�}tt|j|� tt|jtd �|f� d S )N)r r$ r+ TFrJ �r$ r$ r3 ) r �onesr r r rE r5 �bool�slice)r rB rA r r r �test_boolean_shape_mismatch� s z(TestIndexing.test_boolean_shape_mismatchc C sJ t �g d�g�}t jdgtd�}t|| |� d||< t|g d�g� d S )N)r r r Tr3 r" )r" r" r" )r r rX r rF r r r � test_boolean_indexing_onedim� s z)TestIndexing.test_boolean_indexing_onedimc C sP t �d�}dd� }tt||g � tt||g d�� tt||d d� g d�� d S )Nr$ c S s || | dk< d S )NrR r )r �vr r r �f� s z>TestIndexing.test_boolean_assignment_value_mismatch.<locals>.fr* r )r r= r � ValueError)r r r] r r r �&test_boolean_assignment_value_mismatch� s z3TestIndexing.test_boolean_assignment_value_mismatchc C sZ t �d�}t jdtd�}d|d d�< t jdtd�||< t �d�}d|d d�< t||� d S )N�� r3 T�d r )r r5 rX rW �objectr )r rB �indxZexpectedr r r �!test_boolean_assignment_needs_api� s z.TestIndexing.test_boolean_assignment_needs_apic C s� t �g d�g d�g d�g�}t �g d�g d�g d�g�}t|| g d�� t||d g d�g� t||d ||d � d||< t|g d �g d �g d �g� d S ) Nr* rI rK �TFT)FTF)r r+ r rL rN r r r )r r r )r$ r rJ )r rM r rT rF r r r � test_boolean_indexing_twodim� s � � �z)TestIndexing.test_boolean_indexing_twodimc C sD t �g d��}g d�}t|| ddg� t|d |f ddgg� d S )Nr* re r r+ rT rF r r r �test_boolean_indexing_list� s z'TestIndexing.test_boolean_indexing_listc C s� t �d�}t jdt jd�d d d� }t �d�d d d� }|||< t|d d� t �d�}t �d��dd�d d d� }|||d d �f< t|d ddg� d S ) Nr r3 rR r �r r r: r r )r rW r5 rD r= r r? )r r rG �cr r r �,test_reverse_strides_and_subspace_bufferinit s z9TestIndexing.test_reverse_strides_and_subspace_bufferinitc C sX t �d�d d �d f }t �d�d d d� }t|| ||�d� � t �d��dd�}d S )Nr: rR �C� r )r r= r �copyr? )r r �ir r r �'test_reversed_strides_result_allocation s z4TestIndexing.test_reversed_strides_result_allocationc C sf t �dd�}t �dd�}t �d��ddd�j|ddg< t �d��ddd�j�� |ddg< t||� d S )N)r+ r$ r rR � r r$ r r )r Zfullr= r? �Trm r rF r r r �%test_uncontiguous_subspace_assignment s "z2TestIndexing.test_uncontiguous_subspace_assignmentc C s* t �d�}tt|jt �dg�fd � d S )N) r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r � )r rW r r rE r r- r r r �(test_too_many_fancy_indices_special_case) s z5TestIndexing.test_too_many_fancy_indices_special_casec C sJ t �d�}t|t �d� |t �d� � t|t �d� |t �d� � d S )Nr TF)r r r �bool_r- r r r �test_scalar_array_bool. s z#TestIndexing.test_scalar_array_boolc C sB t �d�}t||d u� t||d u� t||d d � u� d S )Nr r .)r r= r r- r r r � test_everything_returns_views; s z*TestIndexing.test_everything_returns_viewsc C sD t �d�}tt|jddgg d�f� tt|jddgg d�fd� d S )N�r r r r �r r r )r r5 r r rE � __setitem__r- r r r �test_broaderrors_indexingC s z&TestIndexing.test_broaderrors_indexingc C sz t �d�}t jdt jd�}d|d< tt|j|� tt|j|d� t jdt jd�}d|d< tt|j|� tt|j|d� d S )Nr rl r3 r: rR r � )r r5 rW rD r r rE rz )r r �indr r r � test_trivial_fancy_out_of_boundsH s z-TestIndexing.test_trivial_fancy_out_of_boundsc C st t �d�}t jdt jd��ddd�d d �d d �df }t|| |� d||< t �d�}d|d< d|d< t||� d S )NrJ r3 r r r+ r rR )r r= rD r? r )r r �idx�resr r r � test_trivial_fancy_not_possibleS s , z,TestIndexing.test_trivial_fancy_not_possiblec C s� G dd� dt j�}t �d�}|�� j|d�}|�d� ||g d�d d �f< t|dk�� � ||d d �g d�f< t|dk�� � |�d� ||d< t|dk�� � d S ) Nc @ s e Zd Zdd� ZdS )z7TestIndexing.test_nonbaseclass_values.<locals>.SubClassc S s | � d� d S )N�c )�fill�r �oldr r r �__array_finalize__d s zJTestIndexing.test_nonbaseclass_values.<locals>.SubClass.__array_finalize__N��__name__� __module__� __qualname__r� r r r r �SubClassc s r� rx )r6 r �r r r r+ r$ r .)r r7 r5 rm �viewr� r �all)r r� r r8 r r r �test_nonbaseclass_valuesb s z%TestIndexing.test_nonbaseclass_valuesc C s� t �d�}t jdt jd��dd�}t|�|g d�d d �f< t||� t|�|d d �g d�f< t||� t|�|d< t||� d S )Nrx r< r3 r r� .)r r5 r= �float64r? � memoryviewr )r r r8 r r r �test_array_like_valuesw s z#TestIndexing.test_array_like_valuesc C s| t jjg d�ddgd�}t jg d�td�}t|| jj� t �ddg�}t|| jj� t|d jj� t|d jj� d S ) N))ZNGC1001r| )ZNGC1002r" )ZNGC1003r" )�targetZS20)ZV_magz>f4r3 )FTTr r .)r Zrecr rX r �flags� writeable)r �dr} r r r �test_subclass_writeable� s �z$TestIndexing.test_subclass_writeablec C sR t �d�}t �d��dd�j}t|| jj� |�dd�}t||df jj� d S )Nr: r r rR r r )r r= r? rq r r� � f_contiguousrF r r r �test_memory_order� s z TestIndexing.test_memory_orderc C sB G dd� d�}|� }G dd� d�}t �d�}tt|d t j�� t �d�}tt|| t j�� t �d�}tt||t �d�f t j�� tt|||� f t j�� t �d�}t jdtd �}||d< tt|d t j�� t �|d g�}tt|| t j�� t �|d gg�}tt||t �d�f t j�� tt|||� f t j�� d S ) Nc @ s e Zd Zdd� ZdS )z2TestIndexing.test_scalar_return_type.<locals>.Zeroc S s dS �Nr r r r r r � __index__� s z<TestIndexing.test_scalar_return_type.<locals>.Zero.__index__N)r� r� r� r� r r r r �Zero� s r� c @ s e Zd Zdd� ZdS )z7TestIndexing.test_scalar_return_type.<locals>.ArrayLikec S s t �d�S r� �r r r r r r � __array__� s zATestIndexing.test_scalar_return_type.<locals>.ArrayLike.__array__N)r� r� r� r� r r r r � ArrayLike� s r� r r )r r r r3 )r r5 r r0 �float_r rb r7 )r r� �zr� r rG r r r �test_scalar_return_type� s& z$TestIndexing.test_scalar_return_typec C s� t �dg�}tr"t�t �t j��}d|t jdgt jd�< d|t jdgt jd�< tt |j t jdgt jd�d� tt |j t jdgt jd�d� tr�t t�t �t j��|� d S )Nr r r3 ) r r r �sys� getrefcountr4 rD Zuint8r r rz r )r r Zrefcountr r r �test_small_regressions� s ��z#TestIndexing.test_small_regressionsc C s< t jdt jd�td� dd� }|�t �d��}t jdt jd�td� dd� }|�t �d��}t jdd d�|d < t �|j�}t || |� |||< t �|jd ��t j�dt � d �j d � }|�t j �d |j� }t �|j�|d < t ||� t j � |� |||� t j �< ||d d k |d d d � ||d d k< d S ) N�@ r3 r r i����ZS8rp Zbbbbbbbb�S.r r )r r5 �int8�ordr� r4 r r= �sizer rD �itemsizer> Zint16)r r\ r� �xrG r r r �test_unaligned� s .zTestIndexing.test_unalignedc C sX t �d�}G dd� dt�}dgdgf}||�}t|| jdk� t||f jdk� d S )Nrx c @ s e Zd ZdS )z7TestIndexing.test_tuple_subclass.<locals>.TupleSubclassN�r� r� r� r r r r � TupleSubclass� s r� r �r )r rW �tupler �shape)r rB r� rA r r r �test_tuple_subclass� s z TestIndexing.test_tuple_subclassc C s^ G dd� d�}t �d�}t||� ||� f � t jdddgd�}t||� ||� f � d S )Nc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )zDTestIndexing.test_broken_sequence_not_nd_index.<locals>.SequenceLikec S s dS r� r r r r r r� � s zNTestIndexing.test_broken_sequence_not_nd_index.<locals>.SequenceLike.__index__c S s dS )Nr r r r r r �__len__� s zLTestIndexing.test_broken_sequence_not_nd_index.<locals>.SequenceLike.__len__c S s t d��d S )Nz Not possible)r )r �itemr r r rE � s zPTestIndexing.test_broken_sequence_not_nd_index.<locals>.SequenceLike.__getitem__N)r� r� r� r� r� rE r r r r � SequenceLike� s r� r: r� )�f1�i8)�f2r� r3 )r r= r r5 )r r� rB r r r �!test_broken_sequence_not_nd_index� s z.TestIndexing.test_broken_sequence_not_nd_indexc C s� t �d�}t �d�}t �d�d d �d d d f }t �|d�}t|| ||�� � t jdt jd�}t|||f ||�� |f � d S )Nr: )r: r )r: �7 r$ r$ r$ r3 )r rW r= Z broadcast_tor rm r5 rD )r r� Zx2r} Zzindr r r �!test_indexing_array_weird_strides s z.TestIndexing.test_indexing_array_weird_stridesc C sH t �d�}|d d d�d d d�f }td �g d�f}d||< t|d� d S )NrV rR �r r r r+ r: g $@)r r5 rY r )r ZarrorB Zslicesr r r �$test_indexing_array_negative_strides s z1TestIndexing.test_indexing_array_negative_stridesc C sH t jddd�}t �d�|d< t|d t jddd�� |d dksDJ �d S )N)r r ri r3 Zasdfgr �r r � s)r r5 Zstr_r r )r rB r r r �test_character_assignment s z&TestIndexing.test_character_assignmentrA TFr �numrs �( � original_ndimr c C s� t �d| �}t�t�� ||f| W d � n1 s<0 Y t�t�� d||f| < W d � n1 st0 Y d S )Nr� r" )r rW �pytest�raisesr )r rA r� r� rB r r r � test_too_many_advanced_indices( s , z+TestIndexing.test_too_many_advanced_indicesc s� ddl m} t�dg�}t�d|fgd �}t�d|fgd �}tjjdd��|�d d �df � tj�� ��fdd ��|d d ��� ��fd d �td�D �}|D ] }|� � q�� j|u s�J �d S )Nr )�ThreadPoolExecutor)� �f8r� r )�p rM )r� c s$ � j dt| �dtjd�}| | d S )Nr r� )r� r4 )Zintegers�lenr rD )rB rc )�rngr r �funcC s z<TestIndexing.test_structured_advanced_indexing.<locals>.funcrM )� max_workersc s g | ]}�� �� ��qS r )Zsubmit)�.0�_)rB r� �tper r � <listcomp>H r zBTestIndexing.test_structured_advanced_indexing.<locals>.<listcomp>r: ) �concurrent.futuresr� r r4 ZrandomZuniformr� Z default_rng�range�result)r r� ZdtZfuturesr] r )rB r� r� r� r �!test_structured_advanced_indexing6 s z.TestIndexing.test_structured_advanced_indexingc C sp t �d��d�}t|ddg t �|d |d g�� t|ddgddgf t �ddg�� tt|jtd �g� d S )Nr< rx r r rJ ) r r= r? r r r r rE rY r- r r r �test_nontuple_ndindexN s $$z"TestIndexing.test_nontuple_ndindexN)1r� r� r� r r' r) r. r2 r9 rC rH rQ rS rU rZ r[ r_ rd rf rg rj ro rr rt rv rw r{ r~ r� r� r� r� r� r� r� r� r� r� r� r� r� r� �mark� parametrizer r r� r� r� r r r r r s\ " � r c @ s e Zd Zdd� ZdS )�TestFieldIndexingc C s< t �ddg�}tt|d t j�� tt|dg t j�� d S )Nr )r r� r )r r5 r r0 r7 r- r r r r� V s z)TestFieldIndexing.test_scalar_return_typeN)r� r� r� r� r r r r r� U s r� c @ s~ e Zd Zdd� Zdd� Zdd� Zdd� Zej� d d d d ge d �fdd gd dfd d d gd d gfg�dd� �Z dd� Z dd� Z d S )�TestBroadcastedAssignmentsc C s |||< |S r( r )r r r} �valr r r �assign_ s z!TestBroadcastedAssignments.assignc C sr t �d�}t �d�|d< t �d�|g d�d d �f< t �d�|d d �ddgf< t �d�|dgdgdggddgf< d S )N)r+ r )r r+ r .ry r r r )r r5 rW r- r r r �test_prepending_onesc s z/TestBroadcastedAssignments.test_prepending_onesc C sx | j }tj}t�d�}tt|||d t�d�� tt|||g d�f t�d�� tt|||dgdggf t�d�� d S )Nr .�r r r* r r )r r r )r� r �s_r5 r r^ rW �r r� r� r r r r �test_prepend_not_onem s z/TestBroadcastedAssignments.test_prepend_not_onec C s� | j }tj}t�d�}tt|||d t�d�� tt|||d t�d�� tt|||d d �dgf t�d�� tt|||d d �dgf t�d�� tt|||dgd d �f t�d�� d S )N)r r .rh )r r r r� )r� r r� r5 r r^ r� r r r � test_simple_broadcasting_errorsw s $$z:TestBroadcastedAssignments.test_simple_broadcasting_errorsrA .r r Nr c C st t �d�}t �d�}t|| j��dd�}t�t��}|||< W d � n1 sR0 Y t|j�� |�spJ �d S )N�ra ra )r+ r$ r rJ rL � r� ) r r5 �strr� �replacer� r� r^ �value�endswith)r rA �valuesrB Z shape_str�er r r �*test_broadcast_error_reports_correct_shape� s &zETestBroadcastedAssignments.test_broadcast_error_reports_correct_shapec C sR t �d�}g d�|dgdgdggg d�f< t|d d�d d�f g d�k�� � d S )Nrx �r r+ r$ r r r ry r+ )r r5 r r� r- r r r �test_index_is_larger� s z/TestBroadcastedAssignments.test_index_is_largerc C sZ t �d�}t �d�d d �d f }t �d�d d d� }|||< t|d d d� |k�� � d S )Nr� ra rR )r r5 r= r r� )r r r\ rG r r r �test_broadcast_subspace� s z2TestBroadcastedAssignments.test_broadcast_subspace)r� r� r� r� r� r� r� r� r� r� rY r� r� r� r r r r r� ^ s � r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestSubclassesc C s. G dd� dt j�}t �d�}|�|�}|d d� }tt|�|u � t|j|u � t||d d� � |g d� }tt|�|u � t|j|u� tt|j�t ju � t||g d� � t|j|g d� � ||dk }tt|�|u � t|j|u� tt|j�t ju � t|||dk � t|j||dk � d S )Nc @ s e Zd ZdS )z+TestSubclasses.test_basic.<locals>.SubClassNr� r r r r r� � s r� r r+ ry r )r r7 r= r� r r6 r/ r )r r� r r8 Zs_slice�s_fancyZs_boolr r r � test_basic� s& zTestSubclasses.test_basicc C sJ G dd� dt j�}t �d�}|�|�}d|j_|g d� }t|jj� d S )Nc @ s e Zd ZdS )z8TestSubclasses.test_fancy_on_read_only.<locals>.SubClassNr� r r r r r� � s r� r Fry )r r7 r= r� r� r� r )r r� r r8 r� r r r �test_fancy_on_read_only� s z&TestSubclasses.test_fancy_on_read_onlyc C s� G dd� dt j�}t �d��|�}|d d� }t|j|� t|j|� |g d� }t|j|� t|j|� ||dk }t|j|� t|j|� d S )Nc @ s e Zd Zdd� ZdS )z=TestSubclasses.test_finalize_gets_full_info.<locals>.SubClassc S s t �| �| _|| _d S r( )r r �finalize_statusr� r� r r r r� � s zPTestSubclasses.test_finalize_gets_full_info.<locals>.SubClass.__array_finalize__Nr� r r r r r� � s r� r: r+ r� r )r r7 r= r� r r� r� )r r� r8 Znew_sr r r � test_finalize_gets_full_info� s z+TestSubclasses.test_finalize_gets_full_infoN)r� r� r� r� r� r� r r r r r� � s r� c @ s e Zd Zdd� ZdS )�TestFancyIndexingCastc C s� d}t �|��t�}d|d< t �|�}t �dg�||< t|d d� tt j|jdgdgft �dg�� t|d d� tt j|j|t �dg�� t|d d� d S ) N)rM �? Tr� r r y @ �?r y �?) r r5 r> rX r r r ZComplexWarningrz )r r� Z bool_indexZ zero_arrayr r r � test_boolean_index_cast_assign� s ��z4TestFancyIndexingCast.test_boolean_index_cast_assignN)r� r� r� r� r r r r r� � s r� c @ s e Zd Zdd� Zdd� ZdS )� TestFancyIndexingEquivalencec C sT t jdtd�}|�� }g d�|d d�< g d�|g d�< t||� t jdtd�d d d �f }g d�g|dgd d�f< t||d � |j}dgdgdgg|d d�dgf< t||d d �df � t jd td�}|�� }dgd gdgd ggg|d d�d f< dgd gdgd ggg|dgd f< t||� |�d �}dgd gdgd ggg|dgd f< t||� d S )Nr r3 )r �r r r+ r+ ry r r r� )r+ r$ r r r$ .�F)r r= rb rm r rq rW )r r rG rB Zcmp_arrr r r �test_object_assign� s& " z/TestFancyIndexingEquivalence.test_object_assignc C s� t �d�}|�� }t �g d��|d d�< t �g d��|g d�< t||� t �d�d d d �f }t �g d�g�|dgd d�f< t||d � d S )Nr )�2�-3�-1r+ )r� r r� )r r r r )r r= rm r r rF r r r �test_cast_equivalence s z2TestFancyIndexingEquivalence.test_cast_equivalenceN)r� r� r� r� r r r r r r� � s r� c @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dS )�TestMultiIndexingAutomateda: These tests use code to mimic the C-Code indexing for selection. NOTE: * This still lacks tests for complex item setting. * If you change behavior of indexing, you might want to modify these tests to try more combinations. * Behavior was written to match numpy version 1.8. (though a first version matched 1.7.) * Only tuple indices are supported by the mimicking code. (and tested as of writing this) * Error types should match most of the time as long as there is only one error. For multiple errors, what gets raised will usually not be the same one. They are *not* tested. Update 2016-11-30: It is probably not worth maintaining this test indefinitely and it can be dropped if maintenance becomes a burden. c C sH t �t �g d����dddd�| _t �d�| _dtdt �g d ��t �d d gd d gg�t �d d gd d ggg�t d dd �t ddd�t ddd�t d d d�t jdt j d�t �g d��t �d gdgdgg�t jddgddggt � d�� � d�t jd dgt j d�t jdgd td�t �ddg�g| _td ddgt �d g�dg| _t d d �dg| _d S )N)r+ r r rJ r+ r r rJ )r+ r r rJ �skipr )TFFTF���r ra r$ rR ������)r r r r3 )r r r rD � r r" )r r= �prodr? r �emptyrG rP r rY rD r4 Z newbyteorderr� r5 �int�complex_indices�simple_indices� fill_indicesr r r r � setup_method= s. " " ��z'TestMultiIndexingAutomated.setup_methodc C s� t |�}g }d}d}d}d}d}d} t|�D �]\} } | du r@q,t| tj�r~| jtkr~d}| jdkrht�|| j7 }|| j7 }q,| t u r�| du r�| } q,t�t| t �r�|d7 }q,t| tj�s�ztj | tj d�} W n t y� t�Y n0 | || < n$| jjdk�r| jjdk�rtd ��| jdk�r"d}|d7 }|d7 }q,|j| dk �rHt�|dk�rhd|v�rh|�� |fS | du�r�t dd�g|j| || | d �< t|�D �]8\} } t| t ��r�tj| �|j| �� } |�d | g� �q�nP| du �r,|�d tj dgtj d�g� |�|jd| � d |j| d� �}�q�t| tj��r | jtk�r | j|j| | | j � k�rft�z*tjt�| �|j| | | j � d d�} W n2 t�y� d}tj dg| �� tj d�} Y n0 | jdk�r|�|jd| � t�|j| | | j � �f |j| | j d� �}| } nt�n2| jdk�rR| |j| k�sN| |j| k �rRt�| jdk�r�| |j| k�s�| |j| k �r�t�t|�dk�r�|d d dk�r�| | k�r�|d �| � n|d7 }|�d| g� �q�|dk�r�|�s�|dd� }t t|j��}g }|� ddg� d}d}|D ]�} |d7 }| d dk�r�|d � | dd� � ||= |d8 }t||t| dd� � �D ]} |�| � |� | � �qx|t| �d 7 }�q |}|j || � }d} |D �]} | d dk�r�t| �dk�r�q�|j}|| | t| dd� � � }|�|jd| � t�|�� t!�f |j| t| dd� � d� �}tj"| dd� � }|j#dk�r�|�rvt�t$| dd� |�D ]@\}}|j#dk�r��q�t�%||k��s�t�%|| k ��r�t��q�t| dd� �t|�k�rJt�&|�dk�r0z tj| dd� |d d�}W n t�y, td��Y n0 ntj| dd� |dd�}nt �|j'|�(� | d�}z0|�|jd| � |j |j| d d� �}W n t �y� t�Y n0 | |j7 } �q�|j'| d | d�}| d7 } �q�||fS )a� Mimic multi dimensional indexing. Parameters ---------- arr : ndarray Array to be indexed. indices : tuple of index objects Returns ------- out : ndarray An array equivalent to the indexing operation (but always a copy). `arr[indices]` should be identical. no_copy : bool Whether the indexing operation requires a copy. If this is `True`, `np.may_share_memory(arr, arr[indices])` should be `True` (with some exceptions for scalars and possibly 0-d arrays). Notes ----- While the function may mostly match the errors of normal indexing this is generally not the case. Tr FNr r3 rG rn z;arrays used as indices must be of integer (or boolean) typer8 �nr� �raise)�moderR r] zinvalid index into 0-sized�wrapr; ))�list� enumerater0 r r7 r4 rX r, r rP rY r rD r^ �kindrm r= �indicesr� �appendr? Zravel_multi_indexZnonzero� Exception�sumr r� r� �insert�extend�removeZ transposer> r � broadcastr� �zip�anyr �take�ravel)r rB r Z in_indices�no_copyZ num_fancyZ fancy_dimZ error_unless_broadcast_to_emptyr, Z ellipsis_posrn rc �axZ flat_indxZ new_indicesZaxesZ fancy_axesZniZaiZ orig_shapeZ orig_slicer� Z_indx�_size�mir r r �_get_multi_indexZ s2 � & � �� " "�� � � "� ��� z+TestMultiIndexingAutomated._get_multi_indexc C s� z| � ||�\}}W nn ty� } zVtr2t�|�}tt|�|j|� tt|�|j|d� trlt |t�|�� W Y d}~dS d}~0 0 | � ||||� dS )a Check a multi index item getting and simple setting. Parameters ---------- arr : ndarray Array to be indexed, must be a reshaped arange. index : tuple of indexing objects Index being tested. r N� r& r r r� r� r r6 rE rz r �_compare_index_result�r rB rA � mimic_getr" r� Z prev_refcountr r r �_check_multi_indexI s z-TestMultiIndexingAutomated._check_multi_indexc C s� z| � ||f�\}}W nn ty� } zVtr4t�|�}tt|�|j|� tt|�|j|d� trnt |t�|�� W Y d}~dS d}~0 0 | � ||||� dS )aW Check a single index item getting and simple setting. Parameters ---------- arr : ndarray Array to be indexed, must be an arange. index : indexing object Index being tested. Must be a single index and not a tuple of indexing objects (see also `_check_multi_index`). r Nr' r) r r r �_check_single_indexa s z.TestMultiIndexingAutomated._check_single_indexc C s� |� � }|| }t||� |jdkrl|jdkrltt�||�|k� trl|r\tt � |�d� ntt � |�d� |� � }|d ||< |jdkr�dS |r�|jdkr�|d7 }t||� dS |j |� � d7 < t||� dS )z4Compare mimicked result to indexing result. r r+ r r` N) rm r r� r, r r Zmay_share_memoryr r r� r� Zflatr! )r rB rA r* r" Z indexed_arrrG r r r r( y s& z0TestMultiIndexingAutomated._compare_index_resultc C s� t �d�}t|t �d� d� d|t �d�< t|d� | �| jt j| jtd�f� | �| jt j| jtd�d f� | �| jt j| jtd�d f� d S )Nr Tr r3 rO )N.)r r r r+ r Z zeros_likerX r- r r r � test_boolean� s ���z'TestMultiIndexingAutomated.test_booleanc s2 t �� �� t �ddt� t �ddtj� dd� � dD ]b}| j| j| j| jg}| j||< t |� D ]6}t � fdd�|D ��}| � | j |� | � | j |� q^q4W d � n1 s�0 Y | � | j d� | � | j d� tt| j jd � tt| j jd d � tt| j jd d d gd d f� tt| j jd d d gd d fd � d S ) N�errorr� c S s t | t�o| dkS )Nr )r0 r� )r r r r �isskip� s z8TestMultiIndexingAutomated.test_multidim.<locals>.isskip)r r r+ c 3 s | ]}� |�s|V qd S r( r )r� rn �r/ r r � <genexpr>� r z;TestMultiIndexingAutomated.test_multidim.<locals>.<genexpr>)r r r r )r r r r r r r )�warnings�catch_warnings�filterwarnings�DeprecationWarningr ZVisibleDeprecationWarningr r r r r� r+ r rG r r rE rz )r Z simple_posZtocheckrA r r0 r � test_multidim� s& � 0 z(TestMultiIndexingAutomated.test_multidimc C s& t �d�}| jD ]}| �||� qd S )Nr: )r r= r r, )r r rA r r r �test_1d� s z"TestMultiIndexingAutomated.test_1dN) r� r� r� �__doc__r r& r+ r, r( r- r6 r7 r r r r r ' s p r c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd S ) �TestFloatNonIntegerArgumentz� These test that ``TypeError`` is raised when you try to use non-integers as arguments to for indexing and slicing e.g. ``a[0.0:5]`` and ``a[0.5]``, or other functions like ``array.reshape(1., -1)``. c C st t �dggg�}|t �dg� |ddg |d d �ddgf |d d �dd d �f |d d �d d �d d �f d S )Nr r r� r- r r r �test_valid_indexing� s z/TestFloatNonIntegerArgument.test_valid_indexingc C s| t �dggg�}|d d � |dd � |d d� |dd� |d d d� |dd d� |d dd� |ddd� d S )Nr r r r r� r- r r r �test_valid_slicing� s z.TestFloatNonIntegerArgument.test_valid_slicingc C sj t �dgg�}ttt j|d� ttt j|t �d�df� ttt j|dgd� ttt j|dgt �d�� d S )Nr )r" r" rR r" rR r )r r r r& r? r r� r- r r r � test_non_integer_argument_errors� s z<TestFloatNonIntegerArgument.test_non_integer_argument_errorsc C s4 dd� }t t|dgt�d�� |dgt�d�� d S )Nc S s | | S r( r )r rG r r r �mult� s zRTestFloatNonIntegerArgument.test_non_integer_sequence_multiplication.<locals>.multr r+ )r r& r r� r1 )r r= r r r �(test_non_integer_sequence_multiplication� s zDTestFloatNonIntegerArgument.test_non_integer_sequence_multiplicationc C sN t �d�}ttt j|d� ttt j|d� ttt j|d� ttt j|d� d S )N)r+ r+ r+ � �?)r? r )r g������@)g�������?g333333�?)r r5 r r& �min)r r� r r r � test_reduce_axis_float_index� s z8TestFloatNonIntegerArgument.test_reduce_axis_float_indexN) r� r� r� r8 r: r; r< r> rA r r r r r9 � s r9 c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestBooleanIndexingc C sz t �dggg�}ttt j|d� ttt j|t �d�df� tttjt �d�� tt tjt j � ttt j |dgdfd� d S )Nr )TrR TrR r F)�args) r r r r& r? ru �operatorrA r r5 ZTrue_r r- r r r � test_bool_as_int_argument_errors s z4TestBooleanIndexing.test_bool_as_int_argument_errorsc sP t �d�� � d jdk � dddgdddgdggf dk tt� fd d �� d S ) Nr� )FT.)r r r+ r$ Tr r r r� c s � dddgdf S )NFr r .r r r r r r r zETestBooleanIndexing.test_boolean_indexing_weirdness.<locals>.<lambda>)r rW r� r r r r r r � test_boolean_indexing_weirdness s "z3TestBooleanIndexing.test_boolean_indexing_weirdnessc s� t �d�� t �dgd g��ttd� �fdd�� t �dgd dg g��ttd� �fd d�� t �dgd g��ttd� �fd d�� t �d �� t �dgdggg��ttd � �fdd�� d S )N)r+ r+ FrN ztboolean index did not match indexed array along dimension 0; dimension is 3 but corresponding boolean dimension is 1c s � � S r( r r )r �idx1r r r # r zETestBooleanIndexing.test_boolean_indexing_fast_path.<locals>.<lambda>rM Tc s � � S r( r r )r �idx2r r r * r r: c s � � S r( r r )r �idx3r r r 1 r )r r r ztboolean index did not match indexed array along dimension 1; dimension is 1 but corresponding boolean dimension is 2c s � � S r( r r )r r r r r 9 r )r rW r r r r r )r r rG rH rI r � test_boolean_indexing_fast_path s, � � � �z3TestBooleanIndexing.test_boolean_indexing_fast_pathN)r� r� r� rE rF rJ r r r r rB s rB c @ s e Zd ZdZdd� ZdS )�TestArrayToIndexDeprecationz7Creating an index from array not 0-D is an error. c C sR t �dggg�}tttjt �dg�� ttt j||df� ttt j|dg|� d S )Nr rR r )r r r r& rD rA r? r r- r r r �test_array_to_index_error@ s z5TestArrayToIndexDeprecation.test_array_to_index_errorN)r� r� r� r8 rL r r r r rK < s rK c @ s e Zd ZdZdd� ZdS )�TestNonIntegerArrayLikez�Tests that array_likes only valid if can safely cast to integer. For instance, lists give IndexError when they cannot be safely cast to an integer. c C s> t �d�}tt|jddg� tt|jddgf� |�g � d S )Nr: r? g �?�1r� )r r= r r rE r- r r r r� P s z"TestNonIntegerArrayLike.test_basicN�r� r� r� r8 r� r r r r rM I s rM c @ s e Zd ZdZdd� ZdS )�TestMultipleEllipsisErrorz/An index can only have a single ellipsis. c sL t �d�� tt� fdd�� tt� jtfd f� tt� jtfd f� d S )Nr: c s � d S )N)..r r r r r r ` r z6TestMultipleEllipsisError.test_basic.<locals>.<lambda>r r+ )r r= r r rE rP r r r r r� ^ s z$TestMultipleEllipsisError.test_basicNrO r r r r rP Z s rP c @ s e Zd Zdd� Zdd� ZdS )�TestCApiAccessc C s� t �td�}tt|t�d�d� tt|t�d�d� tt|t�d�d� tt|t�d�d� tt|t�d�d� t�d�}t|d ||d�� |� dd �}t|d ||d �� d S ) Nr r r: r| ����r: r: r$ r r ���) � functools�partialr r r r rW r= r r? )r Z subscriptr r r r � test_getitemf s zTestCApiAccess.test_getitemc C s� t �td�}tt|t�d�d� tt|t�d�dd� tt|t�d�dd� tt|t�d�dd� tt|t�d�dd� tt|t�d�dd� t�d�}||dd� t |d dk� |� d d �}||dd� t |d ddg� d S ) Nr r: r r r| rR rS r$ r r rR ) rU rV r r r^ r rW r r= r r? r )r r� r r r r � test_setitemv s zTestCApiAccess.test_setitemN)r� r� r� rW rX r r r r rQ e s rQ ) r� r2 rU rD r� Znumpyr Z numpy.core._multiarray_testsr � itertoolsr Z numpy.testingr r r r r r r r r� r� r� r� r� r r9 rB rK rM rP rQ r r r r �<module> s8 $ H C@0 )78
| ver. 1.6 |
Github
|
.
| PHP 8.2.30 | ??????????? ?????????: 0 |
proxy
|
phpinfo
|
???????????