U
|
O ±d ã @ sz d dl Z d dlZd dlZd dlZd dlZd dlm mZ d dlm Z m
|
Z
|
m Z m Z d dl mZmZmZmZmZmZmZmZ dd Zdd Zdd Zejje d
|
d d d Zdd Z dd Z dd Z dd Z dd Z ejje d
|
d dd Z!dd Z"d d Z#d d Z$d d! Z%d"d# Z&d$d% Z'd&d' Z(d(d) Z)d*d+ Z*d,d- Z+d.d/ Z,d0d1 Z-d2d3 Z.d4d5 Z/d6d7 Z0d8d9 Z1d:d; Z2d<d= Z3d>d? Z4d@dA Z5dBdC Z6dDdE Z7dFdG Z8dHdI Z9dJdK Z:dLdM Z;dNdO Z<dPdQ Z=dRdS Z>dTdU Z?ej @dVejAdW ¡ej @dXejAdW ¡ej BdY¡dZd[ ZCd\d] ZDd^d_ ZEd`da ZFdbdc ZGddde ZHdfdg ZIdhdi ZJdjdk ZKdldm ZLdndo ZMdpdq ZNdrds ZOdtdu ZPdvdw ZQdxdy ZRdzd{ ZSd|d} ZTd~d ZUdd ZVdd ZWdd
ZXdd ZYdd ZZdd Z[ejjedd ejje dd dd Z\dd Z]dd Z^dd Z_dd Z`ejjadd Zbdd ZcG dd dZddd Zed d¡ Zfd¢d£ Zgd¤d¥ Zhd¦d§ Zid¨d© Zjej @dªejkd«d¬dejkd®d¬dddd¯
dd
f ejkd°d¬dddd¯
dd
ddd¯
dd
f ejkd±d¬dddd²
ejkd³d¬dddd¯
ddd´
f ejkd°d¬dddd¯
dd
ddd¯
dddµ
f g¡d¶d· Zlej @d¸d¹gddµd gfejkdºd»dd d¼gfejkdºd»ddfejkd½d»ddµd gfejkd¾d»ddµdµgfejkd¾d»ddfg¡d¿dÀ ZmdÁd ZndÃdÄ ZodÅdÆ ZpdÇdÈ ZqdÉdÊ ZrdËdÌ ZsdÍdÎ ZtdÏdÐ Zuej @dÑejvdÒewddd
ddd¯
dd
f ejvdÒewdÓdÔdd
ddd¯
dd
f ejvdÒewddd
ddd¯
dd
f xdÓ¡g¡dÕdÖ Zyd×dØ ZzdÙdÚ Z{dÛdÜ Z|dÝdÞ Z}dßdà Z~dádâ Zejje d
|
d dãdä Zejje jdd¯
dåko e jdækdçd ej @dèdégdêdëdìdíg¡ej @dîd¼d¯d²g¡dïdð Zejje d
|
d ej @dèdégdëdíg¡dñdò Zdódô Z
dS )õé N)ÚarrayÚarangeÚnditerÚall)Úassert_Ú assert_equalÚassert_array_equalÚ assert_raisesÚIS_WASMÚ HAS_REFCOUNTÚsuppress_warningsÚ break_cyclesc C s$ g }| j s | | j¡ | ¡ q|S ©N)ÚfinishedÚappendÚ multi_indexÚiternext©ÚiÚret© r úSd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/core/tests/test_nditer.pyÚiter_multi_index s
|
|
r c C s$ g }| j s | | j¡ | ¡ q|S r )r r Úindexr r r r r Ú iter_indices s
|
|
r c C s$ g }| j s | | j¡ | ¡ q|S r )r r Ú iterindexr r r r r Úiter_iterindices s
|
|
r zPython lacks refcounts)Úreasonc C sr t d} t d¡ ¡ }t | ¡}t |¡}t| g ddggd|gd6}t|j tt | ¡|k tt |¡|k W 5 Q R X d }t t | ¡| t t |¡| t ddd} t d¡}t | ¡}t |¡}t| g dgg|gd}t | ¡}t |¡}|
|
¡ }tt | ¡|k tt |¡|k d }t t | ¡| t t |¡| d }t t | ¡| t t |¡| ~d S ) Né Úf4Ú readwriteÚ updateifcopyÚunsafe©ÚcastingÚ op_dtypes©Údtype©r% ) r Únpr' Ú newbyteorderÚsysÚ getrefcountr r Úiterationneedsapir Úcopy)ÚaÚdtZrc_aZrc_dtÚitZrc2_aZrc2_dtÚit2r r r Útest_iter_refcount% sH
|
|
ý
|
|
|
þ
|
|
r3 c C s dD ] } t t | ¡}tdt| D ]Ü}td gt| }tt| D ] }d| |@ rJtd d d||< qJt|}| | ¡| }t|g dgg}t dd |D | t|j
|
g dgg}t dd |D | t| dkr(t| dd ¡g dgg}t d
|
d |D | q(qd S ) N©)é ©é é ©é r7 r8 )r: r7 r8 r7 )r: r7 r: r: r7 r: éÿÿÿÿÚreadonlyc S s g | ]}|qS r r ©Ú.0Úxr r r Ú
|
<listcomp>b s z(test_iter_best_order.<locals>.<listcomp>c S s g | ]}|qS r r r= r r r r@ e s r é c S s g | ]}|qS r r r= r r r r@ i s ) r r) ÚprodÚrangeÚlenÚsliceÚtupleÚreshaper r ÚTÚswapaxes©Úshaper/ ÚdirsÚ
|
dirs_indexÚbitÚaviewr r r r Útest_iter_best_orderP s
|
rP c C s" dD ]} t t | ¡}tdt| D ]ò}td gt| }tt| D ] }d| |@ rJtd d d||< qJt|}| | ¡| }t|dd}t dd |D |j
|
dd t|j dd}t dd |D |j j
|
dd t| dkr(t| d d
|
¡dd}t d d |D | d d
|
¡j
|
dd q(qd S ) Nr4 r: r; ÚC©Úorderc S s g | ]}|qS r r r= r r r r@ | s z%test_iter_c_order.<locals>.<listcomp>c S s g | ]}|qS r r r= r r r r@ s r rA c S s g | ]}|qS r r r= r r r r@ s © r r) rB rC rD rE rF rG r r ÚravelrH rI rJ r r r Útest_iter_c_orderk s$
|
ÿrV c C s" dD ]} t t | ¡}tdt| D ]ò}td gt| }tt| D ] }d| |@ rJtd d d||< qJt|}| | ¡| }t|dd}t dd |D |j
|
dd t|j dd}t dd |D |j j
|
dd t| dkr(t| d d
|
¡dd}t d d |D | d d
|
¡j
|
dd q(qd S ) Nr4 r: r; ÚFrR c S s g | ]}|qS r r r= r r r r@ s z%test_iter_f_order.<locals>.<listcomp>c S s g | ]}|qS r r r= r r r r@ s r rA c S s g | ]}|qS r r r= r r r r@ s rT rJ r r r Útest_iter_f_order s$
|
ÿrX c C s" dD ]} t t | ¡}tdt| D ]ò}td gt| }tt| D ] }d| |@ rJtd d d||< qJt|}| | ¡| }t|dd}t dd |D |j
|
dd t|j dd}t dd |D |j j
|
dd t| dkr(t| d d
|
¡dd}t d d |D | d d
|
¡j
|
dd q(qd S ) Nr4 r: r; ÚArR c S s g | ]}|qS r r r= r r r r@ ² s z*test_iter_c_or_f_order.<locals>.<listcomp>c S s g | ]}|qS r r r= r r r r@ µ s r rA c S s g | ]}|qS r r r= r r r r@ ¹ s rT rJ r r r Útest_iter_c_or_f_order¡ s$
|
ÿrZ c C sH t d¡ dd¡} t j| dgd}d|_tdd |D ddd d
|
g d S ) Nr r: r7 r ©Úflags©r r: c S s g | ]}|qS r r )r> r r r r r@ Ä s z/test_nditer_multi_index_set.<locals>.<listcomp>r8 r5 )r) r rG r r r )r/ r1 r r r Útest_nditer_multi_index_set¼ s r^ c C sL d} t jt ddddg¡dgd}t | ¡}| f|_t | ¡}t|| d S )Nr éo éÞ iM i¼ r r[ )r) r r r+ r, r r )r r Z start_countZ end_countr r r Ú$test_nditer_multi_index_set_refcountÆ s
|
|
ra c C sf t d} t| dgdgg}tt|ddddg t| d d d
dgdgg}tt|ddddg d S ) Nr8 r r< ©r ©rA ©r: ©r7 r; )r r r r ©r/ r r r r Ú#test_iter_best_order_multi_index_1dÓ s
|
rg c C s, t d} t| dd¡dgdgg}tt|dddd d
|
d g t| dd¡jd d dgdgg}tt|dd dd
|
dd g t| dd¡d d d
dgdgg}tt|d d
|
d dddg t| dd¡d d
d d d
f dgdgg}tt|dddd d
|
d g t| dd¡d d d
d d d
f dgdgg}tt|d d
|
d dddg t| dd¡jd d d d d
dgdgg}tt|d dd
|
dd dg t| dd¡jd d d d
d d d
f dgdgg}tt|dd dd
|
dd g t| dd¡jd d d d d
d d d
f dgdgg}tt|d dd
|
dd dg d S )Nr r: r7 r r< ©r r ©r rA r] ©rA r ©rA rA ©rA r: rW rR r; ©r r rG r r r. rf r r r Ú#test_iter_best_order_multi_index_2dÞ s. "$,.,& ÿ( ÿrn c C s¤ t d} t| ddd¡dgdgg}tt|dddd d
|
d d d ddddg t| ddd¡jdddgdgg}tt|dd ddd
|
ddd d dd dg t| ddd¡d d d
dgdgg}tt|d d dddddddd d
|
d g t| ddd¡d d
d d d
f dgdgg}tt|d
|
d dd ddddddd d g t| ddd¡d d
d d
d d d
f dgdgg}tt|ddd dd d
|
d d ddddg t| ddd¡jddd d d
dgdgg}tt|d ddddd
|
d ddd dd g t| ddd¡jddd d
d d d
f dgdgg}tt|d
|
ddddd d dd ddd g t| ddd¡jddd d
d d
d d d
f dgdgg}tt|dd d dd ddd ddd
|
dg d S )Né r: r7 r r< )r r r )r r rA )r rA r )r rA rA )r r: r )r r: rA )rA r r )rA r rA )rA rA r )rA rA rA )rA r: r )rA r: rA rW rR r; rm rf r r r Ú#test_iter_best_order_multi_index_3dù sÄ ÿÿ$ ÿÿ& ÿÿ. ÿÿ4 ÿÿ ÿ ÿÿ( ÿ ÿÿ. ÿ ÿÿrp c C sf t d} t| dgdgg}tt|ddddg t| d d d
dgdgg}tt|ddddg d S ) Nr8 Úc_indexr< r rA r: r7 r; ©r r r r rf r r r Ú test_iter_best_order_c_index_1d% s
|
rs c C s, t d} t| dd¡dgdgg}tt|dddddd g t| dd¡jd
|
d dgdgg}tt|dddddd g t| dd¡d d d
dgdgg}tt|ddd dddg t| dd¡d d
d d d
f dgdgg}tt|dddd ddg t| dd¡d d d
d d d
f dgdgg}tt|d dddddg t| dd¡jd
|
d d d d
dgdgg}tt|ddddd dg t| dd¡jd
|
d d d
d d d
f dgdgg}tt|dd ddddg t| dd¡jd
|
d d d d
d d d
f dgdgg}tt|d dddddg d S ) Nr r: r7 rq r< r rA r8 r5 rW rR r; ©r r rG r r r. rf r r r Ú test_iter_best_order_c_index_2d0 s: ÿ$,. ÿ& ÿ( ÿru c C s¤ t d} t| ddd¡dgdgg}tt|dddddd d
|
d d d ddg t| ddd¡jdddgdgg}tt|dd
|
dd dddd dd d dg t| ddd¡d d d
dgdgg}tt|d
|
d d d dddddddd g t| ddd¡d d
d d d
f dgdgg}tt|dd ddddddd d d
|
d g t| ddd¡d d
d d
d d d
f dgdgg}tt|ddddd dd d
|
d d ddg t| ddd¡jddd d d
dgdgg}tt|d
|
dd dddd dd ddd g t| ddd¡jddd d
d d d
f dgdgg}tt|dddd dd
|
d ddd dd g t| ddd¡jddd d
d d
d d d
f dgdgg}tt|dd dd d ddd
|
dd ddg d S )Nro r: r7 rq r< r rA r8 r5 r é é é é
|
é rW rR r; rt rf r r r Ú test_iter_best_order_c_index_3dM sZ ÿ ÿÿ&ÿ.ÿ4ÿ ÿÿ( ÿÿ. ÿÿr{ c C sf t d} t| dgdgg}tt|ddddg t| d d d
dgdgg}tt|ddddg d S ) Nr8 Úf_indexr< r rA r: r7 r; rr rf r r r Ú test_iter_best_order_f_index_1dr s
|
r} c C s, t d} t| dd¡dgdgg}tt|dddddd g t| dd¡jd
|
d dgdgg}tt|dddddd g t| dd¡d d d
dgdgg}tt|ddd dddg t| dd¡d d
d d d
f dgdgg}tt|dddd ddg t| dd¡d d d
d d d
f dgdgg}tt|d dddddg t| dd¡jd
|
d d d d
dgdgg}tt|ddddd dg t| dd¡jd
|
d d d
d d d
f dgdgg}tt|dd ddddg t| dd¡jd
|
d d d d
d d d
f dgdgg}tt|d dddddg d S ) Nr r: r7 r| r< r r8 rA r5 rW rR r; rt rf r r r Ú test_iter_best_order_f_index_2d} s: ÿ$,. ÿ& ÿ( ÿr~ c C s¤ t d} t| ddd¡dgdgg}tt|ddddd d
|
d d dd ddg t| ddd¡jdddgdgg}tt|dd ddd ddd dd d
|
dg t| ddd¡d d d
dgdgg}tt|d d dd ddddddd d
|
g t| ddd¡d d
d d d
f dgdgg}tt|d d
|
dddddddd d d g t| ddd¡d d
d d
d d d
f dgdgg}tt|ddddd
|
d d d d dddg t| ddd¡jddd d d
dgdgg}tt|d ddddd d dd ddd
|
g t| ddd¡jddd d
d d d
f dgdgg}tt|d ddddd d
|
ddd dd g t| ddd¡jddd d
d d
d d d
f dgdgg}tt|dd dd d
|
ddd ddd dg d S )Nro r: r7 r| r< r r rw r8 ry rA rv rx r5 rz rW rR r; rt rf r r r Ú test_iter_best_order_f_index_3d sZ ÿ ÿÿ&ÿ.ÿ4ÿ ÿÿ( ÿÿ. ÿÿr c C s6 dD ]*} t | ¡}t|}tdt| D ] }td gt| }tt| D ] }d| |@ rPtd d d||< qPt|}| | ¡| }t|dgdgg}t |j
|
d t |d j |f t|j dgdgg}t |j
|
d t |d j |f t| dkr,t| dd¡dgdgg}t |j
|
d t |d j |f q,qd S )Nr4 r: r; Ú external_loopr< rA r )r) rB r rC rD rE rF rG r r ÚndimrK rH rI )rK Úsizer/ rL rM rN rO r r r r Ú test_iter_no_inner_full_coalesce¿ s.
|
|
ÿ r c C sV t d ddd¡d d
d d
d d
f } t| dgdgg}t|jd t|d jd t d ddd¡d d
d d
d d
f } t| dgdgg}t|jd t|d jd
|
t d ddd¡d d
d d
d d
f } t| dgdgg}t|jd t|d jd t d d d dd d dd d dd d ¡ } t| dgdgg}t|jd t|d jd d S )Né r: r7 r8 r; r r< r re ©rw rA )ro ©r )r rG r r r rK rf r r r Ú!test_iter_no_inner_dim_coalescingÝ s ( ( ( " r c C s t d ddd¡} t| dgdgg}t|jd t d ddd¡}t|dgdgg}t|jd t| d d¡dgdgg}t|jd t|jdgdgg}t|jd t|jd
|
gdgg}t|jd t|j d d¡d
|
gdgg}t|jd t d ddd¡}t|d d }t|jd t|jd d }t|jd t|d d }t|jd t|jd d }t|jd t|dd }t|jd t|jdd }t|jd d S )Nr r: r7 r8 r r< rq rA r r| rQ rR rW rY )r rG r r r rI rH )r/ r Za3dr r r Útest_iter_dim_coalescingõ s6 r c C sþ t tdt d¡gdgdggd } t| jd t| jd t td dd¡t d¡gdgdggd } t| jd t| jd t td dd¡tdgdgdggd } t| jd t| jd t td dd¡tdgdgdggd } t| jd t| jd t td dd¡td dd¡gdgdggd } t| jd t| jd t t d¡td d
|
dd¡gdgdggd } t| jd t| jd t tdtd d
|
dd¡gdgdggd } t| jd t| jd t tdtd d
|
dd¡gdgdggd } t| jd t| jd t td dd¡td d
|
dd¡gdgdggd } t| jd t| jd t td dd¡td d
|
dd¡gdgdggd } t| jd t| jd t td dd¡td d
|
dd¡gdgdggd } t| jd t| jd t td ddd¡td ddd¡td
|
d
|
dd¡gdgdggd } t| jd t| jd t td ddd¡td
|
d
|
dd¡gdgdggd } t| jd t| jd t td d
|
dd¡td d
|
dd¡gdgdggd } t| jd t| jd d S )Nr r: r r< ©r r7 ©r: r7 rA r r8 ©r8 r: r7 rw ro )r r r) Úint32r ÚitersizerK rG ©r r r r Útest_iter_broadcasting s¦ $
|
ÿ
|
ÿ
|
ÿ
|
ÿ
|
ÿ
|
ÿ
|
ÿ "
|
ÿ "
|
ÿ "
|
ÿ "ÿ
|
þ $
|
ÿ $
|
ÿ r c C sR t jddd dd¡} t| d gg dgddggd d
|
d gd gd d }t|jd
|
jd t|jd
|
jd t| jd gg dgddggd d
|
d gd gd d }t|jd
|
jd t|jd
|
jd t| jd gg dgddggdd d
|
d gd gd d}t|jd
|
jd t|jd
|
jd t t
|
t| d gg dgddggd d
|
d gd gdd t jt d¡d d gdd}d S )Nr Úi2r& r: r7 r< Ú writeonlyÚallocater rA )r; r; r8 ©Úop_axesÚ itershaper9 )r rw r: )r7 r: r8 )rw r r: rW )rS r r )r: r ro )r; rA r8 rd ©r ) r) r rG r r ÚoperandsrK ÚstridesrH r Ú
|
ValueErrorÚonesrf r r r Útest_iter_itershape_ s4 þ þ ý ýr c
|
C s* t tttdtdgg dggd t tttd dd¡tdgg dggd t tttd dd¡td dd¡gg dggd t tttd dd¡td dd¡gg dggd t tttd ddd¡td ddd¡gg dggd t tttd ddd
|
¡td ddd¡gg dggd zRttd d
|
dd
|
¡td d
|
d¡td dd¡gg dgdgd d gg td W n\ tk
|
rÖ } z<t| }t| d¡dkd| t| d¡dkd| W 5 d } ~ X Y nX zHttd dd¡tdgg dgdggdd
|
gdt j
|
ggdd td W nt tk
|
r } zTt| }t| d¡dkd| t| d¡dkd| t| d¡dkd| W 5 d } ~ X Y nX z2tt d¡t d ¡gg d d gdgg td W n\ tk
|
r$ } z<t| }t| d ¡dkd | t| d ¡dkd | W 5 d } ~ X Y nX d S )!Nr: r7 r< r rx r8 é$ r rw rA r Ú no_broadcastz$Should have raised a broadcast errorz(2,3)r z0Message "%s" doesn't contain operand shape (2,3)z(1,2,3)z4Message "%s" doesn't contain broadcast shape (1,2,3))r8 r7 r z (2,3)->(2,3)z7Message "%s" doesn't contain operand shape (2,3)->(2,3)z(2,)->(2,newaxis)zDMessage "%s" doesn't contain remapped operand shape(2,)->(2,newaxis)z(4,3)z6Message "%s" doesn't contain itershape parameter (4,3)©r: rA rA rd z(2,1,1)z2Message "%s" doesn't contain operand shape (2,1,1)z(2,1,2)z8Message "%s" doesn't contain the broadcast shape (2,1,2)) r r r r rG ÚAssertionErrorÚstrr Úfindr) ÚnewaxisÚzeros)ÚeÚmsgr r r Ú test_iter_broadcasting_errors~ s¢
|
ÿ
|
þ
|
þ
|
þ"
|
þ"
|
þþü ÿÿ
|
ü ÿÿÿÿ þ ÿÿr¦ c C sö t d} tttg g g ttt| gd g dggd ttt| gdgdgg ttt| gg ddgg ttt| gg dggdd ttt| gg dggdd ttt| gd g dggd
|
ttt| d d gdgg ttt| d dgdgg ttt| d d gdgg ttt| d d gdgg ttt| g g g ttt| g ddgg ttt| g ddgg ttt| g ddgg ttt| g dddgg tttdg dgg tttdg dgg tttt d¡g dgg tttt d¡g dgg d| j_ttt| g dgg ttt| g dgg d| j_tt dg dgg}ttdd | ttdd | dd }dd }d d }d d }tt dd g}tt|| tt|| tt|| tt|| tt dd g}tt|| tt|| tt|| tttt g ¡ d S )!Nr éd r< zbad flagÚGrR Znoon©r$ r7 r: rq r| r r r r ç ø?rA FTc S s | j S r ©r r r r r Ú<lambda>ú ó z(test_iter_flags_errors.<locals>.<lambda>c S s | j S r ©r r r r r r¬ ü r c S s
|
d| _ d S )Nrb r« r r r r Úassign_multi_indexÿ s z2test_iter_flags_errors.<locals>.assign_multi_indexc S s
|
d| _ d S ©Nr r® r r r r Ú assign_index s z,test_iter_flags_errors.<locals>.assign_indexc S s
|
d| _ d S r° )r r r r r Úassign_iterindex s z0test_iter_flags_errors.<locals>.assign_iterindexc S s
|
d| _ d S )Nri )Ú iterranger r r r Úassign_iterrange s z0test_iter_flags_errors.<locals>.assign_iterrangeÚbuffered)
|
r r r r Ú TypeErrorr) r r\ Ú writeabler )r/ r r¯ r± r² r´ r r r Útest_iter_flags_errorsÉ sr ÿ ÿ ÿ ÿ
|
ÿ r¸ c C s¤ t d¡t d¡t d¡ } }}t| ||gg dg}|` d|dd
< t| dddg t|dddg t|dddg d|d< t|dd
ddg W 5 Q R X d S ) Nr7 ç @r )r7 r7 r r: rA ro )r) r r r )r/ ÚbÚcr r r r Útest_iter_slice s " r¼ c C sâ t jddd ddd¡j} t j| g ddggd t d
|
¡gd }|" d|jd d < d|jd d < W 5 Q R X t| d t j| g ddggd t d
|
¡gd }|. |jd dd
}d|d < d|jd d < W 5 Q R X t| d d }d }d S )Nr Úf8r& r: r7 r8 r r! Ú same_kindr r# r .é r; rA i.ûÿÿ)r) r rG rH r r' r r )r/ r1 r? r r r Útest_iter_assign_mapping! s(
|
ÿ
|
|
ÿ
|
rÀ c C sö t jddd} | ¡ ¡ }t| jj|jjk t|g ddggdt d¡gd}|V t|j d j| jj t|j
|
d jj| jj t|j
|
d | d |j
|
d d d
< W 5 Q R X t|d gd ~t jddd} | ¡ ¡ }t| jj|jjk t|g ddd
|
ggdd h}t|j d j| jj t|j
|
d jj| jj t|j
|
d | d |j
|
d d d
< d |j
|
d d d
< W 5 Q R X t|d gd t j d dddd
} d| _t jddd| d d
< t| j j t| g dgg}t|j
|
d j j t|j
|
d | t| g dddgg:}t|j
|
d j j t|j
|
d | d|j
|
d d d
< W 5 Q R X t| dgd td} t| d d
g dgg}t|j
|
d j j t|j
|
d | d d
t| d d d
ddgddggdd}t|d j j t|d | d d d
d S )Nr r r& r r! Úequivr# r r: Únbor© i90 )é Úi1rA r< Úalignedr7 ro rµ r Zcontigry ©Ú
|
buffersize)r) r Úbyteswapr* r r' Ú byteorderr r Údtypesr r£ r\ rÅ Ú
|
contiguous)r/ Úaur r r r Útest_iter_nbo_align_contig4 sb
|
þ ÿ þrÍ c C s t jddd dd¡} t| g dggt d¡gd}|. t|jd | t|jd jt d¡ W 5 Q R X t jdd d dd¡} t| g dd
|
ggd t d ¡gd .}t|jd | t|jd jt d ¡ W 5 Q R X t jddd ddd¡ dd¡} t| g ddggdt d¡gd }t|jd | t|jd jt d¡ t|jd jd | d d d
d d
d d d
f } t| g ddggdt d¡gd }t|jd | t|jd jt d¡ t|jd jd t jddd ddd¡j } t| g dd
|
ggdt d¡gd ^}t|jd | t|jd jt d¡ t|jd jd d|jd d< t
|
| d dk W 5 Q R X t| d d t jdddd d d
} t| g d d
|
ggd t d¡gd H}t|jd jt d¡ t|jd jd dddg|jd d d
< W 5 Q R X t| dddg d S ) Nr r r& r: r7 r r( r ú<f4r! rÁ ú>f4r# r r8 rA r< r. Úsafer½ )é` rw é r; r¾ )r8 é é0 g )Àr Úi4éþÿÿÿr r" )r8 ) r) r rG r r' r r rI r rH r rf r r r Útest_iter_array_castr sj "
|
þ"
|
þ
|
þ
|
ý
|
ý"r× c
|
C s¸ t tttdddg dggt d¡gd t tttdddg ddggdt d¡gd t tttdddg ddggd
|
t d¡gd t tttdddg d d ggdt d¡gd t tttdddg d d ggd
|
t d¡gd t tttdd dg ddggdt d¡gd t tttdddg dd ggdt d¡gd t tttdddg dd ggdt d¡gd t tttdddg ddggdt d¡gd t tttdddg d d ggdt d¡gd d S )Nr: r r& r< r½ r( r. Únor# rÁ r r! rÎ rÏ r rÐ r¾ rÕ )r r¶ r r r) r' r r r r Útest_iter_array_cast_errors° sb
|
ÿ
|
þ
|
þ
|
ý
|
ý
|
þ
|
ý
|
ý
|
ý
|
ýrÙ c C s° t t d¡g dggt d¡gd} t| jd t d¡ t| jjt d¡ t| jd t t d¡g ddggdt d¡gd } t| jd t d¡ t| jjt d¡ t| jd t t d¡g ddggd
|
t d¡gd } t| jd t d¡ t| jjt d¡ t| jd t t d ¡g ddggd t d ¡gd } t| jd t d ¡ t| jjt d ¡ t| jd t dg dggt d¡gd} t| d jt d¡ t| d d d S )Nç @r< r r( r r. rÐ r½ r# r¾ r¹ r" rÕ r7 )r r) Úfloat32r' r rÊ ÚvalueÚfloat64r r r r Útest_iter_scalar_castÜ sB
|
ÿ
|
ý
|
ý
|
ý rÞ c
|
C s t ttt d¡g dggt d¡gd t ttdg dggt d¡gd t ttt d¡g dggd t d¡gd
|
t ttt d¡g dggd t d ¡gd
|
d S ) Nr: r r½ r( rÚ r gDóÂääécLr< rÐ r# r¾ rÕ )r r¶ r r) rÛ r' rÝ r r r r Ú test_iter_scalar_cast_errors s$
|
ÿ
|
|
ÿ
|
ý
|
ýrß c C sº ddd} t jdddgd | d gdd}tr4t | ¡}ttt| trTtt | ¡| t|dgd g}d
|
d |D }tt j|dd| d gd \}}}trªtt | ¡| t| dd¡j
|
dd gd gd d}t |j dd |D }tt j|dd| dd¡j dd d gd \}}}tr,tt | ¡| t| dd¡j
|
dd gdgd d}|* |D ]}d |d< qVd gd \}}}W 5 Q R X trt t | ¡|d k t|t jd gd dd d S )Nr7 Úd©r/ rº rA r: ÚOr& Úrefs_okr< c S s g | ] }|d qS ©r r ©r> Zx_r r r r@ # s z1test_iter_object_arrays_basic.<locals>.<listcomp>rµ rQ rR c S s g | ] }|d qS rä r rå r r r r@ , s rW r .r8 )r) r r r+ r, r r¶ r r rG rH r r- rU )Úobjr/ Úrcr Úvalsr? r r r Ú test_iter_object_arrays_basic sD
|
|
ÿ
|
$ ÿ ré c C s t jddd} t| ddgdgddd }| |D ]}|d
|
d 7 < q0W 5 Q R X t| t d¡d t jddd} t| ddgdgddd }| |D ]}|d
|
d 7 < qW 5 Q R X t| t d¡d t jd d dgd} | d } t d¡| d d
< t| ddgdgddd }|" |D ]}|d
|
d 7 < qW 5 Q R X t| t d¡d t jd d dgd} | d } t d¡d | d d
< t| ddgdgddd }|> |d d }tr¶t |¡}|D ]}|d
|
d 7 < qºW 5 Q R X trøtt |¡|d k t| t d¡d d S )Nr râ r& rã rµ r r" rÕ r# .rA r )ÚprÄ ©r/ râ r/ )r/ rÕ iHþÙr r iIþÙ) r) r r r r£ r r+ r, r )r/ r r? Úobrç r r r Ú#test_iter_object_arrays_conversions< sZ ÿ ÿ ÿ ÿ
|
rí c C s t tdgddtdgddgdgddggd d
|
d } t| jd t d¡ t| jd t d¡ t tdgd dtdgddgdgddggd d
|
d } t| jd t d¡ t| jd t d¡ t tdgddtdddgdgddggd dd } t| jd t d¡ t| jd t d¡ t tdgddtdd dgdgddggd d
|
d } t| jd t d¡ t| jd t d¡ t tdgddtdd dgdgddggd d
|
d } t| jd t d¡ t| jd t d¡ t tdgddtdd dtdgddtdgddgdgddggd d
|
d } t| jd t d¡ t| jd t d¡ t| jd t d¡ t| jd t d¡ t| jd t tdgd dd tdgddgg ddgddgdggd
|
d } t| jd t d ¡ t| jd t d ¡ t| jd t d¡ t tdgd dd tdgddgdgddgddgdggd
|
d } t| jd t d¡ t| jd t d¡ t| jd t d¡ d S )Nr7 r r& r r½ Z common_dtyper< r. r: rÐ r© rA rÕ r¾ Úu4éôÿÿÿÚi8ù @Úc8rx r8 Úc16)r7 rï rñ rx r r )r r r rÊ r) r' rÜ r r r r Útest_iter_common_dtypei s ý ý ý ý ý ÿ ü þü þûrô c
|
C sú dD ]B} t d}t|gdg| gg}| t|jd |k W 5 Q R X qt d}|dd
}|d d
}t||gdgdgdgg}ttj|j W 5 Q R X t d}|}|}t||gdgdd gdd gg}|& t|jd |koð|jd |k W 5 Q R X t||gdgdgdgg.}t|jd |ko>t |jd |¡ W 5 Q R X t d}|d d d
|
}|dd d
|
}t||gdgdgd gg}t|jd |ko¨|jd |k t d tjd }|dd
}| tj¡d d
}t||gdgdgd gg}ttj|j W 5 Q R X dD ]~} t ddg¡}|}|j
|
}|}t|||gdgdgdg| gg6}|j\}}}tt ||¡ tt ||¡ W 5 Q R X qt ddg¡}|}|j
|
}|}t|||gdgdgdgdgg}|j\}}}t||k t||k t||k t ddg¡}|}t ddg¡}|j
|
}t|||gdgdgd gdgg}|j\}}}t||k t||k t||k t d¡}|d d
}|dd
}|d d
}t|||gdgdgd gd gg}|j\}}}t||k t||k t||k d S )N)r< r r ry Zcopy_if_overlapr rA r; r< r Zoverlap_assume_elementwiser: r r8 r& r7 )r r rv r ) r r r r r) Z shares_memoryÚint8Úviewr r rH ) Úflagr/ r r? rº r» Za2Zb2Úc2r r r Útest_iter_copy_if_overlap© s ÿ* 2" ÿ ÿ ÿ
|
ÿ rù c C sv t d dd¡} t| | jgg dggd ddgddggd}ttdd |D t d
|
ddd ¡} t| j| gg dggd dddgd gd}ttd d |D t dd ddd¡} t dd}t| |gg dggd d dddggd}tdd |D | | ddd¡ ¡ t dd }t| |gg dggd d dddggd}tdd |D | | ddd¡ ¡ t dd}t| |gg dggd d tj tj dggd}tdd |D | | ddd¡ ¡ t d
|
ddd ¡} t d ddd ¡}t| |gdgdggd ddddgddddggd}t|j
|
d t d dd ¡} t d d d¡}t| |gdgdggd ddgddggd}t|j
|
d d S )Nr r: r7 r< r rA ©r c S s g | ]\}}||kqS r r ©r> r? Úyr r r r@ s z%test_iter_op_axes.<locals>.<listcomp>r r8 c S s g | ]\}}||kqS r r rû r r r r@ s é r5 r; c S s g | ]\}}|| qS r r rû r r r r@ s c S s g | ]\}}|| qS r r rû r r r r@ s c S s g | ]\}}|| qS r r rû r r r r@ s é( r )r: r7 r5 r: ro é )r7 r5 ) r rG r rH r r r rU r) r¢ rK )r/ r rº r r r Útest_iter_op_axes s< *(
|
&&
|
&&
|
ÿ&ÿ ÿr c
|
C sJ t d dd¡} ttt| | gg dggd dgdgdggd ttt| | gg dggd ddgddggd ttt| | gg dggd ddgddggd ttt| | gg dggd ddgddggd ttt| | gg dggd ddgddggd ttt| | gg dggd ddgdddggd ttt| | gg dggd ddgddggd d S ) Nr r: r7 r< r rA rú r; )r rG r r r ©r/ r r r Útest_iter_op_axes_errors) s, ÿÿÿÿÿÿÿr c C sà t d ddd¡} t| }| ¡ }tdd |D dd |D d|_| ¡ }tdd |D dd |D t| dd gd
|
dd }| ¡ }td d |D d d |D d|_| ¡ }td d |D d d |D d|_| ¡ }tdd |D dd |D d|_t| t| | ¡ }tdd |D dd |D t| dgd
|
dddd}| ¡ }W 5 Q R X tdd |D | jd
|
d t ddd ddd¡} t| dgd
|
dddd}| ¡ }W 5 Q R X tdd |D | jd
|
d d S ) Nr r: r7 r8 c S s g | ] }|d qS rä r r= r r r r@ J s z"test_iter_copy.<locals>.<listcomp>c S s g | ] }|d qS rä r r= r r r r@ N s rµ ÚrangedrW ©rS rÇ c S s g | ] }|d qS rä r r= r r r r@ S s c S s g | ] }|d qS rä r r= r r r r@ W s ©r7 rx c S s g | ] }|d qS rä r r= r r r r@ [ s )r: é c S s g | ] }|d qS rä r r= r r r r@ a s r" r½ r5 )rS r$ r% rÇ c S s g | ] }|d qS rä r r= r r r r@ g s rR z<i4r& z>f8c S s g | ] }|d qS rä r r= r r r r@ m s ) r rG r r. r r r³ ÚnextrU )r/ r Újr r r Útest_iter_copyC sH ÿ ÿr r' ZAllÚ
|
loop_dtypez ignore::numpy.ComplexWarningc C sä | ¡ dkr|d }nt |¡jdkr.|d }tjdt | ¡ ¡ d}z| |¡}W n tk
|
rj Y d S X tj|fddd g|gd
|
d }t t |¡tj
|
¡r°t |tjd|d | ¡ }t |}~t |}~t || t || d S ) NÚmz[ms]r Z50éè r& rµ r rã r" ©r% r$ )Úlowerr) r' Úitemsizer r* ÚastypeÚ Exceptionr Z
|
issubdtypeÚnumberr r. r )r' r
|
ÚarrÚexpectedr1 Úit_copyÚresZres_copyr r r Útest_iter_copy_castsp s,
|
ÿ
|
r c C s t dt d¡fdt d¡fg¡} t dt d¡fdt d¡fg¡}t jd| d}t j|fd d
|
d g|gd d }| ¡ }t|}~t|}~|d |d ¡}t|d | t|d | | d jD ]D}|d | |d | j ¡}t|d | | t|d | | qÂd S )Nr/ zi,rº z>i,<i,>d,S17,>d,(3)f,O,i1râ z>i,>i,S17,>d,>U3,(3)d,i1,Or r& rµ r rã r" r )
|
r) r' r r r. r r r ÚnamesÚbase) Úin_dtypeZ out_dtyper r1 r Zres1Zres2r Úfieldr r r Ú test_iter_copy_casts_structured s. ÿ ÿ ÿ r c C s` t d} t| d gg dgddggd t d¡gd}t|jd j| j t|jd jt d¡ d S )Nr r< r r r r( rA )r r r) r' r r rK rf r r r Ú test_iter_allocate_output_simple´ s ÿr c C s t d} t| d gddgdgddgg}|Z d|jd d d
< | ¡ |D ] }|d d |d d 7 < qJt|jd | d W 5 Q R X d S )
|
Nr rµ Údelay_bufallocr< r r rA .r )r r r Úresetr ©r/ r r? r r r Ú,test_iter_allocate_output_buffered_readwrite¾ s ÿ r! c C s t ddd dd¡} t| d gg dgddggd t d ¡gd
|
}t|jd j| j t|jd j| j t|jd jt d ¡ t d dd ddd ¡j } t| d gg dgddggd t d ¡gd
|
}t|jd j| j t|jd j| j t|jd jt d ¡ t d dd ddd ¡
|
dd ¡} t| d gg dgddggdd t d ¡gd}t|jd j| j t|jd jd t|jd jt d ¡ d S )Nr rÕ r& r: r7 r< r r r r( rA r r8 r rQ )rS r% )rÒ rÓ r8 ) r rG r r) r' r r rK r rH rI rf r r r Ú!test_iter_allocate_output_itorderË s. ÿ ÿ
|
ýr" c C s t ddd ddd¡} td | gg ddgd ggt d
|
¡d gd dd gd gd }t|jd jd t|jd jd t|jd jt d
|
¡ d S )Nr rÕ r& r: r7 r8 r r r< rî rA r ©r% r r )r8 rÔ rÓ ) r rG r r) r' r r rK r rf r r r Ú test_iter_allocate_output_opaxesæ s þr$ c C s t tdgddtdgddd gg dggd dd gg } t| jd t d¡ t tdgd
|
dtdgddd gg dggd dd gg } t| jd t d¡ t tdgddtdddd gg dggd dd gg } t| jd t d¡ t tdgd dtdd
|
dd gg dggd dd gg } t| jd t d ¡ t tdgd dtd d
|
dd gg dggd dd gg } t| jd t d ¡ d S )Nr7 r r& r r½ r< r: r r rÕ rî rï rð )r r r rÊ r) r' r r r r Ú)test_iter_allocate_output_types_promotionñ s( ÿ ÿ ÿ ÿ ÿr% c C s t dgdd ¡ } t| d gg dgddgg}t|jd |jd t| | d gg dgdgddgg}t|jd |jd k t|jd d
|
¡|jd d S ) Nr7 rî r& r< r r r rA r: ú=)r r* r r rÊ r rf r r r Ú*test_iter_allocate_output_types_byte_order s
|
ÿ ÿr' c C sb t d ddt d¡t d¡gg ddggdggd } t| jd jt d
|
¡ t| jd jd d S ) NrA gffffff@ro r7 r r r< r8 r Ú
|
complex128)r r) rÛ r( r r r' r r r r r Ú&test_iter_allocate_output_types_scalar s
|
ÿr) c C s G dd dt j} t ddgddgg¡ | ¡}t d¡ dd¡j}t||d gg dgdgdd gg}tt |t |j
|
d t t |t |j
|
d k t|j
|
d j d
|
t||d gg dgdgdd d gg}tt |t |j
|
d t t |t |j
|
d k t|j
|
d j d
|
d S ) Nc @ s e Zd ZdZdS )z4test_iter_allocate_output_subtype.<locals>.MyNDArrayé N)Ú__name__Ú
|
__module__Ú __qualname__Z__array_priority__r r r r Ú MyNDArray s r. rA r: r7 r8 r< r r ©r: r: Z
|
no_subtype) r) Zndarrayr rö r rG rH r r Útyper r rK )r. r/ rº r r r r Ú!test_iter_allocate_output_subtype s" ÿ ÿÿr1 c C s¤ t d} ttt| d gg dgddgg ttt| d gg dgddgg ttt| d gdgddg tttd d gg ddgddggd t d¡gd t d d
|
d d d d¡} ttt| d gg dgddggd t d¡gd dtjdggd ttt| d gg dgddggd t d¡gd dd dggd ttt| d gg dgddggd t d¡gd dd ddggd t d d
|
d d d d¡} ttt| d gdgdgddggd t d¡gd dtjd ggd d S )Nr r r r< rµ r r r( r rÕ r& r: r7 r8 r rA r# Ú reduce_ok) r r r¶ r r r) r' rG r¢ r r r r Ú test_iter_allocate_output_errors0 sL ÿ ÿÿÿ ý ý ý ý ýr3 c C sh t jd gdgd} | jd jdks&t| jt d¡fks<tt jd gdgdd} | jd jdksdtd S )NÚint64r( r r r9 )r% r )r) r r rK r rÊ r' r r r r Útest_all_allocated[ s
|
r5 c C s² t d ddd¡} t| dg}| d¡ tdd |D | d d
d d d
f ¡ | d d d
|
d d
d d
f } t| dg}| d ¡ td d |D | d d d
d d
f ¡ d S ) Nr r: r7 r8 r rA c S s g | ]}|qS r r r= r r r r@ i s z)test_iter_remove_axis.<locals>.<listcomp>r r; c S s g | ]}|qS r r r= r r r r@ n s )r rG r Z remove_axisr rU rf r r r Útest_iter_remove_axisd s
|
*
|
r6 c C s t d ddd¡} t| dg}t|jd t|jd t|jd jd dd |D }| ¡ d
|
d |D }t|| t|jd tt d d | t|jd jd |
|
¡ t|j d t|d jt | ¡ t|j d t|d jd t|jt d d S )Nr r: r7 r8 r r9 r c S s g | ]}|qS r r r= r r r r@ { s z;test_iter_remove_multi_index_inner_loop.<locals>.<listcomp>c S s g | ]}|qS r r r= r r r r@ } s rA c S s | j S r ©rK r r r r r¬ r z9test_iter_remove_multi_index_inner_loop.<locals>.<lambda>r )r rG r r r rK ÚitviewsZremove_multi_indexr r r r rF Zenable_external_looprÜ )r/ r ZbeforeÚafterr r r Ú'test_iter_remove_multi_index_inner_loopp s&
|
r: c C sÔ d} t d ddd¡}g dgfD ]¬}t||| d}tt|ttd d|_tt|ttdd t||d| d }tt|ttd d|_tt|ttdd t|d d d
|
|d| d }tt|ttd d |_tt|ttd d t|d d d
|
d d d
|
f |d | d }tt|ttd d |_tt|ttd d t|d d d
d d d
|
f || d}tt|ttd d|_tt|ttdd | ¡ d|_tt|ttdd q d S )Nr5 r r8 r7 r: rµ rÆ rW r r; rx rQ é rA é ) r rG r r r ÚlistrC r r )rÇ r/ r\ r r r r Útest_iter_iterindex s4 $"r> c C s d} t ddd ddd¡}|jdd }t|d
|
gd gd| d }t|jd tdd |D | dD ]:}||_t|j| tdd |D ||d |d
q`t|d
|
dgd gdd| d}t|jd tdd |D | dD ]:}||_t|j| tdd |D ||d |d
qÚdd }t|d
|
ddgd gdd| d}t|jd t||| dD ]6}||_t|j| t||||d |d
qXd S ) Nr5 r rÕ r& r8 r7 r: rW rR r r< r ©r r c S s g | ] }|d qS rä r r= r r r r@ ¹ s z'test_iter_iterrange.<locals>.<listcomp>)r? rl )r7 r )r5 r5 )r rÿ )r< r c S s g | ] }|d qS rä r r= r r r r@ ½ s r rA rµ r½ )rS r% rÇ c S s g | ] }|d qS rä r r= r r r r@ Â s c S s g | ] }|d qS rä r r= r r r r@ Æ s c S s* t jg dd}| D ]}t ||f¡}q|S )Nr½ r& )r) r Ú concatenate)r Úvalr? r r r Ú get_arrayÈ s z&test_iter_iterrange.<locals>.get_arrayr )r rG rU r r r³ )rÇ r/ Za_fortr ÚrrB r r r Útest_iter_iterrange¯ sF ÿ & ÿ & þ rD c
|
C s( g } | tjddd ddd¡j ¡ ¡ ¡ | tjddd¡ tjd d
|
dd d
}d |_tjd d d|d d
< | |¡ | tjdd d dddd¡j¡ | D ]}dD ]v}g }t |ddgdddggdd|d}|j
|
st |d j |k | |d ¡ ¡ | ¡ qÒtt |¡|jdd qªq¢d S )Nr ró r& r: r7 r8 ry r )éA rÄ rA rÕ rÓ éx r5 )rA r: r7 r5 rw rz rÓ é rµ r r< r rÅ rQ rÁ )rS r$ rÇ r rR )r r) r rG rH r* rÈ r£ r' r r r r r. r r r@ rU )Úarraysr/ rÇ rè r r r r Útest_iter_bufferingØ s8
|
ÿ ÿ
|
"
|
|
ü
|
rI c C s t d¡ ddd¡j ¡ ¡ } t| dgdddggd d
|
d d }d }|& |jsf||d < |d7 }| ¡ qFW 5 Q R X t | j
|
d
|
dt d¡ d S )Nr r: r7 r8 rµ r r rÅ rÁ rQ rÓ )r$ rS rÇ r rA rR ) r) r rG rH r* rÈ r r r r rU r r r r Útest_iter_write_bufferingõ s
|
ürJ c C s t d¡} t jddd}t| |gddddgd gd
|
dd }t|j ttd d | ttdd | ttdd | dd }tt|| | ¡ t|j t|j d |Z t|d d d|d< t|dd
ddg tdd |D t
|
t t ddgd W 5 Q R X d S )Nr rA r r& rµ r r r2 r r" r# c S s | j S r r« r r r r r¬ r z3test_iter_buffering_delayed_alloc.<locals>.<lambda>c S s | d S r° r r r r r r¬ r c S s | dd
S )Nr r: r r r r r r¬ r c S s d| d< d S r° r r r r r Ú assign_iter s z6test_iter_buffering_delayed_alloc.<locals>.assign_iterrb r r: c S s$ g | ] }|d d |d d gqS )r r rA r r= r r r r@ s z5test_iter_buffering_delayed_alloc.<locals>.<listcomp>) r) r r r Úhas_delayed_bufallocr r r r r r= ÚziprC )r/ rº r rK r r r Ú!test_iter_buffering_delayed_alloc s*
|
ý
|
rN c C s| t jddd} t| ddgdddggd t d
|
¡gd d }| |D ]}|d d9 < q@W 5 Q R X t| dt jddd d S )Nry r r& rµ r r r rÅ r¾ r½ r7 ©r$ r% rÇ .r: )r) r r r' r ©r/ r Úvr r r Ú test_iter_buffered_cast_simple" s
|
|
|
ü rR c
|
C s* t jddd ¡ ¡ } t| ddgdddggd t d
|
¡ ¡ gd d }| |D ]}|d d9 < qLW 5 Q R X t| dt jddd t }| t j ¡ t jdd
|
d ¡ ¡ } t| ddgdddggdt d¡ ¡ gd d }| |D ]}|d d9 < qäW 5 Q R X t| dt jdd
|
d W 5 Q R X d S )Nry r r& rµ r r r rÅ r¾ r½ r7 rO .r: r" rò )
|
r) r r* rÈ r r' r r ÚfilterZComplexWarning)r/ r rQ Úsupr r r Ú#test_iter_buffered_cast_byteswapped1 s0
|
|
ü
|
|
ü rU c C s0 t jddd ¡ ¡ } | d7 } t| ddgddd ggd
|
t d ¡gd d }| |D ]}|d d9 < qPW 5 Q R X t| dt jddd d t jddd} | d7 } t| ddgddd ggd
|
t d ¡ ¡ gd d }| |D ]}|d d9 < qØW 5 Q R X t| dt jddd d t jdt jd ¡ ¡ } | d7 } t| ddgddd ggd
|
t d ¡gd d }|" |D ]}|d d9 < qfW 5 Q R X t| dt jdt jd d t jdt jd ¡ ¡ } t| ddgddd ggd
|
t d¡gdd }|" |D ]}|d d9 < qðW 5 Q R X t| dt jdt jd d S )Nry rò r& rñ rµ r r r rÅ r¾ ró r7 rO .r: y @r rv ) r) r r* rÈ r r' r Z clongdoubleZ
|
longdoublerP r r r Ú+test_iter_buffered_cast_byteswapped_complexO s^
|
|
|
ü
|
|
ü
|
|
|
ü
|
|
|
ü rV c s ddddg} t jdddd t d d
|
gd gd | d }dd |D }t|d d d t|d d d t|d d dgd gd t|d d d t|d d d t|d d d t|d d dgd gd t|d d d t|d jt | ¡ ddddg} t jddd dddddgdddggdf d< dddddgdddggdf d< dddddgdddggdf d< trt d ¡}t d d
|
gd gd | d }dd |D }t|d d d t|d d d t|d d dgd gd t|d d d t|d d d t|d d d t|d d dgd gd t|d d d t|d jt | ¡ d gd \}}}trtt d ¡| dg} t j d d g| d t d d
|
gd gd d d }td d |D d d!g dddg} t j d"d#g| d t
|
t fd$d% dddg}d&d'd(g}t j d)d*g|d t d d
|
gd gd |d }t|d jt |¡ td+d |D t j d)|dt j d*|dg d S ),N©r/ r ©rº rð ©r» rò r ©rà râ r7 r r& ç à?rµ rã r< r" r# c S s g | ]}t |¡qS r ©r) r r= r r r r@ s z;test_iter_buffered_cast_structured_type.<locals>.<listcomp>r r/ rº r» r: rà rA rª re râ rÚ c S s g | ] }| ¡ qS r )r. r= r r r r@ s )ç @r
rÕ c S s g | ] }|d qS rä r rå r r r r@ ´ s r5 rw )r] rv Útest)rw ry rz c s t ddgdgdddS )Nrµ rã r< r" rÕ r# )r r r r r r¬ ¹ s þz9test_iter_buffered_cast_structured_type.<locals>.<lambda>)rà Úu2rë )rº r½ ©rA r: r7 ©r8 r5 r c S s g | ]}t |¡qS r r\ rå r r r r@ Æ s ) r) r r r r' r£ r r+ r, r r r¶ )Úsdtr rè rç r? Úsdt1Úsdt2r r r Ú'test_iter_buffered_cast_structured_type s~ þ þ þ
|
|
|
þ ÿÿre c
|
C s| dddg} ddg}t jddg| d}d D ]N}t jd
|
d gd d}t t¡( t||fd dg||gdd|gd W 5 Q R X q(d S )NrW rX rZ )rº râ )r/ r½ r` ra r& )r r< r rA r: zi,irµ rã r" zf,fr# )r) r ÚpytestÚraisesr¶ r )rc rd r/ ZintentZ
|
simple_arrr r r Ú<test_iter_buffered_cast_structured_type_failure_with_cleanupË s
|
ÿrh c
|
C s t t¡( tjtjdddfdgddgd W 5 Q R X tjtjdddfdgd gddgd
|
} t t¡& | t| }d|d < W 5 Q R X W 5 Q R X d S ) Nr/ ZS1r& r r" rµ )r% r$ r\ rA r )r% Úop_flagsr$ r\ .)rf rg r r) r r r )r1 Úbufr r r Ú test_buffered_cast_error_pathsÞ s ÿ ÿ rk zCannot start subprocesszPyPy seems to not hit this.c C s4 t d¡} tjtjd| gtjdd}d|ks0td S )Na:
|
import numpy as np
|
|
it = np.nditer((np.array(1, dtype="i"),), op_dtypes=["S1"],
|
op_flags=["writeonly"], casting="unsafe", flags=["buffered"])
|
buf = next(it)
|
buf[...] = "a"
|
del buf, it # Flushing only happens during deallocate right now.
|
z-cT)ÚstderrÚtextr )ÚtextwrapÚdedentÚ
|
subprocessÚ check_outputr+ Ú
|
executableÚSTDOUTr )Úcoder r r r Ú)test_buffered_cast_error_paths_unraisableì s
|
ÿru c C s¸ dg} dg}t jd| d}t d¡|d< t|ddgd gd
|
|d }t|d jt |¡ t|ttdD ] \}}t t
|
|d |k¡ qhd g} dg}t jd| d}t d¡|d d d
d d f< t|ddgdgd
|
|d }|\ t|d jt |¡ d }|D ]6}t t
|
|d |k¡ |d d d7 < |d7 }qúW 5 Q R X t|d t d¡ ddd¡d dg} dg}t jd| d}t d¡|d d d
d d d f< t|ddgdgd
|
|d }|R t|d jt |¡ d }|D ],}t|d | |d d7 < |d7 }qÒW 5 Q R X t|d t d¡ dddd¡t d¡ d dg} dg}t jd| d}t d¡|d d d
d d d f< t|ddgd gd
|
|d }t|d jt |¡ d }|D ] }t|d | |d7 }q¦dg} dg}t jd| d}t d¡|d d d
d d d f< t|ddgd gd
|
|d }t|d jt |¡ d }|D ] }t|d | |d7 }q2dg} dg}t jd| d}t d¡ dddd¡|d< t|ddgd gd
|
|d }t|d jt |¡ d }|D ]$}t|d || d |d7 }qºdg} dg}t jd| d}t d¡ dd¡|d< t|ddgd gd
|
|d }t|d jt |¡ d }|D ],}t|d || d d d
|d7 }qFdg} d g}t jd| d}t d ¡ dd¡|d< t|ddgd gd
|
|d }t|d jt |¡ d }|D ]J}t|d d d
|| d t|d dd
d d d d g |d7 }qÚdg} d g}t jd| d}t d ¡ dd¡|d< t|ddgd gd
|
|d }t|d jt |¡ d }|D ]B}t|d d || d t|d d || d |d7 }qd g} d g}t jd| d}t d ¡ ddd¡|d< t|ddgd gd
|
|d }t|d jt |¡ d }|D ]}t|d d d
d f || d d d
d f t|d d d
df || d d d
d f t|d dd d
f d d g |d7 }q8d!g} d g}t jd| d}t d¡ ddd¡|d< t|ddgd gd
|
|d }t|d jt |¡ d }|D ]}t|d d d
d f || d d d
d f t|d d d
df || d d d
df t|d dd d
f d d g |d7 }q*d S )"NrW )r/ r½ ©r7 r: r: r r& r r/ rµ rã r< r" r# r )r/ râ rk )r/ râ rv r r: rA )r/ râ rc )rA r7 r: r: )r/ r rc )r/ r rv éH r7 )r/ r½ r )r/ r rd r )r/ r½ rd )r/ r r ro )r/ r r/ )r/ r½ )r: rA )r/ r )r7 r: )r/ r½ r ) r) r£ r r r r' rM r= rC r r rG r )rc rd r/ r r? Úcountr r r Ú test_iter_buffered_cast_subarray s2 þ þ" þ. þ þ þ þ þ þ þ.. þ.. ry c C s t d¡ ddd¡} t d¡ ddd¡}ttt| |gddgdgd ggd
|
d t| |gddgd gd ggd
|
d t d¡ ddd¡} t| |gddd gdgd ggd
|
d t d¡ ddd¡} ttt| |gddgdgd ggd
|
d t d¡ ddd¡} ttt| |gddgdgd ggd
|
d d S )Nr r: r7 rA ro rµ r r r rQ rR r< r2 r8 )r) r rG r r r rá r r r Ú test_iter_buffering_badwriteback¨ s8
|
ý
|
þ
|
þ
|
ý
|
ýrz c C sô t jdddgt jd} t| jt d¡ ttt| dgdgdd t| dgdgd
|
d }t|d d t|d jt d
|
¡ t jdddgt jd} t| jt d ¡ ttt| dgdgdd t| dgdgdd }t|d d t|d jt d¡ d S )NÚabcr/ Zabcdr& ZS4rµ r< ZS2r( ZS6r s abcZU4ZU2ZU6) r) r Úbytes_r r' r r¶ r Zunicode_rf r r r Útest_iter_buffering_stringÊ s ÿÿr} c C s4 t d¡} t| dddgdd}t|d j| j d S )Né rµ Z growinnerr r5 rÆ r )r) r r r r rf r r r Ú test_iter_buffering_growinnerÜ s
|
|
ÿr c sb t d¡dd
dddddg} d d
|
g}d d gd d ggt jg} fdd}| D ]\}}}t j| ¡ d g|| ||d}|J d|jd d< | ¡ ||_|D ]\}} | | t j
|
¡7 } q¤|jd }
|
W 5 Q R X t ddD ]~} t j|d g|| || |d} | L d| jd d< | ¡ || _| D ]\} }|| t j
|
¡7 }q | jd }W 5 Q R X t ||
|
qÜqVd S )Niæ éó éô rµ r r r2 rã )r< )r r )r rA r: )r rA r; )r r; r; c 3 sn t ddD ]^} t | dD ]N}D ]D}| j | j jf}tjj d|¡}dD ]}|||fV qRq qq
|
d S )Ni÷ÿÿÿry )r7 r7 r7 ri )rC r r) ÚlibÚ stride_tricksÚ
|
as_strided)ÚxsZysr r r Úskip©r/ Z op_axes_listr r Ú
|
get_paramsï s z3test_iter_buffered_reduce_reuse.<locals>.get_params)r r\ ri r% r r; .é )r r\ ri rÇ r% ) r) r Úfloatr' r r. r r r r Úint_rC r )r\ ri r% r r r r Znditer2Za2_inZb2_inZcomp_resÚbufsizeZnditer1Za1_inZb1_inr r r r Ú test_iter_buffered_reduce_reuseå sH
|
þ
|
þ r c C s¦ t d¡ ddd¡} t d¡ ddd¡}t d¡ dd¡}t| ||gg dd gdgdgg ttt| ||gg dgdd gdgg ttt| ||gg dgdgdd gg d S )
|
Nr r: r7 r8 r rA ro r< r )r) r rG r r r )r/ rº r» r r r Útest_iter_no_broadcast s ÿÿÿÿr c @ sL e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dS )ÚTestIterNestedc s t d ddd¡}t |dgddgg¡\} fdd|D }t|dddddd gd
|
d d d ddgg t |ddgdgg¡\} fdd|D }t|ddgddgdd gd
|
d gd d gddgg t |ddgdgg¡\} fdd|D }t|dddgddd gd
|
d dgd d dgg d S )Nro r: r7 r rA c s g | ] }t qS r ©r= ©r> Ú_©r r r r@ / s z-TestIterNested.test_basic.<locals>.<listcomp>r8 r5 r rv rw rx ry rz c s g | ] }t qS r r r r r r r@ 3 s c s g | ] }t qS r r r r r r r@ 7 s ©r rG r) Ú nested_itersr ©Úselfr/ r rè r r r Ú
|
test_basic* s &.zTestIterNested.test_basicc s& t d ddd¡}t |dgddgg¡\} fdd|D }t|dddddd gd
|
d d d ddgg t |ddgdgg¡\} fdd|D }t|ddgddgdd gd
|
d gd d gddgg t |ddgdgg¡\} fdd|D }t|dddgddd gd
|
d dgd d dgg tj|dgddggdd\} fdd|D }t|dddddd gd
|
d dd d dgg tj|ddgdggdd\} fdd|D }t|ddgd
|
d gddgd d gdd gddgg tj|ddgdggdd\} fdd|D }t|dddgd
|
d dgddd gd d dgg d S )Nro r: r7 r rA c s g | ] }t qS r r r r r r r@ @ s z/TestIterNested.test_reorder.<locals>.<listcomp>r8 r5 r rv rw rx ry rz c s g | ] }t qS r r r r r r r@ D s c s g | ] }t qS r r r r r r r@ H s rQ rR c s g | ] }t qS r r r r r r r@ M s c s g | ] }t qS r r r r r r r@ Q s c s g | ] }t qS r r r r r r r@ U s r r r r r Ú test_reorder: s& &.* & . zTestIterNested.test_reorderc sB t d ddd¡d d d
d d d
d d d
f }t |dgddgg¡\} fdd|D }t|ddddd d
|
gd d d dddgg t |ddgdgg¡\} fdd|D }t|ddgddgd d
|
gd d gd dgddgg t |ddgdgg¡\} fdd|D }t|ddd gddd
|
gd d dgd ddgg tj|dgddggdd\} fdd|D }t|dddd d d gd
|
d ddddgg tj|ddgdggdd\} fdd|D }t|ddgdd gd d gd
|
d gddgddgg tj|ddgdggdd\} fdd|D }t|ddd gdd d gd
|
ddgd ddgg d S )Nro r: r7 r; r rA c s g | ] }t qS r r r r r r r@ ^ s z1TestIterNested.test_flip_axes.<locals>.<listcomp>r8 r5 r rv rw rx ry rz c s g | ] }t qS r r r r r r r@ b s c s g | ] }t qS r r r r r r r@ f s rQ rR c s g | ] }t qS r r r r r r r@ k s c s g | ] }t qS r r r r r r r@ o s c s g | ] }t qS r r r r r r r@ s s r r r r r Útest_flip_axesX s& .&.* & . z TestIterNested.test_flip_axesc sæ t d dd¡}t d dd¡}t ||gdgdgg¡\} fdd|D }t|ddgddgddggddgddgddggg t ||gdgdgg¡\} fdd|D }t|ddgddggddgddggddgddggg d S )Nr: rA r7 r c s g | ] }t qS r r r r r r r@ | s z1TestIterNested.test_broadcast.<locals>.<listcomp>c s g | ] }t qS r r r r r r r@ s r )r r/ rº r rè r r r Útest_broadcastv s 2 z TestIterNested.test_broadcastc
|
s t ddd dd¡}tj|dgdggdd gd
|
d \} t d jt d
|
¡ fd d |D }t|dddgdddgg d }t ddd dd¡}tj|dgdggddgdd
|
d\} |j Z t d jt d
|
¡ |D ] } D ]}|d d7 < qâqÚt|dddgdddgg W 5 Q R X W 5 Q R X t|dddgdddgg t ddd dd¡}tj|dgdggddgdd
|
d\} t d jt d
|
¡ |D ]"} D ]}|d d7 < qqt|dddgdddgg | ¡ ¡ t|dddgdddgg d S )Nr rÕ r& r: r7 r rA r< r. r½ )ri r% c s g | ] }t qS r r r r r r r@ s z2TestIterNested.test_dtype_copy.<locals>.<listcomp>r8 r5 r r r! r¾ )ri r$ r% .)r rG r) r r r' Úclose)r r/ r rè r? rü r r r Útest_dtype_copy sH þ
|
ý
|
.ý
|
z TestIterNested.test_dtype_copyc C s t ddd dd¡}tj|dgdggdgd gd
|
d d \}}t|d jt d ¡ |D ] }|D ]}|d d7 < q\qTt|dddgdddgg d S ©Nr r r& r: r7 r rA rµ r r¾ r½ )r\ ri r$ r% .r8 r5 ©r rG r) r r r' ©r r/ r r r? rü r r r Útest_dtype_buffered s ü
|
z"TestIterNested.test_dtype_bufferedc sB t d¡ ddd¡}t |g dddgg¡\} fdd|D }t|dddddd d
|
d d d ddg g t |dddgg g¡\} fdd|D }t|dgdgdgdgdgd gd
|
gd gd gd gdgdgg t |ddgg dgg¡\} }g }|D ]"} D ]}| dd |D ¡ qøqðt|dddgddd gd
|
d dgd d dgg d S )Nro r: r7 rA r c s g | ] }t qS r r r r r r r@ ¿ s z*TestIterNested.test_0d.<locals>.<listcomp>r8 r5 r rv rw rx ry rz c s g | ] }t qS r r r r r r r@ Ã s c S s g | ]}|qS r r )r> Úzr r r r@ Ê s )r) r rG r r r )r r/ r rè Úkr? rü r r r Útest_0d¼ s $: zTestIterNested.test_0dc
|
C s² t ddd dd¡}tj|dgdggdgd gd
|
d d \}}|P |@ t|d jt d ¡ |D ] }|D ]}|d d7 < qhq`W 5 Q R X W 5 Q R X t|dddgdddgg d S r r r r r r Ú%test_iter_nested_iters_dtype_bufferedÍ s ü
|
(z4TestIterNested.test_iter_nested_iters_dtype_bufferedN) r+ r, r- r r r r r r¡ r¤ r¥ r r r r r ( s *r c C sz t d¡} ttt| d gg dgddggdgdggd t d¡ dd ¡} ttt| d gd
|
gdgddggdd gddggd d S ) Nr r< r r r r; rú r: r7 r rA )r) r r r r rG r r r r Útest_iter_reduction_errorÝ s
|
|
þ þr¦ c
|
C s t d¡} t| d gdgdgddggdgdggd}|Z d|jd d
|
< |D ]\}}|d
|
|7 < qJt|jd jd t|jd t | ¡ W 5 Q R X t d¡ d d ¡} t| d gdd gdgddggdd gddggd}| d|jd d
|
< t|d jd t|d j d |D ]2\}}t
|
t |D ]}|| || 7 < q"qt|jd jd t|jd t | ¡ W 5 Q R X t d¡} t| d gdd gdgddggd ddd ggd}t| d gdd ddgdgddggd ddd ggdd}|² |¢ |jd d¡ |jd d¡ | ¡ |D ] }|d d
|
|d 7 < q|D ] }|d d
|
|d 7 < q8t|jd |jd t|jd ¡ | j W 5 Q R X W 5 Q R X d S )Nr r2 r< r r r r; rú rA .r: r7 r r rb )r: r7 r5 rµ r ry )r rÇ )r) r r r r r ÚsumrG rK r rC rD r Úfillr r )r/ r r? rü r Zit1r2 r r r Útest_iter_reductioné s`
|
|
þ þ
|
þ ÿ ý r© c
|
C s t d¡} t jddd ¡ ¡ }t| |gddgdgdd ggd
|
gd ggd }|N t|d jt d¡ t|d j|jk |D ]\}}|d |7 < q~W 5 Q R X t|t | ¡ t d¡
|
dd¡} t jd
|
d
|
gdd ¡ ¡ }t| |gdddgdgdd ggd
|
d gd
|
d ggd }|^ t|d j d t|d j d |D ]2\}}t t|D ]}|| || 7 < qLq8W 5 Q R X t|t j | d d t d¡d }t j|d gddddgdgddggd d
|
gd d ggdd}|4 |jd d
|
¡ | ¡ t|d
|
d dd dg W 5 Q R X t dt j¡dd
d dd
d d
f d dd
|
¡}t |j¡
|
|j ¡|d< t jd t jd
|
dd¡}| ¡ } |d d d
d d
d f }t ||gdddgdgdgg¡}| |D ]\} }| d¡ q¶W 5 Q R X t|d d d
| d d d
t|d d d
d d S ) Nr g r½ r& r2 rµ r< r r r r; rú rA .r: r7 r re rb )Úaxisr r r/ r )rv r; rw r8 rz r5 rÓ )r) r r rÈ r* r r r' r r§ rG rK r rC rD r r¨ r r rõ Z transposer r. )
|
r/ rº r r? rü r rê r1 Zy_baseZ y_base_copyr r r Ú test_iter_buffering_reduction$
|
sd
|
|
þ þ &
|
|
ü 0
|
|
þ r« c C s| t d¡} t d¡}t j| |gdddgdgdggdd }| d
|
d |D }W 5 Q R X t|dd dd g tt|| j d S ) N)r: rv )rA rv r2 r rµ r< r r5 )r\ ri rÇ c S s g | ]\}}|j d qS rb r7 rû r r r r@ p
|
s zDtest_iter_buffering_reduction_reuse_reduce_loops.<locals>.<listcomp>r: )r) r£ r r r§ r )r/ rº r1 Zbufsizesr r r Ú0test_iter_buffering_reduction_reuse_reduce_loopsd
|
s
|
|
|
þr¬ c C s t d¡} t d¡}t dddgdddgg¡}t dddg¡}t jdddgdd}t jdddgd d}t jdddgd
|
d}ttt| |gg d d gd gg ttt| |gg d d gd dgg ttt| |gg d gd dd gg ttt| ||gg d d gd dgd dgg ttt| |gg d gd dgg ttt| ||gdgd gd d gd dgg t | ||gdgd gd d gd dgg¡ ttt| ||gdgd gd d gd dgg t j| |gdgd d gd dggd
|
d gdd ttt j| |gdgd d gd dggd
|
d gdd ttt j| |gdgd d gd dggd
|
d gdd d S )Nr re TFr rA Úu1r& rÄ r r Ú writemaskedr< Ú arraymaskr2 rµ r¾ r )r) r£ r r r r r¶ )r/ rº r Úm2Zm3Zmbad1Zmbad2r r r Ú test_iter_writemasked_badinputt
|
s
|
|
ÿÿÿþÿ ÿþÿþÿþÿÿüÿüÿür± c C s& z
|
| j W n tk
|
r Y dS X dS )NTF)r8 r )Úiteratorr r r Ú _is_buffered·
|
s
|
|
r³ r/ re r½ r& )é& r* r: )r8 i8 é| r7 )rx r7 )r´ r~ r5 r; c C s¦ | j }|d d }tj|td}dddg| |d< t | |gg ddgd d
|
gg¡}| |D ]\}}d |d< qZW 5 Q R X t| t d d d g| |¡ t | |gd gddgd d
|
gg¡}d}|, |D ] \}}d |d< t || ¡rÀd}qÀW 5 Q R X |st| t d d d g| |¡ n$t| t d d d g| |¡ d | d< tj| |gd gddgd d
|
ggdd gdd}| |D ]\}}d|d< qhW 5 Q R X t| t ddd g| |¡ d S )Nr; r7 r& TF.r r® r< r¯ rA rµ rÚ rð r" r )rK r) ÚemptyÚboolr r Z broadcast_toZmay_share_memory)r/ rK ZrepsZmskr1 r? r Z is_bufferedr r r Útest_iter_writemasked¾
|
sN ÿÿ ÿÿ ÿü r¸ ÚmaskÚ mask_axes)rA r8 r· rA r8 r c C sn t d¡}dg}dddg}ddg}|d kr0d }n |dd gg}tt t j| |f|||g|d
|
W 5 Q R X d S ) Nr6 r2 r¯ r r r r® r rA )r\ ri r )r) r£ r r r )r¹ rº r ZitflagsZ
|
mask_flagsZa_flagsr r r r Ú%test_iter_writemasked_broadcast_error s
|
|
|
ÿr» c C s t d¡ d¡} | ¡ }t jjdddd t¡}t j| |gddgdd gd
|
d ggd d gd}t }t rnt
|
|¡}|D ]\}}d|f|d< qr~~~t r®t
|
|¡| t |¡ks®t t| | || | | t d|f| j¡k ¡ sæt ~ t rþt
|
|¡|ksþt d S )Ni' z>i,Or r: )r rµ rã r r® r< r¯ z<i,Oú?r( r7 .)r) r r r. ÚrandomÚrandintr· r Úobjectr r+ r, Z count_nonzeror r r r' r )r Úoriginalr¹ r1 Z singletonrx rj Zmask_bufr r r Ú test_iter_writemasked_decref s, ÿý
|
"rÁ c C sJ t t d¡¡} dddddddd d
|
d d d dg }|D ]}ttt| | q2d S )Nr: rÜ rK r r8 rL r- Zhas_multi_indexZ has_indexrÊ r Znopr r ©r) r r r ÚAttributeErrorÚdelattr©r1 ÚattrÚsr r r Ú)test_iter_non_writable_attribute_deletion= s
|
þrÈ c C s8 t t d¡¡} ddddg}|D ]}ttt| | q d S )Nr: r r r³ r r rÅ r r r Ú%test_iter_writable_attribute_deletionG s rÉ c C sT t t d¡¡} z| d= | dd
= W n* tk
|
r8 Y n tk
|
rN tY nX d S )Nr7 rA r: )r) r r r¶ r r ©r1 r r r Útest_iter_element_deletionN s rË c C sH t jdddgd fd dgd} | D ] \}}|d |d< |d |d< q t| jd ddgddgd d
|
gg t jdddggd fd dgd gd d gd } | D ] \}}|d |d< |d |d< qt| jd ddgddgd d
|
gg t jdd d dgd} | D ]:\}}}|| |d< || |d< || |d< || |d< qìt| jd ddgddgg d S )NrA r7 rÿ )rÕ rd r( r r: r8 é é r2 )r; r )r% r\ r )ry r: N)rÕ r/ rh ri rj rk rw ro r5 )r) r r r )r1 r/ rº r» r r r Ú test_iter_allocated_array_dtypesX s( " ÿ " rÎ c C s t ddgdgdggd } t| jd tt| d t| jd t| jd ttt| | ¡ tt| d ttt| t t
|
d¡dgdggdgd } t| jd tt | d
|
t t
|
d¡dgdggdgdd } t| jd tt | d
|
tt " t t
|
d¡dgdggdd W 5 Q R X d dddg}t j ddd}t |ddgdgd|d} t| }t|d d t|d d t|d dgd gd t|d d d S ) Nr: r7 r r< r r r r5 rú rA r r rW rX rY rZ r[ r r& rµ rã r" r# r/ rº r» rà )r r r r r r r Ú StopIterationr r) r rD r r )r rb r/ rè r r r Ú test_0d_iters s> ÿ
|
& ÿrÐ c s t tdd t tjd ¡ dd¡ t¡ t¡d d d
df< t t fdd G dd d} t ttj j
|
tj | | gdd d S )
|
Nc S s t jdddd S )N)ihB r: r r& )r) r£ r r r r r¬ r z*test_object_iter_cleanup.<locals>.<lambda>ry r; c s0 t d d
d d d
f d d
d d d
f ¡S )Nr; )r) Úaddr ©r Zoarrr r r¬ ¥ r c @ s e Zd Zdd ZdS )z#test_object_iter_cleanup.<locals>.Tc S s t dd S )NZ Ambiguous)r¶ )r r r r Ú__bool__ª s z,test_object_iter_cleanup.<locals>.T.__bool__N)r+ r, r- rÓ r r r r rH © s rH râ r& ) r r¶ r) r ÚBUFSIZErG r r r¿ Ú
|
logical_orÚreducer )rH r rÒ r Útest_object_iter_cleanup s
|
|
ÿr× c C sP t d dgddgd dgddgg¡d d d
} t t¡ t | ¡ W 5 Q R X d S )NrA r; r: )r) r rf rg r¶ r§ )r r r r Ú test_object_iter_cleanup_reduce¯ s , rØ r )é@ r8 r: rW )r' rS c C s< t jdt jd}t j| dt|d}t|t jddtd d S )NrÙ r& rl )rª r' Úoutr8 )r) r Úintpr§ r¿ r Úfull)r rÚ r r r r Ú%test_object_iter_cleanup_large_reduce· s rÝ c C sn t t j¡jd } t jj t d¡| fd¡}tt t
|
||d d
d f f tt t
|
||d d
d f fdgd d S )NrG rA rb r r[ ) r) ÚiinforÛ Úmaxr r r r£ r r r )r r r r r Útest_iter_too_large s ÿrà c C s d} d}| | t t j¡jk r(|d7 }qddg| }g }t|D ],}|d d
}d||d < | t |¡¡ q@t|}tdD ]&}tt t
|
|d|¡ W 5 Q R X q~t
|
|dd¡ t|D ]N}tdD ]@}t
|
||d |¡ tt t
|
||d d |¡ W 5 Q R X qÈq¼d S )NrG rA r: r r; rv ) r) rÞ rÛ rß rC r r¶ rF r r Ú_multiarray_testsZtest_nditer_too_large)Z base_sizeÚnumZshape_templaterH r rK Úmoder r r Ú#test_iter_too_large_with_multiindexÏ s(
|
|
|
rä c
|
C s t jddd} | ¡ ¡ }t| jj|jjk t|g ddggdt d¡gd}| d|jd d d
< W 5 Q R X t |d t|g ddggdt d¡gd}z<|. t |j
|
j d
|
d |jd d d
< t d W 5 Q R X W n Y nX t |d t |j
|
j d t t t|d t|g ddggdt d¡gd}|( |jd }d|d d
< t|j
|
j W 5 Q R X t |d t|j
|
j d|d d
< t |d t|g ddggdt d¡gd}|* | |D ]}d|d< qºW 5 Q R X W 5 Q R X t|g ddggdt d¡gd}|8 | |D ]}d|d< qW 5 Q R X t t t|d W 5 Q R X t|g ddggdt d¡gd}~| |D ]}d|d< qlW 5 Q R X |j}t t| d S )Nr r r& r r! rÁ r# r§ r Fz!exit context manager on exceptionTr é{ .)r) r rÈ r* r r' rÉ r r r r\ r· r r ÚgetattrZwritebackifcopyÚ __enter__Ú RuntimeError)r/ rÌ r1 r? Zenterr r r Útest_writebacksñ s
|
ÿ
|
|
ÿ
|
|
ÿ
|
|
|
|
þ
|
þ
|
þré c C s` d
|
dd} d dd}| t dt d}t|t ddd |t dt d}t|t ddd dS ) zC using a context amanger and using nditer.close are equivalent
|
Nc S s\ t j}t | ||gg dgdgddgg¡}|D ]\}}}||||d q,|jd }| ¡ |S ©Nr< r r )rÚ r: )r) rÑ r r r ) r? rü rÚ Úaddopr1 r/ rº r» r r r r Ú add_close5 s ÿ
|
z(test_close_equivalent.<locals>.add_closec
|
S sl t j}t | ||gg dgdgddgg¡}|6 |D ]\}}}||||d q2|jd W 5 Q R £ S Q R X d S rê )r) rÑ r r )r? rü rÚ rë r1 r/ rº r» r r r Ú add_context? s ÿz*test_close_equivalent.<locals>.add_contextr5 r ry r: )N)N)rC r )rì rí r¢ r r r Útest_close_equivalent2 s
|
|
|
rî c C sD t t d¡¡} tt| d | ¡ ttt| ttt | d d S )Nr7 r r )
|
r) r r r r r r rÏ r ræ rÊ r r r Útest_close_raisesL s
|
rï c C s" t t d¡¡} tt| jd d S )Nr7 rA )r) r r r r¶ r rÊ r r r Útest_close_parametersS s rð c C sp t jddd} | ¡ ¡ }t F}| t¡ t j|g ddggdt d¡gd}~t |j
|
dksbt W 5 Q R X d S ) Nr r r& r r! rÁ r# rA ) r) r rÈ r* r ÚrecordÚRuntimeWarningr r' rD Úlogr )r/ rÌ rT r1 r r r Útest_warn_nocloseW s
|
|
ÿrô r Úwin32z!Errors with Python 3.9 on Windowsr Ú buf_dtype)r râ )râ r )úi,OzO,O)zO,ir÷ Ústepsc C s d}t tt jd |¡ | ¡}t j|t |¡gdddgdd}t|D ] }t| qH~t j|t |¡gdddgdd}t|D ] }| ¡ q~dS ) a
|
Checks for reference counting leaks during cleanup. Using explicit
|
reference counts lead to occasional false positives (at least in parallel
|
test setups). This test now should still test leaks correctly when
|
run e.g. with pytest-valgrind or pytest-leaks
|
i @rÚ rµ r rã r" ©r% r\ r$ N)
|
r) rÜ ÚintrÔ r r r' rC r r )r rö rø rÜ r r1 Ústepr r r Ú test_partial_iteration_cleanupc s ÿ
|
ÿ
|
rü c C sæ d}t tt jd |¡ | ¡}| dkr<d |tt jd < nd |tt jd d< t |¡}t j|t |¡gdddgd d
|
}t
|
t ¡ t | t | W 5 Q R X | ¡ t
|
t ¡ | ¡ | ¡ W 5 Q R X |t |¡ksâtd S ) Nrå rÚ râ rª Zf0rµ r rã r" rù )r) rÜ rú rÔ r r+ r, r r' rf rg r¶ r r r r )r rö rÜ r rx r1 r r r Ú test_partial_iteration_error
s$
|
ÿ rý c C sÄ d ¡ ¡ }tjdtjd dd¡dd
ddd
f }t d¡}tj||fdd gd
|
d d gd gdggd}| ¡ | ¡ j }| ¡ ¡ }t
|
|t
|
|kst t ||D ]\}}| | ¡ ¡s¤t q¤dS )a
|
Matches the expected output of a debug print with the actual output.
|
Note that the iterator dump should not be considered stable API,
|
this test is mainly to ensure the print does not crash.
|
|
Currently uses a subprocess to avoid dealing with the C level `printf`s.
|
a£
|
------ BEGIN ITERATOR DUMP ------
|
| Iterator Address:
|
| ItFlags: BUFFER REDUCE REUSE_REDUCE_LOOPS
|
| NDim: 2
|
| NOp: 2
|
| IterSize: 50
|
| IterStart: 0
|
| IterEnd: 50
|
| IterIndex: 0
|
| Iterator SizeOf:
|
| BufferData SizeOf:
|
| AxisData SizeOf:
|
|
|
| Perm: 0 1
|
| DTypes:
|
| DTypes: dtype('float64') dtype('int32')
|
| InitDataPtrs:
|
| BaseOffsets: 0 0
|
| Operands:
|
| Operand DTypes: dtype('int64') dtype('float64')
|
| OpItFlags:
|
| Flags[0]: READ CAST ALIGNED
|
| Flags[1]: READ WRITE CAST ALIGNED REDUCE
|
|
|
| BufferData:
|
| BufferSize: 50
|
| Size: 5
|
| BufIterEnd: 5
|
| REDUCE Pos: 0
|
| REDUCE OuterSize: 10
|
| REDUCE OuterDim: 1
|
| Strides: 8 4
|
| Ptrs:
|
| REDUCE Outer Strides: 40 0
|
| REDUCE Outer Ptrs:
|
| ReadTransferFn:
|
| ReadTransferData:
|
| WriteTransferFn:
|
| WriteTransferData:
|
| Buffers:
|
|
|
| AxisData[0]:
|
| Shape: 5
|
| Index: 0
|
| Strides: 16 8
|
| Ptrs:
|
| AxisData[1]:
|
| Shape: 10
|
| Index: 0
|
| Strides: 80 0
|
| Ptrs:
|
------- END ITERATOR DUMP -------
|
r§ r& ry Nr: g @rà rÕ r" r2 rµ r< r )r% r$ r\ ri )ÚstripÚ
|
splitlinesr) r r4 rG r Ú debug_printZ
|
readouterrrÚ rD r rM Ú
|
startswith)Zcapfdr Zarr1Zarr2r1 r Zres_lineZ expected_liner r r Útest_debug_print¦ s
|
7*
|
|
þ
|
r )r+ rf rn rp Únumpyr) Z numpy.core._multiarray_testsÚcorerá r r r r Z numpy.testingr r r r r
|
r r r r r r ÚmarkZskipifr3 rP rV rX rZ r^ ra rg rn rp rs ru 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 Z parametrizeZ typecodesÚfilterwarningsr r r r! r" r$ r% r' r) r1 r3 r5 r6 r: r> rD rI rJ rN rR rU rV re rh rk ru ry rz r} r Zslowr 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ô Ú version_infoÚplatformrü rý r r r r r Ú<module> s8 (
|
*
|
|
, % % $F KM >>,&'-@Z&-
|
!
|
+ ") 2J ("
|
3 6 ;@C , .ùÿ
|
<
|
÷
|
|
|
|
&$&*ý
|
"A
|
"ÿ þÿ ÿÿ
|
