Current File : //opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyo
�
� 9Zc @` ss d d l m Z m Z m Z d d l Z d d l m Z m Z m Z m
Z
m Z m Z m
Z
m Z d d l m Z m Z m Z m Z m Z m Z m Z m Z m Z m Z 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 � Z d � Z! d � Z" d � Z# d � Z$ e% d k roe � n d S( i ( t divisiont absolute_importt print_functionN( t run_module_suitet TestCaset assert_t assert_equalt assert_array_equalt assert_almost_equalt assert_array_almost_equalt
assert_raises(
t mgridt ndenumeratet
fill_diagonalt diag_indicest diag_indices_fromt index_expt ndindext r_t s_t ix_t TestRavelUnravelIndexc B` s# e Z d � Z d � Z d � Z RS( c C` s� t t j d d" � d# � t t j d$ d% � d � t t j d d& � d' � t t j d( 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 t t j d/ d0 � t t t j d1 d2 � t t t j d3 d4 � t t t j d5 d6 � t t j d: d; � d d d
g � t t j d d d
g d< � d@ � t j d d d g d
d d g g � } t t j | dA � d d d g � t t j | dB d d �d d d g � t t j | dC d d �d d d g � t t j | dD d dE �d d d g � t t j dF dG � d! � t t j t j d d d g � dH � d d d g d
d d g g � t t j t j d d d g � dI d d �d d d g d
d d g g � t t j d! dJ � d d d
d g � d S(K Ni i i i� i i^ iB i����g �?i i����g�������?g i i i i i i) i% t ordert Fi i
t modet clipi i t wrapi i i iU ( i i ( i i ( i i ( i i ( i i^ ( i iB ( i iB ( i i^ ( i i ( i i ( i i ( i����i ( i i ( i i ( i i ( i i����( i i ( i i ( i i ( g�������?g ( i i i i i* i. ( i i i ( i i i i i i* i. ( i i ( i i ( i i ( i i ( R R ( i i i i ( i i i i ( i i ( i i ( i i i i ( R t npt
unravel_indext ravel_multi_indexR
t
ValueErrort TypeErrort array( t selft arr( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyt
test_basic s< " '"%%$ ' c
C` s( x!t j t j t j t j t j t j g D]�} t j d d d d d d g d d d d d d g g d | �} d } d | d | d } t t j | | � | � t | t j
| | � � | d d | d } t t j | | d
d �| � t | t j
| | d
d �� t j d d d d d d g d d d d d d g d d d d d d g g d | �} d } d
d | d | d | d } t t j | | � | � t | t j
| | � � | d d | d d | d } t t j | | d
d �| � t | t j
| | d
d �� q+ Wd S( Ni i i i i i t dtypei i R R i i
( i i ( i i i
( R t int16t uint16t int32t uint32t int64t uint64R R R R ( R! R$ t coordst shapet uncoords( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyt test_dtypes2 s. 9 E "" c C` s� t t j d d d d g d d d
�t j d d d d g d � � t t j d d d d g d d d �t j d d d d g d � � t t t j d d d d g d � d S( Ni i i����i i i i i R R i t raiseR i ( i i i i ( i i i i ( i i i i ( R R/ R R/ ( i i i i ( i i i i ( R R R R
R ( R! ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyt test_clipmodesM s ! ( t __name__t
__module__R# R. R0 ( ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyR s # t TestGridc B` s# e Z d � Z d � Z d � Z RS( c C` s� t d d d � } t d d d � } t | j d k � t | j d
k � t | d d k � t | d d � t | d d k � t | d | d d d � t | d | d d d � t | d | d d d � d S( Ni����i y $@g�������?i
i i i i g @g "@( i
( i ggfffff�?g �q �q�?( R R R, R ( R! t at b( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyR# [ s c C` sL t j d d d d �\ } } t | d � t | t d d d � d � d S(
Ni i
t retstepi i g �H@y I@i
g9/����?( R t linspaceR R R ( R! t yt st( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyt test_linspace_equivalenceg s
c C` s� t d d d � d d d � f } t d d d � d d d � f } t | j d
k � t | j d k � t | d
d
d d � f t j d d � � t | d d d � d
f d t j d d � � t | d
d d d � f t j d d � d � t | d d d � d f d t j d d � d � t | d
d d d � f | d
d
d d � f d t j d d � d � t | d d d � d f | d d d � d
f d t j d d � d � d S( Ni����i y $@i����i g�������?g�������?i
i i t di ( i i
i
( i i i ( R R R, R R t onesR ( R! t cR; ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyt test_ndl s "".10400( R1 R2 R# R: R> ( ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyR3 Z s t TestConcatenatorc B` s, e Z d � Z d � Z d � Z d � Z RS( c C` s� t t d t j d d d d d d g � � t j d � } t | d d | f } t | d d d d d d d d d 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! R5 R= ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyt test_1d| s ,c C` s- t d d d � f } t | j d k � d S( Ng333333$@i i
t f8( R R R$ ( R! t g( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyt test_mixed_type� s c C` sK t d t j d g � t j d d d g � d f } t | j d k � d S( Ng333333$�i i i i g $@RA ( R R R R R$ ( R! RB ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyt test_more_mixed_type� s 4c C` s t j j d d � } t j j d d � } t d | | f } t | j d k � t | d d � d d � f | � t | d d � d d � f | � t | | f } t | j d k � t | d d � d d � f | � t | d d � d d � f | � d S( Ni t 1i
( i i
( i
i ( R t randomt randR R R, R ( R! R5 R= R; ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyt test_2d� s ###( R1 R2 R@ RC RD RH ( ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyR? { s t TestNdenumeratec B` s e Z d � Z RS( c C` sJ t j d d g d d g g � } t t t | � � d d d d
g � d S( Ni i i i i ( i i ( ( i i i ( i i ( ( i i i ( i i ( ( i i i ( i i ( ( i i i ( R R R t listR ( R! R4 ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyR# � s !( R1 R2 R# ( ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyRI � s t TestIndexExpressionc B` s e Z d � Z d � Z RS( c C` sE t j d � } t | d | t d � t | d | t d � d S( Ni i����( R t arangeR R R ( R! R4 ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyt test_regression_1� s c C` s� t j j d d d � } t | d d � d d � d d g f | t d d � d d � d d g f � t | d d � d d � d d g f | t d d � d d � d d g f � d S( Ni i i i i i ( R RF RG R R R ( R! R4 ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyt
test_simple_1� s O( R1 R2 RM RN ( ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyRK � s t TestIx_c B` s5 e Z d � Z d � Z d � Z d � Z d � Z RS( c C` sI xB t t j f D]1 } t j | d � � \ } t | j t j � q Wd S( Ni ( t rangeR RL R R R$ t intp( R! t funcR4 ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyRM � s c ` s� d } x� t t j f D]� } t j g | D] } | | � ^ q) � } x t t | | � � D]h \ � \ } } t | j � | � t t � f d � t | j � D� � � t t j
| j t � � qZ Wq Wd S( Ni i i i c 3` s- | ]# \ } } | � k r | d k Vq d S( i N( ( t .0t jt sh( t k( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pys <genexpr>� s ( i i i i (
RP R RL R t enumeratet zipR R, R t allt
issubdtypeR$ t int( R! t sizesRR t szt arraysR4 ( ( RV sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyt test_shape_and_dtype� s (()c C` sB t t t t g } t j | � \ } t t j | � d | � d S( Ni ( t Truet FalseR t nonzeroR R ( R! t bool_at int_a( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyt test_bool� s c C` s5 d d d g d d d g g } t t t j | � d S( Ni i i i i i ( R
R R R ( R! t idx2d( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyt test_1d_only� s c C` so d } t j | � } t | | � } t | d j | d f � t | d j d | f � t | j | f � d S( Ni i i ( R RL R R R, ( R! t length_of_vectort xt out( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyt test_repeated_input� s ( R1 R2 RM R_ Re Rg Rk ( ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyRO � s
c
C` so t j t j d d d g g � d d t j d d d g g � f } t | d d d d d d d d g g � d S( Ni i i i i i i ( R t c_R R ( R4 ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyt test_c_� s Cc c` s� t j d t � } t | d � t | t j d d d g d d d g d d d g g � f Vt j d t � } t | d � t | t j d d d g d d d g d d d g d d d g d d d g d d d g d d d g d d d g d d d g d d d g g
� f Vt j d
t � } t | d t � t | t j d d d g d d d g d d d g d d d g d d d g d d d g d d d g d d d g d d d g d d d g g
� f Vt j d t � } t | d � t | t j d d d d d d d d d d g
d d d d d d d d d d g
d d d d d d d d d d g
g � f Vt j d t � } t | d � t j d d d g � } t t j | d k � | | | | f f Vd S(
Ni i i i
i i i ( i i ( i
i ( i
i ( i i
( i i i i ( R t zerosR[ R
R R R` R t where( R4 t i( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyt test_fill_diagonal� sT
'!,
c
c` s$ t d � } t j d d d d g d d d d g d d
d d g d
d d d g g � } d | | <t | t j d d d d g d d d d g d d
d d g d
d d d g g � f Vt d d � } t j d t � } d | | <t | t j d d g d d g g d d g d d g g g � f Vd S( Ni i i i i i i i i i
i i i
i i i id i ( i i i ( R R R R Rn R[ ( t diR4 t d3( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyt test_diag_indices s&
c C` sT t j j d � } t | � \ } } t | t j d � � t | t j d � � d S( Ni ( i i ( R RF R R RL ( Ri t rR= ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyt test_diag_indices_from& s c C` s t t d d d � � } g t t j d � � D] \ } } | ^ q. } t | | � t t d � � } t | | � t t d � � } t | t t d � � � t t � � } t | d g � t t d � � } t | d
g � t t d g � � } t | g � d S( Ni i i i ( i i i ( i i i ( i ( ( ( ( RJ R R R Rn R R ( Ri t ixt et expected( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyt test_ndindex- s .
t __main__(& t
__future__R R R t numpyR t
numpy.testingR R R R R R R R
t numpy.lib.index_tricksR R R
R R R R R R R R R3 R? RI RK RO Rm Rq Rt Rv Rz R1 ( ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_index_tricks.pyt <module> s :FL! % 5
Mini Shell