File: //opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pyc
�
� 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 m Z m
Z
m Z m Z d d l
m Z m Z d d l m Z d d l m Z e j d d k r� e Z n d Z d Z e e g 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 d � Z- d � Z. d � Z/ d � Z0 e1 d k r�e � n d S( i ( t divisiont absolute_importt print_functionN( t run_module_suitet assert_t
assert_raisest assert_equal( t solve_diophantinet internal_overlap( t
as_strided( t longi i i
i����c C` s� | d k r t d g Sg } xh d D]` } xW d D]O } t d | | | d } | | | | } | j t | | | | � � q1 Wq$ W| S( sG Returns slices of length nelems, from start onwards, in direction sign.i i i i����( i i ( i����i ( t sizet appendt slice( t nelemst rest stept signt startt stop( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pyt _indices_for_nelems s
%c C` sZ g } xM d D]E } t | � } | j g | D] } | D] } | | f ^ q6 q, � q
W| S( s$ Returns (src, dst) pairs of indices.i i i ( i i i ( R t extend( R R t indt at b( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pyt _indices_for_axis) s
7c C` so t � } g g } xV t | � D]H } g } x3 | D]+ } x" | D] } | j | g | � q? Wq2 W| } q W| S( sK Returns ((axis0_src, axis0_dst), (axis1_src, axis1_dst), ... ) index pairs.( R t rangeR ( t ndimsR R t it newrest elemt others( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pyt _indices6 s
c C` ss t j t j t � � j t � } | j � } | | | | <| | | | <t t j | | k � d | | f � d S( s1 Check assignment arr[dstidx] = arr[srcidx] works.s assigning arr[%s] = arr[%s]N( t npt aranget productt shapet reshapet copyR t all( t srcidxt dstidxt arrt cpy( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pyt _check_assignmentH s ! c c` su t t � } xb | D]Z } t g | D] } | d ^ q# � } t g | D] } | d ^ qF � } t | | f Vq Wd S( sA Test automatically generated assignments which overlap in memory.i i N( R R t tupleR, ( t indsR R R( R) ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pyt test_overlapping_assignmentsV s
##c ` s! t j j d � � t j t j � j } x�t d � D]�} d } d } d | d } g } x�t | | � | k rd � j d d d t j �d � � j d d d t j �d � t | � � � t | d � � � t � � f d � t | � D� � } t � � f d
� t | � D� � } t | d t
d � t | | � D� � � } � j d
| d d t j �� | d k r�| | k r�d � n t | | � � } | d k r�t | | � d d �}
t |
d k | | � |
f � y# t d � t | | � D� � } Wn t k
rq] n Xd } x | D] }
| t |
� 9} q(W| d k rt t � f d � t j | � D� � � | d 7} qq] t | | � d d �}
t |
d k | | � |
f � t t
d � t | | � D� � � k � t t d � t | | � D� � � | d 7} q] Wq4 Wd S( Ni� i
i i� i i t dtypei c 3` s4 | ]* } t � j d � d d t j �� Vq d S( i R0 N( t intt randintR! t intp( t .0t j( t A_maxt rng( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pys <genexpr>w s c 3` s4 | ]* } t � j d � d d t j �� Vq d S( i i R0 N( R1 R2 R! R3 ( R4 R5 ( t U_maxR7 ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pys <genexpr>y s i c s` s | ] \ } } | | Vq d S( N( ( R4 R t ub( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pys <genexpr>| s i����t simplifyc s` s/ | ]% \ } } t d | | d | � Vq d S( i i N( t xrange( R4 R R9 ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pys <genexpr>� s i�� c 3` s! | ] } t | � � k Vq d S( N( t sum( R4 t w( R ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pys <genexpr>� s c s` s | ] \ } } | | Vq d S( N( ( R4 R t x( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pys <genexpr>� s c s` s1 | ]' \ } } d | k o& | k n Vq d S( i N( ( R4 R> R9 ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pys <genexpr>� s ( R! t randomt RandomStatet iinfoR3 t maxR t minR2 R- R< t zipR t NoneR t
OverflowErrort lent anyt itertoolsR# R' ( t max_intt ndimt feasible_countt infeasible_countt min_countt numberst At Ut b_ubt Xt X_simplifiedt rangesR t r( ( R6 R8 R R7 sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pyt test_diophantine_fuzzb sN # , #
* )#c C` s� t j t j � j } t j t j � j } | | k r� | d | d d f } | d | d d f } d | d d } t t | | | � d � n d S( Ni i
i ( i i ( R! RA R3 RB t int64R R ( t max_intpt max_int64RP RQ R ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pyt test_diophantine_overflow� s c
C` s' t j | | d t �} t t j | | � t j | | d t �� | j d � | j d � | j d � | j � } d } | | k rd d j d | j d d | j d d f d | j f d
| j f d | j
f d | j
f d
| j f d | j f g � } n t | | d | �d S( Nt max_worki i t s s
s base_a - base_b = %rt datas shape_a = %rs shape_b = %rs strides_a = %rs strides_b = %rs size_a = %rs size_b = %rt err_msg( R! t may_share_memoryt MAY_SHARE_EXACTR t MAY_SHARE_BOUNDSt fillRH t joint __array_interface__R$ t stridesR ( R R t gott exactR_ ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pyt check_may_share_memory_exact� s$
$
c C` s� t j d d d d g d t j �t j d d d d g d t j �d d � d d � d d d � d d � f g } g } x^ | D]V } xM t j t d � t d d d � g f d � D] } | | } | j | � q� Wq� Wx�| D]x} t t j | d d � d d d � f | d d � d
d d � f � � t t j | d d � d d d � f | d d � d
d d � f d d �� t
| d d � d d d � f | d d � d
d d � f � t
| d d � d d d � f | d d � d
d d
� f � y4 | j � } | j j
r w� n | j t j � } Wn t k
rEq� n Xt
| j � d d !| j d d d d � d d � d d d � f � t
| d d � d d d � f | j d d d d � � t
| d d � d d d � f | j d d d d � d d � d
d d
� f � t
| j � d d !| j d d d d � d d � d d d � f � q� Wt j d
g d t j �} t
| | � t
| | j � � d S( Ni
i i i R0 i i����i i i R\ i i i i i. ( R! t zerost int8RI R# R
RE R R R` Ri t ravelt flagst owndatat viewt int16t
ValueErrorR% R& ( t xs0t xsR> t sst xp( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pyt test_may_share_memory_manual� s< !R
5
HN??
2 26
c ` s] t j d d d d g d t j �} t j j d � � � f d � � d } d } xt | | � | k rXt � f d � t | j � D� � } | r� | } n% t � f d
� t | j � D� � } t j | j � } � j
| � t j | j � } � j
| � t � f d � t | j | � D� � }
t � f d � t | j | � D� � } | |
j
| � } | | j
| � }
t j | |
� } t j | |
� } t j | |
d
| | |
� �} t j | |
d
t �} | | k rt | | d t |
| f � �n | | k r/t | | d t |
| f � �n | rT | rH| d 7} qU| d 7} qT qT Wd S( Ni i" iG ia R0 i� c ` s� � j d | d d t j �} � j | | d d t j �} � j d d d t j �d k ry | | } } | d 9} n t | | | � S( Ni i R0 i i����( R2 R! R3 R
( t nR R R ( R7 ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pyt random_slice s !
i c 3` sQ | ]G } � j d d d t j �d k rE � j d d d t j �n d Vq d S( i i R0 i i N( R2 R! R3 ( R4 R5 ( R7 ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pys <genexpr> s c 3` sQ | ]G } � j d d d t j �d k rE � j d d d t j �n d Vq d S( i i R0 i i N( R2 R! R3 ( R4 R5 ( R7 ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pys <genexpr> s c 3` s$ | ] \ } } � | | � Vq d S( N( ( R4 t pt s( Rx ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pys <genexpr>" s c 3` s$ | ] \ } } � | | � Vq d S( N( ( R4 Ry Rz ( Rx ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pys <genexpr># s R\ R_ i ( R! Rj Rp R? R@ RC R- R RK R" t shuffleRD R$ t transposeR` Ra R t repr( t get_max_workt
same_stepsRN R> t feasiblet
infeasiblet stepst steps2t t1t t2t s1t s2R R t bounds_overlapt may_share_answert easy_answert exact_answer( ( Rx R7 sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pyt check_may_share_memory_easy_fuzz� sB $
((! "
c C` s t d d � d t d d � d S( NR~ c S` s d S( Ni ( ( R R ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pyt <lambda>? s R RN i� ( R� t True( ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pyt test_may_share_memory_easy_fuzz; s c C` s t d d � d t d d � d S( NR~ c S` s t | j | j � d S( Ni ( RB R ( R R ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pyR� L s R RN i� ( R� t False( ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pyt! test_may_share_memory_harder_fuzzD s c C` s/ t j d d d g d t j �} t t j | | � t � t t j | | j � � t � | d d � d d d � d d d � f } | d d � d d d � d d d � f } t t j | | � t � t t j | | d d �t � t t j
t j | | d d �t t j
t j | | d t d � �d S( Ni i i R0 i i R\ i ( R! Rj Rk R t
shares_memoryR� R& R� RE R t TooHardErrorR
( R> R R ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pyt test_shares_memory_apiQ s ! ++ c C` sB d d � } | d d d f d t �| d d d f d t �d S( Nc S` s t | | d d d �} | d k r3 | d k } n | d k r� t t d � t | | � D� � t d � t | | � D� � k � t t d � t | | � D� � � t t d � t | | � D� � � n | r� t | d k t | � � n t | d k t | � � d S( Ni t require_ub_nontriviali c s` s | ] \ } } | | Vq d S( N( ( R4 R R> ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pys <genexpr>g s c s` s# | ] \ } } | | d Vq d S( i N( ( R4 R t u( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pys <genexpr>g s c s` s1 | ]' \ } } d | k o& | k n Vq d S( i N( ( R4 R> R� ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pys <genexpr>h s c s` s% | ] \ } } | | d k Vq d S( i N( ( R4 R> R� ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pys <genexpr>i s ( R RE R R< RD R' RH R} ( RP RQ t existsRS ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pyt check` s B#& i i R� i i ( i i i i i ( i i i i i ( RE R� R� ( R� ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pyt! test_internal_overlap_diophantine_ s c ` s t j d d d d g d t j �} t j j d � � � f d � � d } d } x� | | k r� t � f d
� t | j � D� � } t j | j � } � j | � t � f d � t
| j | � D� � } | | j | � } t
t | � � | d 7} qT Wd S(
Ni i" iG ia R0 i� c ` s� � j d | d d t j �} � j | | d d t j �} � j d d d t j �d k ry | | } } | d 9} n t | | | � S( Ni i R0 i i����( R2 R! R3 R
( Rw R R R ( R7 ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pyRx | s !
i i� c 3` sQ | ]G } � j d d d t j �d k rE � j d d d t j �n d Vq d S( i i R0 i i N( R2 R! R3 ( R4 R5 ( R7 ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pys <genexpr>� s c 3` s$ | ] \ } } � | | � Vq d S( N( ( R4 Ry Rz ( Rx ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pys <genexpr>� s i ( R! Rj Rp R? R@ R- R RK R" R{ RD R$ R| R R ( R> t casesRN R� R� R� R ( ( Rx R7 sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pyt test_internal_overlap_slicesu s $
(c C` s� t | � } t � } t d � | j D� � } xb t j | � D]K } t d � t | j | � D� � } | | k r| t } Pq> | j
| � q> Wt } | | k r� t | | d t
| j | j f � �n | d k r� | | k r� t | | � n | S( Nc s` s | ] } t | � Vq d S( N( R; ( R4 Rw ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pys <genexpr>� s c s` s | ] \ } } | | Vq d S( N( ( R4 Rz R= ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pys <genexpr>� s R_ ( R t setR- R$ RI R# R< RD Rf R� t addR� R R} RE ( R t manual_expectedRg t mRU t vt offsett expected( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pyt check_internal_overlap� s " (c C` sj t j d � j t j � } t | t � t | j g � t � t | d d d d
�} t | t � t | d d d d �} t | t � t | d d d d �} t | t � t | d d d d �} t | t � t | d d d d �} t | t � t | d d d d �} t | t � t | d d d d �} t | t � t | d d d d �} t | t � d S( Ni Rf i i R$ i i i i���iW i ( i i ( i i ( i i ( i i ( i ( i ( i ( i ( i ( i ( i i���( iW i ( i i���( i i ( i i���( i i ( R! R" t astypeRk R� R� R% R R� ( R> R ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pyt test_internal_overlap_manual� s&
c ` s� t j d � j t j � } d } d } d } t j j d � � x� t | | � | k r� � j d d d t j �} t � f d � t
| � D� � } t � f d � t
| � D� � } t | d | d
| �} t | � } | r� | d 7} qB | d 7} qB Wd S( Ni i id i� i R0 c 3` s* | ] } � j d d d t j �Vq d S( i���i� R0 N( R2 R! R3 ( R4 R5 ( R7 ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pys <genexpr>� s c 3` s* | ] } � j d d d t j �Vq d S( i i R0 N( R2 R! R3 ( R4 R5 ( R7 ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pys <genexpr>� s Rf R$ (
R! R" R� Rk R? R@ RC R2 R3 R- R R R� ( R> t overlapt
no_overlapRN RK Rf R$ R t result( ( R7 sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pyt test_internal_overlap_fuzz� s
c C` s/ d t f d � � Y} d t f d � � Y} x� | | g D]� } t j d � } t t j | | d d d � � | d d d � � � t t j | | d d d � � | d d d � � � t t j | | d d d � � | d d d � � � t t j | | d d d � � | d d d � � � q9 Wd S( Nt MyArrayc B` s e Z d � Z e d � � Z RS( c S` s
| | _ d S( N( R^ ( t selfR^ ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pyt __init__� s c S` s
| j j S( N( R^ Re ( R� ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pyRe � s ( t __name__t
__module__R� t propertyRe ( ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pyR� � s t MyArray2c B` s e Z d � Z d � Z RS( c S` s
| | _ d S( N( R^ ( R� R^ ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pyR� � s c S` s | j S( N( R^ ( R� ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pyt __array__� s ( R� R� R� R� ( ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pyR� � s i i i i ( t objectR! R" R R` R� ( R� R� t clsR> ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pyt test_non_ndarray_inputs� s 676t __main__(2 t
__future__R R R t sysRI t numpyR! t
numpy.testingR R R R t numpy.core.multiarray_testsR R t numpy.lib.stride_tricksR t numpy.compatR
t version_infoR R; R R R- R$ Rb Ra R R R R, R/ RW R[ Ri Rv R� R� R� R� R� R� RE R� R� R� R� R� ( ( ( sT /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_mem_overlap.pyt <module> sF "
A 3 <
%