OwlCyberSecurity - MANAGER
Edit File: test_indexing.cpython-38.pyc
U �p�]�� � @ sT d dl mZmZmZ d dlZd dlZd dlZd dlZd dlZd dl Z d dl m Z d dl mZ d dlmZmZmZmZmZmZmZ G dd� de�ZG dd � d e�ZG d d � d e�ZG d d � d e�ZG 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� de�Z"G d d � d e�Z#G d d � d e�Z$dS ) � )�division�absolute_import�print_functionN)�array_indexing)�product)�assert_� assert_equal� assert_raises�assert_array_equal� assert_warns� HAS_REFCOUNT�suppress_warningsc @ 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&dIS )J� 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 �@/usr/lib/python3/dist-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 1 r z5TestIndexing.test_slicing_no_floats.<locals>.<lambda>c s � dd �dd�f S )Nr r � r r r r r r 2 r c s � dd d�d d�f S �Nr r r r r r r r r 3 r c s � ddd�d d �f S )Nr � r r r r r r r 4 r c s � d d �dd �f S r r r r r r r 5 r c s � d d� S r r r r r r r 7 r c s � d d�dd�f S )Nr r" � @r r r r r r 8 r c s � d dd�d d�f S r! r r r r r r 9 r c s � d d�d d �f S r 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 � 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 A r c s � ddd� S )Nr% r r# r r r r r r C r c s � dd d� S )Nr% r# r r r r r r D r c s � dd �d dd�f S )Nr r# r r r r r r E r c s � ddd�d d�f S )Nr% r" r$ r r r r r r r F r c s � ddd�d d �f S )Nr% r& r r r r r r G r c s � d d �ddd�f S )Ng�������?r$ r# r r r r r r H r c s � d d d� S r r r r r r r J 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 P r z;TestIndexing.test_index_no_array_to_index.<locals>.<lambda>r( r r r r � test_index_no_array_to_indexL 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 Znewaxis�ndim�r r r r r �test_none_indexR 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 �test_empty_tuple_indexX 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 r6 �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_castingf 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- r5 )r r r r6 �intpr r � __getitem__�r r �br r r �test_empty_fancy_indexz 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 r1 �EllipsisrH 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 rK rL rM rN r ���i @l )r r r r r rG 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 rK rL rM rN 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- TFrK �r' r' r5 ) r �onesr r r rG r7 �bool�slice)r rD rC 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 )Nr Tr5 r% )r r rY r rH 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 )NrR r )r �vr r r �f� s z>TestIndexing.test_boolean_assignment_value_mismatch.<locals>.fr" r 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�� r5 T�d r" )r r7 rY rX �objectr )r rD �indxZexpectedr 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 rT rU rH r r r � test_boolean_indexing_twodim� s � � �z)TestIndexing.test_boolean_indexing_twodimc C sH t �dddg�}dddg}t|| ddg� t|d |f ddgg� d S )Nr" r r- TFrU rH 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 r5 rR r �r r r< r r" )r rX r7 rF r? r rA )r r rI �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 �copyrA )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? rA �Trm r rH 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 rX r r rG 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 k � t||d k � t||d d � k � d S )Nr 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 r7 r r rG � __setitem__r/ r r r �test_broaderrors_indexingD 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 r5 r< rR r � )r r7 rX rF r r rG ry )r r �indr r r � test_trivial_fancy_out_of_boundsI 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 )NrK r5 r r" r- r rR )r r? rF rA r )r r �idx�resr r r � test_trivial_fancy_not_possibleT s , z,TestIndexing.test_trivial_fancy_not_possiblec 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__e s zJTestIndexing.test_nonbaseclass_values.<locals>.SubClass.__array_finalize__N��__name__� __module__� __qualname__r� r r r r �SubClassd s r� rx )r8 r" r r r- r' .)r r9 r7 rm �viewr� r �all)r r� r r: r r r �test_nonbaseclass_valuesc 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)ZNGC1001r{ )ZNGC1002r% )ZNGC1003r% )�targetZS20)ZV_magz>f4r5 FTr r" .)r Zrecr rY r �flags� writeable)r �dr| r r r �test_subclass_writeablex 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? rA rq r r� � f_contiguousrH r r r �test_memory_order� 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)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 r5 )rc r r7 r r2 �float_r r9 )r r� �zr� r rI 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" r5 ) r r r �sys� getrefcountr6 rF Zuint8r r ry 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�@ r5 r r" i����ZS8rp Zbbbbbbbb�S.r r )r r7 �int8�ordr� r6 r r? �sizer rF �itemsizer@ Zint16)r r] r� �xrI 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 rX �tupler �shape)r rD r� rC 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 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 rG � s zPTestIndexing.test_broken_sequence_not_nd_index.<locals>.SequenceLike.__getitem__N)r� r� r� r� r� rG r r r r � SequenceLike� s r� r< r� )�f1�i8)�f2r� r5 )rc r r? r r7 )r r� rD 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' r5 )r rX r? Z broadcast_tor rm r7 rF )r r� Zx2r| 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 �ddddgf}d||< t|d� d S ) NrW rR r r" r r- r< g $@)r r7 rZ r )r ZarrorD Zslicesr r r �$test_indexing_array_negative_strides s z1TestIndexing.test_indexing_array_negative_stridesN)'r� r� r� r r* r, r0 r4 r; rE rJ rQ rS rV r[ r\ r` re rf rg rj ro rr rt rv rw rz r} r� r� r� r� r� r� r� r� r� r� r� r r r r r sH " r c @ s e Zd Zdd� ZdS )�TestFieldIndexingc C s< t �ddg�}tt|d t j�� tt|dg t j�� d S )Nr )r Zf8r )r r7 r r2 r9 r/ r r r r� s z)TestFieldIndexing.test_scalar_return_typeN)r� r� r� 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 r+ r )r r r| �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 r7 rX 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_r7 r r_ rX �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" .rh )r r r r� )r� r r� r7 r r_ r� r r r � test_simple_broadcasting_errors4 s $$z:TestBroadcastedAssignments.test_simple_broadcasting_errorsc C sX t �d�}dddg|dgdgdggdddgf< t|d d�d d�f dddgk�� � d S )Nrx r r- r' r r" )r r7 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 )N)rb rb rb rR )r r7 r? r r� )r r r] rI r r r �test_broadcast_subspaceF s z2TestBroadcastedAssignments.test_broadcast_subspaceN) r� r� r� r� r� r� r� r� r� r r r r r� s r� c @ s> e Zd Zdd� Zdd� Zdd� Zejje dd�d d � �Z d S ) �TestSubclassesc C s4 G dd� dt j�}t �d�}|�|�}|d d� }tt|�|k� t|j|k� t||d d� � |dddg }tt|�|k� t|j|k � tt|j�t jk� t||dddg � t|j|dddg � ||dk }tt|�|k� t|j|k � tt|j�t jk� t|||dk � t|j||dk � d S )Nc @ s e Zd ZdS )z+TestSubclasses.test_basic.<locals>.SubClassNr� r r r r r� T s r� r r- r r" r )r r9 r? r� r r8 r1 r )r r� r r: Zs_slice�s_fancyZs_boolr r r � test_basicO s& zTestSubclasses.test_basicc C sL G dd� dt j�}t �d�}|�|�}d|j_|dddg }t|jj� d S )Nc @ s e Zd ZdS )z8TestSubclasses.test_fancy_on_read_only.<locals>.SubClassNr� r r r r r� o s r� r Fr r" r )r r9 r? r� r� r� r )r r� r r: r� r r r �test_fancy_on_read_onlyl s z&TestSubclasses.test_fancy_on_read_onlyc 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 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 )r r9 r? r� r r� r� )r r� r: Znew_sr r r � test_finalize_gets_full_infoy s z+TestSubclasses.test_finalize_gets_full_infozPython lacks refcounts)�reasonc C s� G dd� dt j�}t �dg��|�}|dd� t|jtdd�� tt�|j�d� z|dd� t �W nB t k r� t td�r�t � �}|� t� t�� W 5 Q R X Y nX t|jtdd�� tt�|j�d� d |dd �< t|jtdd �� tt�|j�d� zd|dd �< t �W nF t k �rZ t td��rVt � �}|� t� t�� W 5 Q R X Y nX t|jtdd �� tt�|j�d� d S ) Nc @ s e Zd Zdd� Zdd� ZdS )zETestSubclasses.test_slice_decref_getsetslice.<locals>.KeepIndexObjectc S s || _ |tdd�krt�d S )Nr rL �rd rZ r_ )r rd r r r rG � s zQTestSubclasses.test_slice_decref_getsetslice.<locals>.KeepIndexObject.__getitem__c S s || _ |tdd�krt�d S )Nr r' r� )r rd r� r r r ry � s zQTestSubclasses.test_slice_decref_getsetslice.<locals>.KeepIndexObject.__setitem__N)r� r� r� rG ry r r r r �KeepIndexObject� s r� r" r r r rL � exc_clearrK r- r' )r r9 r r� r rd rZ r� r� �AssertionErrorr_ �hasattrr �filter�DeprecationWarningr� )r r� �kZsupr r r � test_slice_decref_getsetslice� s8 z,TestSubclasses.test_slice_decref_getsetsliceN) r� r� r� r� r� r� �pytestZmarkZskipifr r� r r r r r� N 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 �? T)r r" r" r y @ �?r y �?) r r7 r@ rY r r r ZComplexWarningry )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 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 r5 r" �r" r r- r r )r- r' r r' .�F)r r? rc rm r rq rX )r r rI rD 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? rm r r rH 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 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 rK )r- r r rK �skipr TF���r rb r' rR ������)r r" r" r5 rF � r r% )r r? �prodrA r �emptyrI rP r rZ rF r6 Z newbyteorderr� r7 �int�complex_indices�simple_indices� fill_indicesr r r r �setup s. & " ��z TestMultiIndexingAutomated.setupc C s� t |�}g }d}d}d}d}d}d} 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�ztj | tj d�} W n t k r� t�Y nX | || < n$| jjdk�r| jjdk�rtd ��| jdk�r$d}|d7 }|d7 }q,|j| dk �rJt�|dk�rjd|k�rj|�� |fS | dk �r�t dd�g|j| || | d �< t|�D �]:\} } 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�rht�z*tjt�| �|j| | | j � d d�} W n4 tk �r� d}tj dg| �� tj d�} Y nX | jdk�r |�|jd| � t�|j| | | j � �f |j| | j d� �}| } nt�n2| jdk�rV| |j| k�sR| |j| k �rVt�| 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 ]} |�| � |� | � �q||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�|�rzt�t$| dd� |�D ]@\}}|j#dk�r��q�t�%||k��s�t�%|| k ��r�t��q�t| dd� �t|�k�rPt�&|�dk�r6z tj| dd� |d d�}W n tk �r2 td��Y nX ntj| dd� |dd�}nt �|j'|�(� | d�}z0|�|jd| � |j |j| d d� �}W n t k �r� t�Y nX | |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" r5 rI rn z;arrays used as indices must be of integer (or boolean) typer: �nr� �raise)�moderR r^ zinvalid index into 0-sizedZwrapr= ))�list� enumerater2 r r9 r6 rY r. r rP rZ r rF r_ Zkindrm r? �indicesr� �appendrA Zravel_multi_indexZnonzero� Exception�sumr� �len�range�insert�extend�removeZ transposer@ r� Z broadcastr� �zip�anyr �take�ravel)r rD r� Z in_indices�no_copyZ num_fancyZ fancy_dimZ error_unless_broadcast_to_emptyr. Z ellipsis_posrn rd ZaxZ flat_indxZ new_indicesZaxesZ fancy_axesZniZaiZ orig_shapeZ orig_slicer Z_indxZ_size�mir r r �_get_multi_index7 s4 � & � �� " "�� � � "� ��� z+TestMultiIndexingAutomated._get_multi_indexc C s� z| � ||�\}}W np tk r� } zRtr4t�|�}tt|�|j|� tt|�|j|d� trnt |t�|�� W Y �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 r� r r� r� r r8 rG ry r �_compare_index_result�r rD rC � mimic_getr� �eZ prev_refcountr r r �_check_multi_index& s z-TestMultiIndexingAutomated._check_multi_indexc C s� z| � ||f�\}}W np tk r� } zRtr6t�|�}tt|�|j|� tt|�|j|d� trpt |t�|�� W Y �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 Nr 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 ra N) rm r r� r. r r Zmay_share_memoryr r r� r� Zflatr� )r rD rC r r� Z indexed_arrrI r r r r V 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" r5 rO )N.)r r r r r Z zeros_likerY r/ r r r � test_booleanv s � � �z'TestMultiIndexingAutomated.test_booleanc s t �� �� t �ddt� t �ddtj� dd� � dD ]b}| j| j| j| jg}| j||< t |� D ]6}t � fdd�|D ��}| � | j |� | � | j |� q^q4W 5 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� )r2 �str)r~ r r r �isskip� s z8TestMultiIndexingAutomated.test_multidim.<locals>.isskip)r r r- c 3 s | ]}� |�s|V qd S r+ r )�.0rn �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�filterwarningsr� r ZVisibleDeprecationWarningr� r� r� r r� r r rI r r rG ry )r Z simple_posZtocheckrC r r r � test_multidim� s( � z(TestMultiIndexingAutomated.test_multidimc C s& t �d�}| jD ]}| �||� qd S )Nr< )r r? r� r )r r rC r r r �test_1d� s z"TestMultiIndexingAutomated.test_1dN) r� r� r� �__doc__r� r r r r r r r 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) rA r� Zfloat64r/ 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 rI r r r �mult� s zRTestFloatNonIntegerArgument.test_non_integer_sequence_multiplication.<locals>.multr" r- )r r) r r� r3 )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 r7 r r) �min)r r� r r r � test_reduce_axis_float_index� s z8TestFloatNonIntegerArgument.test_reduce_axis_float_indexN) r� r� r� r r r r r r r r r r r � s r c @ s e Zd 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) rA ru �operatorrC r r� 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 ) N)r r- r' )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 rX r� r r r r r r � test_boolean_indexing_weirdness� s "z3TestBooleanIndexing.test_boolean_indexing_weirdnessN)r� r� r� 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" rR r )r r r r) r! rC rA r� r/ r r r �test_array_to_index_error� s z5TestArrayToIndexDeprecation.test_array_to_index_errorN)r� r� r� 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< r g �?�1r� )r r? r r rG r/ r r r r� s z"TestNonIntegerArrayLike.test_basicN�r� r� r� 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 rG rP r r r r r� s z$TestMultipleEllipsisError.test_basicNr( 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< r{ ����r< r< r' r r ���) � functools�partialr r r r rX r? r rA )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 r{ r+ r, r' r r rR ) r. r/ r r r_ r rX r r? r rA r )r r� r r r r � test_setitem0 s zTestCApiAccess.test_setitemN)r� r� r� r0 r1 r r r r r* s r* )%Z __future__r r r r� r r. r! r� Znumpyr Z numpy.core._multiarray_testsr � itertoolsr Z numpy.testingr r r r r r r rc r r� r� r� r� r� r� r r r$ r& r) r* r r r r �<module> s: $ 3p0 )7