File: //opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyo
�
� 9Zc @` s� d d l m Z m Z m Z d d l Z d d l Z d d l Z d d l Z d d l m
Z
d d l m Z d d l
m Z m Z m Z m Z m Z m Z m Z y( e j j d e j j j � Z e Z Wn e k
r� e Z n Xd e f d � � YZ d e f d
� � YZ d e f d � � YZ! d
e f d � � YZ" d e f d � � YZ# d e f d � � YZ$ d e f d � � YZ% d e f d � � YZ& e' d k r�e � n d S( i ( t divisiont absolute_importt print_functionN( t array_indexing( t product( t TestCaset run_module_suitet assert_t assert_equalt
assert_raisest assert_array_equalt assert_warnst
multiarrayt TestIndexingc B` s e Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z
d � Z d
� Z d � Z
d � Z d
� Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z RS( c C` sO t j d d d g � } t | d | t j � t | d j | j d � d S( Ni i i ( t npt arrayR t Nonet newaxist ndim( t selft a( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_none_index s c C` s� t j d d d g � } t | d
| � t | d j | k � t j d � } t t | d t j � � t j � � t j d d t
� t j d g � } t t | d t j � � t j t j d � g d t �} t t | d t j
� � Wd QXd S(
Ni i i i t ignoret g �?g t dtype( ( ( ( R R R R t baset
isinstancet int_t warningst catch_warningst filterwarningst DeprecationWarningt float_t objectt ndarray( R R ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_empty_tuple_index s
!c C` s6 t j d � } | j t j � } t j d � } t | | | | � t j d � | | <t | t j d � � t j d � j d d � } t | | | | � t j d � d d � d f | | <t | t j d � d d � d f j d d d �� t j d � j d d � } t | | | f | | | f � d S( Ni i
i t axisi i ( R t aranget astypet uintpR
t reshapeR t repeat( R t indext u_indext arr( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_same_kind_index_casting0 s #5c C` s� t j d d d g � } t | g g � t | g j | j � t j g d t j �} t | g g � t | g j | j � t j g � } t t | j | � d S( Ni i i R ( R R R R t intpR t
IndexErrort __getitem__( R R t b( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_empty_fancy_indexD s c C` s� t j d d d g d d d g d d d g g � } t | d
| � t | d
j | k � t | d | d � t | d
| d d d � f � t | d | d d � d f � t | d t j d � � t j d � } d | t f <t | d � d S( Ni i i i i i i i i .i ( i .( i .( .i ( i .i ( R R R R R t Ellipsis( R R R1 ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_ellipsis_indexR s !!
c C` s� t j d d d g d d d g d d d g g � } t | d
d d d g � t | d d d d g � t t | j d � t t | j d � d S( Ni i i i i i i i i i i����i i@ i @l ( R R R R R/ R0 ( R R ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_single_int_indexi s c C` so t j d d d g d d d g d d d g g � } t t | j t j t � � t t | j t j t � � d S(
Ni i i i i i i i i ( R R R R/ R0 t Truet False( R R ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_single_bool_indexw s
c C` sk t j d d d g g � } t j t g d t �} t | | | � d | | <t | d d d g g � d S( Ng R g �?( R R R6 t boolR ( R R R1 ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_boolean_indexing_onedim� s
c C` sk t j d � } d � } t t | | g � t t | | d d d g � t t | | d d d d g � d S( Ni c S` s | | | d k <d S( Ni����( ( R t v( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt f� s i i i ( R R% R t
ValueError( R R R< ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt& test_boolean_assignment_value_mismatch� s
c C` sq t j d � } t j d d t �} t | d *t j d d t �| | <t j d � } d | d *t | | � d S( Ni� R id i ( R t zerosR9 R6 t onesR! R
( R R, t indxt expected( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt! test_boolean_assignment_needs_api� s
c C` s t j d d d g d d d g d d d g g � } t j t t t g t t t g t t t g g � } t | | d d d d d g � t | | d d d d g g � t | | d
| | d � d
| | <t | d
d d
g d d
d g d
d d
g g � d S( Ni i i i i i i i i i ( R R R6 R7 R ( R R R1 ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_boolean_indexing_twodim� s !
c C` s� t j d � } t j d d t j �d d d � } t j d � d d d � } | | | <t | d d � t j d � } t j d � j d d � d d d � } | | | d d � f <t | d d d g � d S( Ni R i����i i i
i ( i i ( R R@ R? R. R% R R( ( R R R1 t c( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt, test_reverse_strides_and_subspace_bufferinit� s %
(c C` sx t j d � d d � d f } t j d � d d d � } t | | | | j d � � t j d � j d d � } d S( Ni
i����t Ci i ( R R% R R
t copyR( ( R R t i( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt' test_reversed_strides_result_allocation� s c C` s� t j d d � } t j d d � } t j d
� j d d d � j | d d g <t j d � j d d d � j j � | d d g <t | | � d S(
Ni i i i����i i ( i i i ( i i i i i i i ( R t fullR% R( t TRH R ( R R R1 ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt% test_uncontiguous_subspace_assignment� s
+1c C` s= t j d d � } t t | j t j d g � f d � d S( Ni i i ( i ( R R@ R R/ R0 R ( R R ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt( test_too_many_fancy_indices_special_case� s c C` sa t j d � } t | t j t � | t j t � � t | t j t � | t j t � � d S( Ni ( R R R t bool_R6 R7 ( R R ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_scalar_array_bool� s 'c C` sL t j d � } t | | d k � t | | d k � t | | k � d S( Ni .( ( R R% R ( R R ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_everything_returns_views� s c C` sf t j d � } t t | j d d g d d d g f � t t | j d d g d d d g f d � d S( Ni i i i ( i i ( R R? R R/ R0 t __setitem__( R R ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_broaderrors_indexing s (c C` s� t j d � } t j d d t j �} d | d <t t | j | � t t | j | d � t j d d t j �} d | d <t t | j | � t t | j | d � d S( Ni i R i
i����i i ( R R? R@ R. R R/ R0 RR ( R R t ind( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_trivial_fancy_out_of_bounds s
c C` s� d t j f d � � Y} t j d � } | j � j d | � } | j d � | | d d d d d g d d � f <t | d k j � � | | d d � d d d d d g f <t | d k j � � | j d � | | d
<t | d k j � � d S( Nt SubClassc B` s e Z d � Z RS( c S` s | j d � d S( Nic ( t fill( R t old( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt __array_finalize__ s ( t __name__t
__module__RY ( ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyRV s i t typei i i i i .( i i ( R R" R? RH t viewRW R t all( R RV R t s( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_nonbaseclass_values s
%%
c C` s� t j j d d d g d d d g �} t j t t t g d t �} t | | j j � t j d d g � } t | | j j � t | d
j j � t | d j j � d S( Nt NGC1001i t NGC1002g �?t NGC1003R t targett S20t V_mags >f4i i .( Ra i ( Rb g �?( Rc g �?( Rd Re ( Rf s >f4( R t recR R7 R6 R9 R t flagst writeable( R t dRT ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_subclass_writeable& s c C` sq t j d � } t j d � j d d � j } t | | j j � | j d d � } t | | d f j j � d S( Ni
i i i����i i ( R R% R( RL R Rh t f_contiguous( R R R1 ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_memory_order0 s
c C` s� d t f d � � Y} | � } d t f d � � Y} t j d � } t t | d t j � � t j d � } t t | | t j � � t j d
� } t t | | t j d � f t j � � t t | | | � f t j � � t j d � } t j d d t �} | | d <t t | d t j � � t j | d g � } t t | | t j � � t j | d g g � } t t | | t j d � f t j � � t t | | | � f t j � � d S(
Nt Zeroc B` s e Z d � Z RS( c S` s d S( Ni ( ( R ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt __index__@ s ( RZ R[ Ro ( ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyRn > s t ArrayLikec B` s e Z d � Z RS( c S` s
t j d � S( Ni ( R R ( R ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt __array__G s ( RZ R[ Rq ( ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyRp E s i i R ( ( ( i i ( ( ( R! R R? R R R R R" R ( R Rn t zRp R R1 ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_scalar_return_type; s& )#
)c C` s� t j d g � } t j t j t j � � } d | t j d g d t j �<d | t j d g d t j �<t t | j t j d g d t j �d � t t | j t j d g d t j �d � t
t j t j t j � � | � d S( Ni i R ( R R t syst getrefcountR R. t uint8R R/ RR R ( R R t refcount( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_small_regressions] s c C` s� t j d d t j �t d � d d !} | j t j d � � } t j d d t j �t d � d d !} | j t j d � � } t j d d d
�| d <t j | j � } t | | | � | | | <t j | j d � j t j � d t j
d � j d !} | j t j
� | j } t j | j � | d <t | | j t j
� | � | | | j t j
� <| | d
d k | d d d
� | | d
d k <d S( Ni@ R R i i����t S8i R1 i t S.i i t bbbbbbbb( R R? t int8t ordR] R R R% t sizeR R. t itemsizeR& t int16( R R; Rj t xR1 ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_unalignedk s ))
: c C` sx t j d � } d t f d � � Y} d g d g f } | | � } t | | j d k � t | | f j d k � d S( Ni t
TupleSubclassc B` s e Z RS( ( RZ R[ ( ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyR� � s i ( i i ( i ( i ( R R@ t tupleR t shape( R R, R� R* ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_tuple_subclass� s c C` s� d t f d � � Y} t j d � } t | | � | | � f � t j d d d
d g �} t | | � | | � f � d S( Nt SequenceLikec B` s# e Z d � Z d � Z d � Z RS( c S` s d S( Ni ( ( R ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyRo � s c S` s d S( Ni ( ( R ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt __len__� s c S` s t d � � d S( Ns Not possible( R/ ( R t item( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyR0 � s ( RZ R[ Ro R� R0 ( ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyR� � s i
i R t f1t i8t f2( i ( R� R� ( R� R� ( R! R R% R
R? ( R R� R, ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt! test_broken_sequence_not_nd_index� s
c C` s� t j d � } t j d � } t j d � d d � d d d f } t j | d � } t | | | | j � � t j d d t j �} t | | | f | | j � | f � d S( Ni
i i7 i R ( i
i ( i
i7 i i ( R R@ R% R t broadcast_toR
RH R? R. ( R R� t x2RT t zind( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt! test_indexing_array_weird_strides� s %( RZ R[ R R# R- R2 R4 R5 R8 R: R> RC RD RF RJ RM RN RP RQ RS RU R` Rk Rm Rs Rx R� R� R� R� ( ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyR
s8
" t TestFieldIndexingc B` s e Z d � Z RS( c C` sP t j d d g � } t t | d t j � � t t | d g t j � � d S( NR t f8( ( R R� ( R R? R R R" ( R R ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyRs � s ( RZ R[ Rs ( ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyR� � s t TestBroadcastedAssignmentsc B` s> e Z d � Z d � Z d � Z d � Z d � Z d � Z RS( c C` s | | | <| S( N( ( R R RT t val( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt assign� s
c C` s� t j d � } t j d � | d <t j d � | d d d g d d � f <t j d � | d d � d d g f <t j d
� | d g d g d g g d d g f <d S( Ni i i .i ( i i ( i i i ( i i i ( i i i ( i i i ( R R? R@ ( R R ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_prepending_ones� s
(%c
C` s� | j } t j } t j d � } t t | | | d t j d � � t j � �v t j d t
� t t
| | | d d d g f t j d � � t t
| | | d g d g g f t j d � � Wd QXd S(
Ni .i i t errori ( i i ( i i ( i i i ( R� R t s_R? R R= R@ R R t simplefilterR ( R R� R� R ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_prepend_not_one� s #
"c C` s | j } t j } t j d � } t t | | | d t j d � � t t | | | d t j d � � t t | | | d d � d g f t j d � � t t | | | d d � d g f t j d
� � t t | | | d g d d � f t j d � � d S( Ni i .i i ( i i ( i i ( i i ( i i ( i i ( i i ( R� R R� R? R R= ( R R� R� R ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_simple_broadcasting_errors� s ##22c C` s| t j d � } d d d g | d g d g d g g d d d g f <t | d d � d d � f d d d g k j � � d S( Ni i i i i i ( i i ( R R? R R^ ( R R ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_index_is_larger� s 4c C` s{ t j d � } t j d � d d � d f } t j d � d d d � } | | | <t | d d d � | k j � � d S( Nid i����( id id ( R R? R% R R R^ ( R R R; R1 ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_broadcast_subspace� s
( RZ R[ R� R� R� R� R� R� ( ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyR� � s
t TestSubclassesc B` s# e Z d � Z d � Z d � Z RS( c C` s� d t j f d � � Y} t j d � j | � } t t | d | � � t | d j | k � t t | d d d g | � � t t | | d k | � � d S( NRV c B` s e Z RS( ( RZ R[ ( ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyRV � s i i i i i ( R R" R% R] R R R ( R RV R_ ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt
test_basic� s c C` s� t j d d g d d g g � } t t | d d d g d d � f t j � � t j t j d � j d d � � } t | d d � t j d � f | d d � d f j � d S(
Ni i i i i i2 i i
i����(
R t matrixR R t asmatrixR% R( R R RL ( R t mR� ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_matrix_fancy s !/$c C` s� d t j f d � � Y} t j d � j | � } | d } t | j | � t | j | � | d d d d g } t | j | � t | j | � | | d k } t | j | � t | j | � d S( NRV c B` s e Z d � Z RS( c S` s t j | � | _ | | _ d S( N( R R t finalize_statusRX ( R RX ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyRY s ( RZ R[ RY ( ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyRV s i
i i i i ( R R" R% R] R
R� RX ( R RV R_ t new_s( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_finalize_gets_full_info s
( RZ R[ R� R� R� ( ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyR� � s
t TestFancingIndexingCastc B` s e Z d � Z RS( c C` s� d } t j | � j t � } t | d <t j | � } t j d g � | | <t | d d � t t j | j d g d g f t j d
g � � t | d d � t t j | j | t j d g � � t | d d � d S(
Ni i? i i i y �?( i i? ( i i ( i i y @ �?( i i ( i i (
R R? R& R9 R6 R R R t ComplexWarningRR ( R R� t
bool_indext
zero_array( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_boolean_index_cast_assign( s
( ( RZ R[ R� ( ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyR� ' s t TestFancyIndexingEquivalencec B` s e Z d � Z d � Z RS( c C` s� t j d d t �} | j � } d d
d g | d *d d d g | d d d g <t | | � t j d d t �d d d � f } d d d g g | d g d d � f <t | | d � | j } d g d
g d g g | d d � d g f <t | | d d � d f � t j d d t �} | j � } d g d g d g d g g g | d d � d f <d g d g d g d g g g | d g d f <t | | � | j d � } d g d g d g d g g g | d g d f <t | | � d S( Ni R i i i i i .t F( i i ( i i ( i i ( i i ( i i i ( R R% R! RH R
R RL R@ ( R R R1 R, t cmp_arr( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_object_assign> s&
%% + 1.
.c C` s� t j d � } | j � } t j d d d g � | d *t j d d d g � | d d d g <t | | � t j d � d d d � f } t j d d d g g � | d g d d � f <t | | d � d S( Ni t 2s -3s -1i i i i ( R R% RH R R
R ( R R R1 ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_cast_equivalence^ s %
.( RZ R[ R� R� ( ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyR� = s t TestMultiIndexingAutomatedc B` sV e Z d Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z
RS( s�
These test 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.
c C` s t j t j d d d d g � � j d d d d � | _ t j d � | _ d t d t j t � t j t
t t g � t j t
t g t t
g g � t j t t g t t g g g � t d d d � t d d d � t d
d d � t d d d
� t j d d t j
�t j d d d g � t j d g d g d g g � t j d d g d d g g d t j d � j � �t j d d g d t j �t j d g d d t �t j d d g � g | _ t d d d g t j t
g � d g | _ t d d � d g | _ d S( Ni i i i i t skipi����i id i i����i����i����R R. i g g �?( i i i i ( i i i ( R R% t prodR( R t emptyR1 R3 R R7 R6 t sliceR R. R t newbyteorderR| R? t intt complex_indicest simple_indicest fill_indices( R ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt setUp} s* 9 ! 3*c C` sf t | � } g } t } d } d } t } d } d } xmt | � D]_\ }
} | d k ra qC n t | t j � r� | j t k r� t } | j
d k r� t � n | | j
7} | | j
7} qC n | t k r� | d k r� |
} qC n t � n t | t
� r | d 7} qC n t | t j � sCt j | d t j �} | | |
<n3 | j j d k rv| j j d k rvt d � � n | j
d k r�t } n | d 7} | d 7} qC W| j
| d k r�t � n | d k r�d | k r�| j � | f S| d k r!t
d d � g | j
| | | | d +n x�t | � D]�\ } } t | t
� r�t j | j | j | � � } | j d | g � q.n_ | d k r�| j d t j d g d t j �g � | j | j | d | j | � } q.n t | t j � r�| j t k r�| j | j | | | j
!k r*t � n y6 t j t j | � | j | | | j
!d d
�}
Wn2 t } t j d g | j � d t j �}
n X| j
d k r�| j | j | t j | j | | | j
!� f | j | | j
� } |
} qAt � nB | j
d k rA| | j | k s5| | j | k rAt � qAn | j
d k r�| | j | k sw| | j | k r�t � q�n t | � d k r�| d d d k r�| | k r�| d j | � q.| d 7} | j d | g � q.W| d k r| r| } t t | j
� � } g } | j d d g � d } d } x� | D]� } | d 7} | d d k r�| d j | d � | | =| d 8} xB t | | t | d � � D] } | j | � | j | � q�Wn | t | � d 7} qEW| } | j | | � } n d } xO| D]G} | d d k r5t | � d k r?qn | j } | | | t | d � !} | j | j | t j | � j! t" � f | j | t | d � � } t j# | d � } | j$ d k rL| r�t � n xm t% | d | � D]U \ } } | j$ d k rq�n t j& | | k � s<t j& | | k � r�t � q�q�Wn t | d � t | � k r�t j' | � d k r�y t j | d | d d
�} Wq�t d
� � q�Xq�t j | d | d d �} n t( � | j) | j* � d | �} | j | j | | j | j | d � } | | j
7} qn | j) | d d | �} | d 7} qW| | f S( s� 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.
i i R R1 RI s; arrays used as indices must be of integer (or boolean) typeR_ t nt modet raisei����R< s invalid index into 0-sizedt wrapR$ N( i (+ t listR6 R7 R t enumerateR R R" R R9 R R/ R3 R� R R. t kindRH R% t indicesR� t appendR( t ravel_multi_indext nonzerot sumR� t lent ranget insertt extendt removet transposeR& R� t broadcastR~ t zipt anyR R= t taket ravel( R R, R� t
in_indicest no_copyt num_fancyt fancy_dimt error_unless_broadcast_to_emptyR t ellipsis_posRI RA t axt flat_indxt new_indicest axest
fancy_axest nit ait
orig_shapet
orig_slicet rest _indxt _sizet mi( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt _get_multi_index� s� !
$
+ (%! $)C ' ' 2
$
B + &
c C` s� y | j | | � \ } } Wn` t k
r~ t j | � } t t | j | � t t | j | d � t | t j | � � d SX| j | | | | � d S( s 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.
i N( R� t ExceptionRt Ru R R0 RR R t _compare_index_result( R R, R* t mimic_getR� t
prev_refcount( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt _check_multi_index{ s
c C` s� y | j | | f � \ } } Wn` t k
r� t j | � } t t | j | � t t | j | d � t | t j | � � d SX| j | | | | � d S( sW 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`).
i N( R� R� Rt Ru R R0 RR R R� ( R R, R* R� R� R� ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt _check_single_index� s
c C` s | j � } | | } t | | � | j d k r� | j d k r� t t j | | � | k � | r| t t j | � d � q� t t j | � d � n | j � } | d | | <| j d k r� d S| r� | j d k r� | d 7} t | | � d S| j
| j � c d 7<t | | � d S( s4 Compare mimicked result to indexing result.
i i i i� N( RH R
R~ R R R t may_share_memoryR Rt Ru t flatR� ( R R, R* R� R� t indexed_arrR1 ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyR� � s$
c C` s� t j d � } t | t j t � d � d | t j t � <t | d � | j | j t j | j d t �f � | j | j t j | j d t �d f � | j | j t j | j d t �d f � d S( Ni i R .i ( .i ( N.( R R R R6 R� R t
zeros_likeR9 R ( R R ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_boolean� s
(,c ` s� t j � �� t j d d t � t j d d t j � d � � x� d d d g D]� } | j | j | j | j g } | j | | <xV t | � D]H } t
� f d � | D� � } | j | j | � | j | j
| � q� WqO WWd QX| j | j d � | j | j
d
� t t | j j d � t t | j j d d � t t | j j d d d g d d f � t t | j j d d d g d d f d � d S(
NR� R c S` s t | t � o | d k S( NR� ( R t str( t idx( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt isskip� s i i i c 3` s! | ] } � | � s | Vq d S( N( ( t .0RI ( R� ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pys <genexpr>� s i ( i i i i ( i i i i ( i i i i i ( i i i i i ( R R R R R t VisibleDeprecationWarningR� R� R� R R� R� R R1 R R/ R0 RR ( R t
simple_post tocheckR* ( ( R� sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt
test_multidim� s$
!(c C` s` t j d � } t j � �? t j d d t j � x! | j D] } | j | | � q< WWd QXd S( Ni
R� R ( R R% R R R R� R� R� ( R R R* ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_1d� s
( RZ R[ t __doc__R� R� R� R� R� R� R� R� ( ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyR� m s � t TestCApiAccessc B` s e Z d � Z d � Z RS( c C` s� t j t d � } t t | t j d � d � t t | t j d � d � t t | t j d � d � t t | t j d
� d � t t | t j d � d � t j d � } t | d | | d � � | j d d � } t | d | | d � � d S( Ni i
i i����i i i i����( ( i
i
( i
i
(
t functoolst partialR R R/ R R@ R% R
R( ( R t subscriptR ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_getitem� s c C` s9 t j t d � } t t | t j d � d � t t | t j d
� d d � t t | t j d � d d � t t | t j d � d d � t t | t j d � d d � t t | t j d � d d � t j d � } | | d d � t | d d k � | j
d d � } | | d d � t | d d d g � d S(
Ni i
i i i����i i i i����( ( i
i
( i
i
( R� R R R R= R R@ R/ R% R R( R
( R R� R ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt test_setitem s ( RZ R[ R R ( ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyR� � s t __main__(( t
__future__R R R Rt R R� t numpyR t numpy.core.multiarray_testsR t itertoolsR t
numpy.testingR R R R R R
R t ctypeslibt load_libraryt coreR t __file__t cdllR6 t
_HAS_CTYPEt ImportErrorR7 R
R� R� R� R� R� R� R� RZ ( ( ( sQ /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_indexing.pyt <module> s0 4
� � <-0� �'