Edit File: test_indexing.cpython-37.pyc
B ��Fd^� � @ s� d dl mZmZmZ d dlZd dlZd dlZd dlZd dlZ d dl m Z d dl m Z d dlmZmZmZmZmZmZmZmZ y`dZeed�r�ye j�de jj j �ZW n e k r� Y nX edkr�e j�de jj j �Zd Z W n e k r� d Z Y nX G d d � d e�Z!G d d� de�Z"G dd� de�Z#G dd� de�Z$G dd� de�Z%G dd� de�Z&G dd� de�Z'G dd� de�Z(G dd � d e�Z)G d d � d e�Z*G d d � d e�Z+G d!d"� d"e�Z,G d#d$� d$e�Z-e.d%k�r�e� dS )&� )�division�absolute_import�print_functionN)�array_indexing)�product)�TestCase�run_module_suite�assert_� assert_equal� assert_raises�assert_array_equal� assert_warns� HAS_REFCOUNT�gettotalrefcountZ multiarray_d� multiarrayTFc @ 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#dCS )D� 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 )Ng � r )�ar ��/builddir/build/BUILDROOT/alt-python37-numpy-1.13.3-8.el8.x86_64/opt/alt/python37/lib64/python3.7/site-packages/numpy/core/tests/test_indexing.py�<lambda># � z3TestIndexing.test_index_no_floats.<locals>.<lambda>c s � d S )N)r g r r )r r r r $ r c s � d S )N)g r r r )r r r r % r c s � dd d �f S )Ng r r )r r r r & r c s � d d �df S )Ng r r )r r r r ' r c s � d d �dd d �f S )Ng r r )r r r r ( r c s � dd d �d d �f S )Ng r r )r r r r ) r c s � d S )N)r r g r r )r r r r * r c s � d S )N)g r r r r )r r r r + r c s � d S )N)r g r r r )r r r r , r c s � d S )Ngffffff��r r )r r r r - r c s � d S )N)r gffffff��r r )r r r r . r c s � d S )N)gffffff��r r r )r r r r / r c s � dd d �f S )Ngffffff��r r )r r r r 0 r c s � d d �df S )Ngffffff��r r )r r r r 1 r c s � d d �dd d �f S )Ngffffff��r r )r r r r 2 r c s � dd d �d d �f S )Ngffffff��r r )r r r r 3 r c s � d S )N)r r gffffff��r r )r r r r 4 r c s � d S )N)gffffff��r r r r )r r r r 5 r c s � d S )N)r gffffff��r r r )r r r r 6 r c s � dd �df S )Ng r r )r r r r 7 r c s � dd �dd d �f S )Ng r r )r r r r 8 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 )Ng r r )r r r r > r z5TestIndexing.test_slicing_no_floats.<locals>.<lambda>c s � dd �dd�f S )Nr g � r r )r r r r ? r c s � dd d�d d�f S )Ng r r r r )r r r r @ r c s � ddd�d d �f S )Ng � r r r )r r r r A r c s � d d �dd �f S )Ng r r )r r r r B r c s � d d� S )Ng r r )r r r r D r c s � d d�dd�f S )Nr r g @r r )r r r r E r c s � d dd�d d�f S )Ng r r r r )r r r r F r c s � d d�d d �f S )Ng r r )r r r r G r c s � d d �ddd�f S )Nr g @r r r )r r r r H r c s � d d d� S )Ng �?r r )r r r r J r c s � dd �d dd�f S )Nr r g @r r )r r r r K r c s � dd d�d d�f S )Nr g @r r r )r r r r L r c s � d d d�d d �f S )Ng @r r )r r r r M r c s � d d �ddd�f S )Nr � g @r r )r r r r N r c s � ddd� S )Ng �?r g @r r )r r r r P r c s � dd d� S )Ng �?g @r r )r r r r Q r c s � dd �d dd�f S )Nr g @r r )r r r r R r c s � ddd�d d�f S )Ng �?r g @r r r )r r r r S r c s � ddd�d d �f S )Ng �?g @r r )r r r r T r c s � d d �ddd�f S )Ng�������?g @g @r r )r r r r U r c s � d d d� S )Ng r r )r r r r W r )r r r � TypeError)r 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 ] r z;TestIndexing.test_index_no_array_to_index.<locals>.<lambda>)r r r r )r r )r r � test_index_no_array_to_indexY s z)TestIndexing.test_index_no_array_to_indexc C s> t �dddg�}t|d |t j � t|d j|jd � d S )Nr r � )r r r �newaxis�ndim)r r r r r �test_none_index_ s z TestIndexing.test_none_indexc C sR t �dddg�}t|d |� t|d j|k� t �d�}tt|d t j�� d S )Nr r r# r r )r r r r �base� isinstance�int_)r r r r r �test_empty_tuple_indexe s z#TestIndexing.test_empty_tuple_indexc 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 )�axis� )r �arange�astype�uintpr �reshape�repeat)r �indexZu_index�arrr r r � test_same_kind_index_castingm s &z)TestIndexing.test_same_kind_index_castingc C s| t �dddg�}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 r r# )�dtype)r r r r6 �intpr r � __getitem__)r r �br r r �test_empty_fancy_index� s z#TestIndexing.test_empty_fancy_indexc C s� t �dddgdddgddd gg�}t|d |k � t|d |� t|d j|k� t|d |d � t|d |d d d �f � t|d |d d �d f � 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' �Ellipsis)r r r9 r r r �test_ellipsis_index� s z TestIndexing.test_ellipsis_indexc C sj t �dddgdddgddd gg�}t|d dddg� t|d ddd g� tt|jd � tt|jd � d S )Nr r r# r r r; r< r= r> r ���i @l )r r r r r r8 )r r r r r �test_single_int_index� s z"TestIndexing.test_single_int_indexc C s^ t �dddgdddgddd gg�}t|t �d � |d � t|t �d � |d d d � � d S ) Nr r r# r r r; r< r= r> TFr )r 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# TFr; )r r )r6 ) r �onesr r r r8 �zeros�bool�slice)r r4 r3 r r r �test_boolean_shape_mismatch� s z(TestIndexing.test_boolean_shape_mismatchc C sN t �dddgg�}t jdgtd�}t|| |� d||< t|dddgg� d S )Ng T)r6 g �?)r r rF r )r r r9 r r r � test_boolean_indexing_onedim� s z)TestIndexing.test_boolean_indexing_onedimc C sT t �d�}dd� }tt||g � tt||dddg� tt||d d� dddg� d S )Nr c S s || | dk< d S )NrA r )r �vr r r �f� s z>TestIndexing.test_boolean_assignment_value_mismatch.<locals>.fr r r# )r r. r � ValueError)r r rK 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 )Ni� )r6 T�d r )r rE rF rD �objectr )r r4 �indx�expectedr r r �!test_boolean_assignment_needs_api� s z.TestIndexing.test_boolean_assignment_needs_apic C s� t �dddgdddgddd gg�}t �d d d gd d d gd d d gg�}t|| ddddd g� t||d dddgg� t||d ||d � d ||< t|d dd gdd dgd dd gg� d S ) Nr r r# r r r; r< r= r> TFr )r r r )r r r9 r r r � test_boolean_indexing_twodim� s z)TestIndexing.test_boolean_indexing_twodimc 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 )r6 rA r )r r r+ r r )r rD rE r7 r. r r1 )r r r9 �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+ rA �C� r )r r. r �copyr1 )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 rA � r r r r )r �fullr. r1 �TrX r )r r r9 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 rD r r r8 r )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 r �test_scalar_array_bool. s z#TestIndexing.test_scalar_array_boolc C sB t �d�}t||d k � t||d k � t||d d � k � d S )Nr r .)r r. r )r r r r r � test_everything_returns_views; s z*TestIndexing.test_everything_returns_viewsc C sH t �d�}tt|jddgdddgf� tt|jddgdddgfd� d S )N)r r r r r )r rE r r r8 � __setitem__)r 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 rW )r6 r+ rA r � )r rE rD r7 r r r8 rd )r r �indr r r � test_trivial_fancy_out_of_boundsH s z-TestIndexing.test_trivial_fancy_out_of_boundsc C s� G dd� dt j�}t �d�}|�� j|d�}|�d� ||ddddd gd d �f< t|dk�� � ||d d �ddddd gf< 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__U s zJTestIndexing.test_nonbaseclass_values.<locals>.SubClass.__array_finalize__N)�__name__� __module__� __qualname__rl r r r r �SubClassT s rp )r r )�typer r r r# r .)r �ndarrayrE rX �viewrj r �all)r rp r �sr r r �test_nonbaseclass_valuesS s z%TestIndexing.test_nonbaseclass_valuesc C s� t jjdddgddgd�}t jdddgtd�}t|| jj� t �d d g�}t|| jj� t|d jj� t|d jj� d S ) N)ZNGC1001rf )ZNGC1002g �?)ZNGC1003g �?)�targetZS20)ZV_magz>f4)r6 FTr r .)r �recr rF r �flags� writeable)r �drg r r r �test_subclass_writeableh 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 rA r r )r r. r1 r] r ry � f_contiguous)r r r9 r r r �test_memory_orderr s z TestIndexing.test_memory_orderc C sF G dd� dt �}|� }G dd� dt �}t�d�}tt|d tj�� t�d�}tt|| tj�� t�d�}tt||t�d�f tj�� tt|||� f tj�� t�d�}tjdt d �}||d< tt|d tj�� t�|d g�}tt|| tj�� t�|d gg�}tt||t�d�f tj�� tt|||� f tj�� 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)rm rn ro 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 )Nr )r r )r r r r � __array__� s zATestIndexing.test_scalar_return_type.<locals>.ArrayLike.__array__N)rm rn ro r� r r r r � ArrayLike� s r� r r )r r r )r6 )rO r rE r r( �float_r rr )r r� �zr� r r9 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 )r6 ) r r r �sys� getrefcountr6 r7 �uint8r r rd r )r r �refcountr 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�@ )r6 r r i����ZS8r[ Zbbbbbbbb�S.r r )r rE �int8�ordrs r6 r r. �sizer r7 �itemsizer/ �int16)r rJ r{ �xr9 r r r �test_unaligned� s .zTestIndexing.test_unalignedc C sX t �d�}G dd� dt�}dgdgf}||�}t|| jdk� t||f jdk� d S )N)r r c @ s e Zd ZdS )z7TestIndexing.test_tuple_subclass.<locals>.TupleSubclassN)rm rn ro r r r r � TupleSubclass� s r� r )r )r rD �tupler �shape)r r4 r� r3 r r r �test_tuple_subclass� s z TestIndexing.test_tuple_subclassc C s` G dd� dt �}t�d�}t||� ||� f � tjdddgd�}t||� ||� f � d S )Nc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )zDTestIndexing.test_broken_sequence_not_nd_index.<locals>.SequenceLikec S s dS )Nr 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 r8 � s zPTestIndexing.test_broken_sequence_not_nd_index.<locals>.SequenceLike.__getitem__N)rm rn ro r r� r8 r r r r � SequenceLike� s r� r+ )r )�f1�i8)�f2r� )r6 )rO r r. r rE )r r� r4 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 )r6 )r rD r. � broadcast_tor rX rE r7 )r r� �x2rg Zzindr r r �!test_indexing_array_weird_strides� s z.TestIndexing.test_indexing_array_weird_stridesc C sL t �d�}|d d d�d d d�f }td �ddddgg}d||< t|d� d S ) N)r r rA r r r r# r+ g $@)r rE rG r )r Zarror4 �slicesr r r �$test_indexing_array_negative_strides� s z1TestIndexing.test_indexing_array_negative_stridesN)$rm rn ro r r! r" r& r* r5 r: r@ rB rC rH rI rM rR rS rU rZ r^ r` rb rc re rh rv r| r~ r� r� r� r� r� r� r� r r r r r sB " r c @ s e Zd Zdd� ZdS )�TestFieldIndexingc C s< t �ddg�}tt|d t j�� tt|dg t j�� d S )Nr )r �f8r )r rE r r( rr )r r r r r r� s z)TestFieldIndexing.test_scalar_return_typeN)rm rn ro r� r r r r r� s r� c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )�TestBroadcastedAssignmentsc C s |||< |S )Nr )r r rg �valr r r �assign s z!TestBroadcastedAssignments.assignc C st t �d�}t �d�|d< t �d�|dddgd d �f< t �d�|d d �ddgf< t �d�|dgdgdggddgf< d S )N)r# r )r r# r .r r r )r rE rD )r r r r r �test_prepending_ones s z/TestBroadcastedAssignments.test_prepending_onesc C sz | j }tj}t�d�}tt|||d t�d�� tt|||dddgf t�d�� tt|||dgdggf t�d�� d S )Nr .)r r r r r# )r r r )r� r �s_rE r rL rD )r r� r� r r r r �test_prepend_not_one 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 .)r r )r r r )r r )r� r r� rE r rL )r r� r� r r r r � test_simple_broadcasting_errors$ s $$z:TestBroadcastedAssignments.test_simple_broadcasting_errorsc C sX t �d�}dddg|dgdgdggdddgf< t|d d�d d�f dddgk�� � d S )N)r r r r# r r r )r rE r rt )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 )N)rN rN rN rA )r rE r. r rt )r r rJ r9 r r r �test_broadcast_subspace6 s z2TestBroadcastedAssignments.test_broadcast_subspaceN) rm rn ro r� r� r� 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��|�}tt|d d� |�� t|d d� j|k� tt|dddg |�� tt||dk |�� d S )Nc @ s e Zd ZdS )z+TestSubclasses.test_basic.<locals>.SubClassN)rm rn ro r r r r rp @ s rp r r# r r r )r rr r. rs r r( r' )r rp ru r r r � test_basic? s zTestSubclasses.test_basicc C s~ t �ddgddgg�}tt|dddgd d �f t j�� t �t �d��dd��}t|d d�t �d �f |d d�d f j � d S ) Nr r r# r r �2 r r+ rA ) r �matrixr r( �asmatrixr. r1 r r r] )r �mr� r r r �test_matrix_fancyJ s "z TestSubclasses.test_matrix_fancyc C s� G dd� dt j�}t �d��|�}|d d� }t|j|� t|j|� |ddddg }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 )N)r r �finalize_statusrk )r rk r r r rl Z s zPTestSubclasses.test_finalize_gets_full_info.<locals>.SubClass.__array_finalize__N)rm rn ro rl r r r r rp Y s rp r+ r# r r r )r rr r. rs r r� rk )r rp ru Znew_sr r r � test_finalize_gets_full_infoW s z+TestSubclasses.test_finalize_gets_full_infoN)rm rn ro 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)r= �? T)r r r r y @ �?r y �?) r rE r/ rF r r r �ComplexWarningrd )r r� Z bool_indexZ zero_arrayr r r � test_boolean_index_cast_assignl s z4TestFancyIndexingCast.test_boolean_index_cast_assignN)rm rn ro r� r r r r r� k s r� c @ s e Zd Zdd� Zdd� ZdS )� TestFancyIndexingEquivalencec C s\ t jdtd�}|�� }dddg|d d�< dddg|dddg< t||� t jdtd�d d d �f }dddgg|dgd d�f< t||d � |j}dgdgdgg|d d�dgf< t||d d �df � t jdtd�}|�� }dgdgdgd ggg|d d�d f< dgdgdgd ggg|dgd f< t||� |�d �}dgdgdgd ggg|dgd f< t||� d S ) Nr )r6 r )r r r# r r )r# r r r .�F)r r. rO rX r r] rD )r r r9 r4 Zcmp_arrr r r �test_object_assign� s& " z/TestFancyIndexingEquivalence.test_object_assignc C s� t �d�}|�� }t �dddg�|d d�< t �dddg�|dddg< t||� t �d�d d d �f }t �dddgg�|dgd d�f< t||d � d S ) Nr �2z-3z-1r# r r r )r r. rX r r )r r r9 r r r �test_cast_equivalence� s z2TestFancyIndexingEquivalence.test_cast_equivalenceN)rm rn ro 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 sP t �t �ddddg���dddd�| _t �d�| _dtdt �dd d g�t �dd gd dgg�t �d d gd d ggg�t d dd �t ddd �t d dd�t d d d�t jdt j d�t �dddg�t �d gdgdgg�t jddgddggt � d�� � d�t jd dgt j d�t jdgd td�t �ddg�g| _td ddgt �dg�dg| _t d d �dg| _d S )Nr# r r r; )r# r r r; �skipr TF���r rN r rA ������)r r r )r6 r7 � g g �?)r r. �prodr1 r �emptyr9 r? r rG r7 r6 � newbyteorderr� rE �int�complex_indices�simple_indices� fill_indices)r r r r �setUp� s* & "z TestMultiIndexingAutomated.setUpc C s� t |�}g }d}d}d}d}d}d} x�t|�D ]�\} } | dkr@q.t| tj�r~| jtkr~d}| jdkrht�|| j7 }|| j7 }q.| t kr�| dkr�| } q.t�t| t �r�|d7 }q.t| tj�s�tj | tj d�} | || < n | jj dkr�| jj dkr�td ��| jdk�rd}|d7 }|d7 }q.W |j| dk �r*t�|dk�rJd|k�rJ|�� |fS | dk �rvt dd�g|j| || | d �< �x>t|�D �]0\} } t| t ��r�tj| �|j| �� } |�d | g� �q�nP| dk�r|�d tj dgtj d�g� |�|jd| � d |j| d� �}�q�t| tj��r�| jtk�r�| j|j| | | j � k�rLt�y*tjt�| �|j| | | j � d d�} W n* d}tj dg| �� tj d�} Y nX | jdk�r�|�|jd| � t�|j| | | j � �f |j| | j d� �}| } nt�n2| jdk�r0| |j| k�s,| |j| k �r0t�| jdk�rb| |j| k�s^| |j| k �rbt�t|�dk�r�|d d dk�r�| | k�r�|d �| � n|d7 }|�d| g� �q�W |dk�r�|�s�|dd� }t t|j��}g }|�ddg� d}d}x�|D ]�} |d7 }| d dk�rz|d �| dd� � ||= |d8 }x8t||t| dd� � �D ]} |�| � |� | � �q\W |t| �d 7 }�qW |}|j || � }d} �x�|D �]�} | d dk�r�t| �dk�rԐq�|j}|| | t| dd� � � }|�|jd| � t�|�� t �f |j| t| dd� � d� �}tj | dd� � }|j!dk�r�|�rbt�xVt"| dd� |�D ]@\}}|j!dk�r��qvt�#||k��s�t�#|| k ��rvt��qvW t| dd� �t|�k�r2t�$|�dk�ry tj| dd� |d d�}W n td��Y nX ntj| dd� |dd�}nt%�|j&|�'� | d�}|�|jd| � |j |j| d d� �}| |j7 } �q�|j&| d | d�}| d7 } �q�W ||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[indicies])` 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 )r6 r9 rY z;arrays used as indices must be of integer (or boolean) typeru �n)r �raise)�moderA rK zinvalid index into 0-sized�wrap)r, )(�list� enumerater( r rr r6 rF r% r r? rG r r7 �kindrX r. �indicesr� �appendr1 �ravel_multi_index�nonzero�sumr� �len�range�insert�extend�remove� transposer/ r� � broadcastr� �zip�anyr rL �take�ravel)r r4 r� Z in_indices�no_copyZ num_fancyZ fancy_dimZ error_unless_broadcast_to_emptyr% Z ellipsis_posrY rP �axZ flat_indxZ new_indices�axesZ fancy_axes�ni�ai� orig_shapeZ orig_slice�resZ_indx�_size�mir r r �_get_multi_index� s & @ " " @ " ( z+TestMultiIndexingAutomated._get_multi_indexc C s� y| � ||�\}}W nb tk rv } zDtr4t�|�}tt|j|� tt|j|d� trft|t�|�� dS d}~X Y nX | � ||||� 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� � Exceptionr r� r� r r8 rd r �_compare_index_result)r r4 r3 � mimic_getr� �e� prev_refcountr r r �_check_multi_index� s z-TestMultiIndexingAutomated._check_multi_indexc C s� y| � ||f�\}}W nb tk rx } zDtr6t�|�}tt|j|� tt|j|d� trht|t�|�� dS d}~X Y nX | � ||||� 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 N) r� r� r r� r� r r8 rd r r� )r r4 r3 r� r� r� r� r r r �_check_single_index� 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 i� N) rX r r� r% r r �may_share_memoryr r r� r� �flatr� )r r4 r3 r� r� Z indexed_arrr9 r r r r� � 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 )r6 ).r )N.)r r r r� r � zeros_likerF )r r r r r � test_boolean s z'TestMultiIndexingAutomated.test_booleanc s& t �� �� t �ddt� t �ddtj� dd� � xndD ]f}| j| j| j| jg}| j||< xBt |� D ]6}t � fdd�|D ��}| � | j |� | � | j |� qbW q6W W d Q R X | � | 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�error� c S s t | t�o| dkS )Nr� )r( �str)�idxr r r �isskip6 s z8TestMultiIndexingAutomated.test_multidim.<locals>.isskip)r r r# c 3 s | ]}� |�s|V qd S )Nr )�.0rY )r r r � <genexpr>> s z;TestMultiIndexingAutomated.test_multidim.<locals>.<genexpr>)r r r r )r r r r r r r )�warnings�catch_warnings�filterwarnings�DeprecationWarningr �VisibleDeprecationWarningr� r� r� r r� r� r r9 r r r8 rd )r Z simple_posZtocheckr3 r )r r � test_multidim- s$ z(TestMultiIndexingAutomated.test_multidimc C sN t �d�}t�� �2 t�ddt j� x| jD ]}| �||� q,W W d Q R X d S )Nr+ r� r� )r r. r r r r r� r� )r r r3 r r r �test_1dK s z"TestMultiIndexingAutomated.test_1dN) rm rn ro �__doc__r� r� r� r� r� r� r r r r r r r� � s i 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 r r �test_valid_indexingY 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 r r �test_valid_slicingc 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 )g �?g �?rA g �?rA r )r r r r r1 r� �float64)r r r r r � test_non_integer_argument_errorsp s z<TestFloatNonIntegerArgument.test_non_integer_argument_errorsc C s4 dd� }t t|dgt�d�� |dgt�d�� d S )Nc S s | | S )Nr )r r9 r r r �multz s zRTestFloatNonIntegerArgument.test_non_integer_sequence_multiplication.<locals>.multr r# )r r r r� r) )r r r r r �(test_non_integer_sequence_multiplicationx 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# g �?)g �?r )r g������@)g�������?g333333�?)r rE r r �min)r r{ r r r � test_reduce_axis_float_index� s z8TestFloatNonIntegerArgument.test_reduce_axis_float_indexN) rm rn ro r r r r r r r r r r r R s r c @ s e Zd Zdd� Zdd� ZdS )�TestBooleanIndexingc C sj t �dggg�}ttt j|d� ttt j|t �d�df� tttjt �d�� ttt j|dgdfd� d S )Nr )TrA TrA r F)�args) r r r r r1 ra �operatorr3 r� )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 ) N)r r# r )FT.)r r r# r Tr r r )r r c s � dddgdf S )NFr r .r r )r r r r � r zETestBooleanIndexing.test_boolean_indexing_weirdness.<locals>.<lambda>)r rD r� r r )r r )r r � test_boolean_indexing_weirdness� s "z3TestBooleanIndexing.test_boolean_indexing_weirdnessN)rm rn ro r r r r r r r � s r c @ s e Zd ZdZdd� ZdS )�TestArrayToIndexDeprecationz:Creating an 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 rA r )r r r r r r3 r1 r� )r r r r r �test_array_to_index_error� s z5TestArrayToIndexDeprecation.test_array_to_index_errorN)rm rn ro r r r r r r r � s r 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+ g �?g �?�1r� )r r. r r r8 )r r r r r r� � s z"TestNonIntegerArrayLike.test_basicN)rm rn ro r r� r r r r r � s r 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 r8 r? )r r )r r r� � s z$TestMultipleEllipsisError.test_basicN)rm rn ro r r� r r r r r � s r 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+ rf i����)r+ r+ r r r ���) � functools�partialr r r r rD r. r r1 )r Z subscriptr r r r � test_getitem� 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 rf i����)r+ r+ r r r rA ) r r! r r rL r rD r r. r r1 r )r r� r r r r � test_setitem� s zTestCApiAccess.test_setitemN)rm rn ro r" r# r r r r r � s r �__main__)/� __future__r r r r� r r r �numpyr �numpy.core.multiarray_testsr � itertoolsr Z numpy.testingr r r r r r r r �cdll�hasattr� ctypeslib� load_library�corer �__file__�OSErrorZ _HAS_CTYPE� ImportErrorr r� r� r� r� r� r� r r r r r r rm r r r r �<module> sR ( f 3-0 $7 '