fume-manage-python.git

U  
O±d¶Ùã@s*ddlZddlZddlZddlZddlZddlZddlmZddl    m
Z
ddlmZm Z mZmZmZmZmZmZGdddZGdddZGd    d
d
ZGdddZGd ddZGdddZGdddZGdddZGdddZGdddZGdddZGdddZGdddZ dS)éN)Úarray_indexing)Úproduct)Úassert_Úassert_equalÚ assert_raisesÚassert_raises_regexÚassert_array_equalÚassert_warnsÚHAS_REFCOUNTÚIS_WASMc@s¤eZdZddZddZddZddZd    d
ZddZd dZ    ddZ
ddZddZddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3d4Zd5d6Zd7d8Zd9d:Zd;d<Z d=d>Z!d?d@Z"dAdBZ#dCdDZ$dEdFZ%dGdHZ&dIdJZ'dKdLZ(e)j* +dMdNdOe, -dPg¡g¡e)j* +dQdRdSg¡e)j* +dTdUdRg¡dVdWZ.e)j*j/e0dXdYdZd[Z1d\d]Z2d^S)_ÚTestIndexingcs t dggg¡ttfddttfddttfddttfddttfddttfddttfd    dttfd
dttfddttfddttfd dttfddttfddttfddttfddttfddttfddttfddttfddttfddttfddttfdddS)NécsdS©Nç©r©ÚarúUd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/core/tests/test_indexing.pyÚ<lambda>óz3TestIndexing.test_index_no_floats.<locals>.<lambda>csdS)N)rrrrrrrrrcsdS)N)rrrrrrrrrcsdddfSrrrrrrrrcsdddfSrrrrrrrrcsdddddfSrrrrrrrrcsdddddfSrrrrrrrrcsdS)N)rrrrrrrrrrcsdS)N)rrrrrrrrrrcsdS)N)rrrrrrrrrrcsdS©Nçffffffö¿rrrrrrrcsdS)N)rrrrrrrr rcsdS)N)rrrrrrrr!rcsdddfSrrrrrrr"rcsdddfSrrrrrrr#rcsdddddfSrrrrrrr$rcsdddddfSrrrrrrr%rcsdS)N)rrrrrrrrr&rcsdS)N)rrrrrrrrr'rcsdS)N)rrrrrrrrr(rcsdddfSrrrrrrr)rcsdddddfSrrrrrrr*r)ÚnpÚarrayrÚ
IndexError©ÚselfrrrÚtest_index_no_floatss.z!TestIndexing.test_index_no_floatscst dgg¡ttfddttfddttfddttfddttfddttfddttfd    dttfd
dttfddttfddttfd dttfddttfddttfddttfddttfddttfddttfddttfddttfddttfddttfdddS)Nr csddSrrrrrrr0rz5TestIndexing.test_slicing_no_floats.<locals>.<lambda>csddddfS)Nrrérrrrrr1rcsdddddfS©Nrrrrrrrrr2rcsdddddfS)Nrérrrrrrr3rcsddddfSrrrrrrr4rcsddSrrrrrrr6rcsddddfS)Nrr ç@rrrrrr7rcsdddddfSrrrrrrr8rcsddddfSrrrrrrr9rcsdddddfS)Nrç@rrrrrrr:rcsdddS)Nçð?rrrrrr<rcsdddddfS)Nrrr!rrrrrr=rcsdddddfS)Nr r"rrrrrrr>rcsdddddfS)Nç@rrrrrr?rcsdddddfS)Nrér!rrrrrr@rcsdddS)Nr#rr!rrrrrrBrcsdddS)Nr#r!rrrrrrCrcsdddddfS)Nrr!rrrrrrDrcsdddddfS)Nr#r r"rrrrrrrErcsdddddfS)Nr#r$rrrrrrFrcsdddddfS)NgÙ?r"r!rrrrrrGrcsdddSrrrrrrrIr©rrrÚ    TypeErrorrrrrÚtest_slicing_no_floats,s.z#TestIndexing.test_slicing_no_floatscs&t dggg¡ttfdddS)Nr csS©NrrrrrrOrz;TestIndexing.test_index_no_array_to_index.<locals>.<lambda>r&rrrrÚtest_index_no_array_to_indexKsz)TestIndexing.test_index_no_array_to_indexcCs>t dddg¡}t|d|tjt|dj|jddS)Nr ré)rrrZnewaxisÚndim©rrrrrÚtest_none_indexQszTestIndexing.test_none_indexcCsRt dddg¡}t|d|t|dj|kt d¡}tt|dtjdS)Nr rr+rr)rrrrÚbaseÚ
isinstanceÚint_r-rrrÚtest_empty_tuple_indexWs
 
z#TestIndexing.test_empty_tuple_indexcCsFtjddd}t|dj|jt|d|tt|dtjdS)NrZV4©Údtype.)rÚzerosrr4ÚtypeÚndarray)rÚsrrrÚtest_void_scalar_empty_tuple_sz)TestIndexing.test_void_scalar_empty_tuplecCsät d¡}| tj¡}t d¡}t||||t d¡||<t|t d¡t d¡ dd¡}t||||t d¡dddf||<t|t d¡dddfjdddt d¡ dd¡}t|||f|||fdS)Nr é
rr ©Zaxisé)rÚarangeÚastypeZuintprÚreshapeÚrepeat)rÚindexZu_indexÚarrrrrÚtest_same_kind_index_castinges
 
&z)TestIndexing.test_same_kind_index_castingcCs|t dddg¡}t|ggt|gj|jtjgtjd}t|ggt|gj|jt g¡}tt|j|dS)Nr rr+r3)rrrr4ÚintprrÚ__getitem__©rrÚbrrrÚtest_empty_fancy_indexys
z#TestIndexing.test_empty_fancy_indexcCsÎt dddgdddgddd    gg¡}t|d
|k    t|d
|t|d
j|kt|d|dt|d|dddft|d |dddft|dt d¡t d¡}d|tf<t|ddS)Nr rr+r%r éééé    .)r.r©.r)r.r )rrrrr/ÚEllipsisrFrrrÚtest_ellipsis_indexsþ
 
z TestIndexing.test_ellipsis_indexcCsjt dddgdddgddd    gg¡}t|d
dddgt|dddd    gtt|jdtt|jd dS)Nr rr+r%r rIrJrKrLréÿÿÿÿi@l)rrrrrrEr-rrrÚtest_single_int_indexsþz"TestIndexing.test_single_int_indexcCs^t dddgdddgddd    gg¡}t|t d
¡|dt|t d¡|ddddS© Nr rr+r%r rIrJrKrLTFr©rrrr-rrrÚtest_single_bool_indexsþz#TestIndexing.test_single_bool_indexcCsxt d¡}t dg¡}tt|j|t dgd¡}tt|j|tjdtd}tt|j|tt|jtd|fdS)N)r r%r+TFrI©r%r%r3)    rÚonesrrrrEr5ÚboolÚslice)rrBrArrrÚtest_boolean_shape_mismatch¶s
z(TestIndexing.test_boolean_shape_mismatchcCsNt dddgg¡}tjdgtd}t|||d||<t|dddggdS)NrTr3r#)rrrWrrFrrrÚtest_boolean_indexing_onedimÄs
z)TestIndexing.test_boolean_indexing_onedimcCsTt d¡}dd}tt||gtt||dddgtt||dddddgdS)Nr%cSs|||dk<dS)NrPr)rÚvrrrÚfÓsz>TestIndexing.test_boolean_assignment_value_mismatch.<locals>.fr rr+)rr=rÚ
ValueError)rrr\rrrÚ&test_boolean_assignment_value_mismatchÎs
 
z3TestIndexing.test_boolean_assignment_value_mismatchcCsZt d¡}tjdtd}d|dd<tjdtd||<t d¡}d|dd<t||dS)Néèr3Tédr )rr5rWrVÚobjectr)rrBÚindxÚexpectedrrrÚ!test_boolean_assignment_needs_apiÚs
 
z.TestIndexing.test_boolean_assignment_needs_apicCs¾t dddgdddgddd    gg¡}t d
dd
gdd
dgd
dd
gg¡}t||ddddd    gt||ddddggt||d||dd||<t|dddgdddgdddggdSrRrSrFrrrÚtest_boolean_indexing_twodimès þþþz)TestIndexing.test_boolean_indexing_twodimcCsHt dddg¡}dddg}t||ddgt|d|fddggdS)Nr rr+TFrSrFrrrÚtest_boolean_indexing_listûs
z'TestIndexing.test_boolean_indexing_listcCst d¡}tjdtjdddd}t d¡ddd}|||<t|ddt d¡}t d¡ dd¡ddd}|||ddf<t|dddgdS)    Nr r3rPr©r rr:rr )rrVr5rDr=rr?)rrrGÚcrrrÚ,test_reverse_strides_and_subspace_bufferinits
 
z9TestIndexing.test_reverse_strides_and_subspace_bufferinitcCsXt d¡dddf}t d¡ddd}t|||| d¡t d¡ dd¡}dS)Nr:rPÚCér)rr=rÚcopyr?)rrÚirrrÚ'test_reversed_strides_result_allocationsz4TestIndexing.test_reversed_strides_result_allocationcCsft dd¡}t dd¡}t d¡ ddd¡j|ddg<t d¡ ddd¡j ¡|ddg<t||dS)N)r+r%rrPérr%rr )rÚfullr=r?ÚTrlrrFrrrÚ%test_uncontiguous_subspace_assignments
"z2TestIndexing.test_uncontiguous_subspace_assignmentcCs*t d¡}tt|jt dg¡fddS)N) r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r ré )rrVrrrErr-rrrÚ(test_too_many_fancy_indices_special_case)s
z5TestIndexing.test_too_many_fancy_indices_special_casecCsJt d¡}t|t d¡|t d¡t|t d¡|t d¡dS)Nr TF)rrrÚbool_r-rrrÚtest_scalar_array_bool.s
z#TestIndexing.test_scalar_array_boolcCsBt d¡}t||dk    t||dk    t||ddk    dS)Nr r.)rr=rr-rrrÚtest_everything_returns_views;s
z*TestIndexing.test_everything_returns_viewscCsHt d¡}tt|jddgdddgftt|jddgdddgfddS)N©r r rr r)rr5rrrEÚ__setitem__r-rrrÚtest_broaderrors_indexingCs
z&TestIndexing.test_broaderrors_indexingcCszt d¡}tjdtjd}d|d<tt|j|tt|j|dtjdtjd}d|d<tt|j|tt|j|ddS)Nr rkr3r:rPré)rr5rVrDrrrEry)rrÚindrrrÚ test_trivial_fancy_out_of_boundsHs
z-TestIndexing.test_trivial_fancy_out_of_boundscCstt d¡}tjdtjd ddd¡dddddf}t|||d||<t d¡}d|d<d|d<t||dS)NrIr3rr r+rrP)rr=rDr?r)rrÚidxÚresrrrÚtest_trivial_fancy_not_possibleSs
,
z,TestIndexing.test_trivial_fancy_not_possiblecCs°Gdddtj}t d¡}| ¡j|d}| d¡||ddddd    gddf<t|dk ¡||ddddddd    gf<t|dk ¡| d¡||d
<t|dk ¡dS)Nc@seZdZddZdS)z7TestIndexing.test_nonbaseclass_values.<locals>.SubClasscSs| d¡dS)Néc)Úfill©rÚoldrrrÚ__array_finalize__dszJTestIndexing.test_nonbaseclass_values.<locals>.SubClass.__array_finalize__N©Ú__name__Ú
__module__Ú__qualname__rrrrrÚSubClasscsrrx)r6r rrr+r%.)rr7r5rlÚviewrrÚall)rrrr8rrrÚtest_nonbaseclass_valuesbs
 
 
z%TestIndexing.test_nonbaseclass_valuescCst d¡}tjdtjd dd¡}t||ddddd    gddf<t||t||ddddddd    gf<t||t||d
<t||dS)Nrxr<r3r rr rr+r%.)rr5r=Úfloat64r?Ú
memoryviewr)rrr8rrrÚtest_array_like_valuesws
 
 
z#TestIndexing.test_array_like_valuescCstjjdddgddgd}tjdddgtd}t||jjt d    d
g¡}t||jjt|djjt|d    jjdS)N)ZNGC1001r{)ZNGC1002r#)ZNGC1003r#)ÚtargetZS20)ZV_magz>f4r3FTrr .)rZrecrrWrÚflagsÚ    writeable)rÚdr|rrrÚtest_subclass_writeablesÿz$TestIndexing.test_subclass_writeablecCsRt d¡}t d¡ dd¡j}t||jj| dd¡}t||dfjjdS)Nr:r rrPr r)rr=r?rqrrÚf_contiguousrFrrrÚtest_memory_orders
 
zTestIndexing.test_memory_ordercCsBGddd}|}Gddd}t d¡}tt|dtjt d¡}tt||tjt d¡}tt||t d¡ftjtt|||ftjt d¡}tjdtd    }||d<tt|dtjt |dg¡}tt||tjt |dgg¡}tt||t d¡ftjtt|||ftjdS)
Nc@seZdZddZdS)z2TestIndexing.test_scalar_return_type.<locals>.ZerocSsdS©NrrrrrrÚ    __index__sz<TestIndexing.test_scalar_return_type.<locals>.Zero.__index__N)rrrrrrrrÚZerosrc@seZdZddZdS)z7TestIndexing.test_scalar_return_type.<locals>.ArrayLikecSs
t d¡Sr©rrrrrrÚ    __array__¦szATestIndexing.test_scalar_return_type.<locals>.ArrayLike.__array__N)rrrrrrrrÚ    ArrayLike¤srrr )r r rr3)rr5rr0Úfloat_rrar7)rrÚzrrrGrrrÚtest_scalar_return_types&
 
 
 
z$TestIndexing.test_scalar_return_typecCsªt dg¡}tr"t t tj¡¡}d|tjdgtjd<d|tjdgtjd<tt    |j
tjdgtjddtt    |j
tjdgtjddtr¦tt t tj¡¡|dS)Nrr r3)rrr
ÚsysÚgetrefcountr4rDZuint8rrryr)rrÚrefcountrrrÚtest_small_regressions¼sÿÿz#TestIndexing.test_small_regressionscCs<tjdtjdtddd}| t d¡¡}tjdtjdtddd}| t d¡¡}tjdd    d|d
<t |j¡}t    ||||||<t |jd¡ tj¡dt 
d¡jd}| tj
¡d|j}t |j¡|d
<t    || tj ¡|||| tj ¡<||ddk|ddd||ddk<dS) Né@r3rr iùÿÿÿZS8roZbbbbbbbbÚS.rr)rr5Úint8Úordrr4rr=ÚsizerrDÚitemsizer>Úint16)rr[rÚxrGrrrÚtest_unalignedÌs  .zTestIndexing.test_unalignedcCsXt d¡}Gdddt}dgdgf}||}t||jdkt||fjdkdS)Nrxc@seZdZdS)z7TestIndexing.test_tuple_subclass.<locals>.TupleSubclassN©rrrrrrrÚ TupleSubclassæsr¯r ©r )rrVÚtuplerÚshape)rrBr¯rArrrÚtest_tuple_subclassâs
z TestIndexing.test_tuple_subclasscCs^Gddd}t d¡}t||||ftjdddgd}t||||fdS)Nc@s$eZdZddZddZddZdS)zDTestIndexing.test_broken_sequence_not_nd_index.<locals>.SequenceLikecSsdSrrrrrrrõszNTestIndexing.test_broken_sequence_not_nd_index.<locals>.SequenceLike.__index__cSsdS)Nr rrrrrÚ__len__øszLTestIndexing.test_broken_sequence_not_nd_index.<locals>.SequenceLike.__len__cSstddS)NzNot possible)r)rÚitemrrrrEûszPTestIndexing.test_broken_sequence_not_nd_index.<locals>.SequenceLike.__getitem__N)rrrrr´rErrrrÚSequenceLikeôsr¶r:r°)Úf1Úi8)Úf2r¸r3)rr=rr5)rr¶rBrrrÚ!test_broken_sequence_not_nd_indexîs
 
 
z.TestIndexing.test_broken_sequence_not_nd_indexcCst d¡}t d¡}t d¡dddddf}t |d¡}t|||| ¡tjdtjd}t|||f|| ¡|fdS)Nr:)r:r)r:é7r%r%r%r3)rrVr=Zbroadcast_torrlr5rD)rr¬Zx2r|ZzindrrrÚ!test_indexing_array_weird_stridess
 
z.TestIndexing.test_indexing_array_weird_stridescCsLt d¡}|ddddddf}tdddddgf}d||<t|ddS)    NrUrPrr rr+r:g$@)rr5rXr)rZarrorBZslicesrrrÚ$test_indexing_array_negative_stridess
 
z1TestIndexing.test_indexing_array_negative_stridescCsHtjddd}t d¡|d<t|dtjddd|ddksDtdS)N)r r rhr3Zasdfgr©rr ós)rr5Zstr_rrÚAssertionError)rrBrrrÚtest_character_assignmentsz&TestIndexing.test_character_assignmentrATFrÚnumrsé(Ú original_ndimr c    CsZt d|¡}t t¡||f|W5QRXt t¡d||f|<W5QRXdS)Nr°r#)rrVÚpytestÚraisesr)rrArÂrÄrBrrrÚtest_too_many_advanced_indices(s
z+TestIndexing.test_too_many_advanced_indiceszno threading)Úreasoncs¾ddlm}t dg¡}t d|fgd¡}t d|fgd¡}tjjdd |¡dddftj ¡fdd    |d
dfdd tdD}|D]}|     ¡qj|ksºt
dS)Nr)ÚThreadPoolExecutor)ÚÚf8rÊr)éprK)r©cs$jdt|dtjd}||dS)NrrÌ)r©r4)ZintegersÚlenrrD)rBrb)ÚrngrrÚfuncDsz<TestIndexing.test_structured_advanced_indexing.<locals>.funcrK)Úmax_workerscsg|]} ¡qSr)Zsubmit)Ú.0Ú_)rBrÏÚtperrÚ
<listcomp>IszBTestIndexing.test_structured_advanced_indexing.<locals>.<listcomp>r:)Úconcurrent.futuresrÉrr4ÚrandomÚuniformrZdefault_rngÚrangeÚresultrÀ)rrÉÚdtZfuturesr\r)rBrÏrÎrÓrÚ!test_structured_advanced_indexing6s 
 
 
z.TestIndexing.test_structured_advanced_indexingcCspt d¡ d¡}t|ddgt |d|dg¡t|ddgddgft ddg¡tt|jtdgdS)Nr<rxrr rI)    rr=r?rrrrrErXr-rrrÚtest_nontuple_ndindexOs$$z"TestIndexing.test_nontuple_ndindexN)3rrrrr(r*r.r2r9rCrHrOrQrTrYrZr^rdrerfrirnrrrtrvrwrzr}rrrrrr r¤rr³rºr¼r½rÁrÅÚmarkÚparametrizerrrÇZskipifrrÛrÜrrrrrs^    
         
"
    ÿ
 
rc@seZdZddZdS)ÚTestFieldIndexingcCs<t ddg¡}tt|dtjtt|dgtjdS)Nr)rrËr)rr5rr0r7r-rrrr Wsz)TestFieldIndexing.test_scalar_return_typeN)rrrr rrrrrßVsrßc    @s~eZdZddZddZddZddZej     d    d
ddge
d fddgd
dfd
ddgddgfg¡ddZddZddZ d S)ÚTestBroadcastedAssignmentscCs|||<|Sr)r)rrr|ÚvalrrrÚassign`sz!TestBroadcastedAssignments.assigncCstt d¡}t d¡|d<t d¡|dddgddf<t d¡|ddddgf<t d¡|dgdgdggddgf<dS)N)r+r)r r+r.rr r)rr5rVr-rrrÚtest_prepending_onesds
 
z/TestBroadcastedAssignments.test_prepending_onescCsz|j}tj}t d¡}tt|||dt d¡tt|||dddgft d¡tt|||dgdggft d¡dS)Nr .©rr r rr+)rrr )rârÚs_r5rr]rV©rrârårrrrÚtest_prepend_not_onens
"z/TestBroadcastedAssignments.test_prepend_not_onecCsº|j}tj}t d¡}tt|||dt d¡tt|||dt d¡tt|||dddgft d¡tt|||dddgft d¡tt|||dgddft d¡dS)N)r r .rg)r rrrä)rârrår5rr]rærrrÚtest_simple_broadcasting_errorsxs
$$z:TestBroadcastedAssignments.test_simple_broadcasting_errorsrA.r rNrc    Cs`t d¡}t d¡}t||j dd¡}t t¡}|||<W5QRXt|j     |¡s\t
dS)N©r`r`)r+r%r rIrJú rÊ)rr5Ústrr²ÚreplacerÅrÆr]ÚvalueÚendswithrÀ)rrAÚvaluesrBZ    shape_strÚerrrÚ*test_broadcast_error_reports_correct_shapes
 
zETestBroadcastedAssignments.test_broadcast_error_reports_correct_shapecCsXt d¡}dddg|dgdgdggdddgf<t|ddddfdddgk ¡dS)Nrxrr+r%rr )rr5rrr-rrrÚtest_index_is_largers
$z/TestBroadcastedAssignments.test_index_is_largercCsZt d¡}t d¡dddf}t d¡ddd}|||<t|ddd|k ¡dS)Nrér`rP)rr5r=rr)rrr[rGrrrÚtest_broadcast_subspaces
 
z2TestBroadcastedAssignments.test_broadcast_subspace)rrrrârãrçrèrÅrÝrÞrXrñròrórrrrrà_s
 
ý
ràc@s$eZdZddZddZddZdS)ÚTestSubclassescCs4Gdddtj}t d¡}| |¡}|dd}tt||kt|j|kt||dd|dddg}tt||kt|j|k    tt|jtjkt||dddgt|j|dddg||dk}tt||kt|j|k    tt|jtjkt|||dkt|j||dkdS)Nc@seZdZdS)z+TestSubclasses.test_basic.<locals>.SubClassNr®rrrrr¨srr r+rr r)rr7r=rrr6r/r)rrrr8Zs_sliceÚs_fancyZs_boolrrrÚ
test_basic£s&
 
zTestSubclasses.test_basiccCsLGdddtj}t d¡}| |¡}d|j_|dddg}t|jjdS)Nc@seZdZdS)z8TestSubclasses.test_fancy_on_read_only.<locals>.SubClassNr®rrrrrÃsrr Frr r)rr7r=rrrr)rrrr8rõrrrÚtest_fancy_on_read_onlyÀs
 
z&TestSubclasses.test_fancy_on_read_onlycCsGdddtj}t d¡ |¡}|dd}t|j|t|j||ddddg}t|j|t|j|||dk}t|j|t|j|dS)Nc@seZdZddZdS)z=TestSubclasses.test_finalize_gets_full_info.<locals>.SubClasscSst |¡|_||_dSr))rrÚfinalize_statusrrrrrrÐszPTestSubclasses.test_finalize_gets_full_info.<locals>.SubClass.__array_finalize__NrrrrrrÏsrr:r+rr r)rr7r=rrrør)rrr8Znew_srrrÚtest_finalize_gets_full_infoÍsz+TestSubclasses.test_finalize_gets_full_infoN)rrrrör÷rùrrrrrô¢s rôc@seZdZddZdS)ÚTestFancyIndexingCastcCs d}t |¡ t¡}d|d<t |¡}t dg¡||<t|ddttj|jdgdgft dg¡t|ddttj|j|t dg¡t|dddS)    N)rKé?Tr¾r ry@ð?ryð?)    rr5r>rWrrr    ZComplexWarningry)rr²Z
bool_indexZ
zero_arrayrrrÚtest_boolean_index_cast_assignãs$
 
 
ÿ
ÿz4TestFancyIndexingCast.test_boolean_index_cast_assignN)rrrrürrrrrúâsrúc@seZdZddZddZdS)ÚTestFancyIndexingEquivalencecCs\tjdtd}| ¡}dddg|dd<dddg|dddg<t||tjdtddddf}dddgg|dgddf<t||d|j}dgdgdgg|dddgf<t||dddftjdtd}| ¡}dgdgdgd    ggg|ddd
f<dgdgdgd    ggg|dgd
f<t||| d¡}dgdgdgd    ggg|dgd
f<t||dS)Nr r3r ©r rr+rr)r+r%r r%.ÚF)rr=rarlrrqrV)rrrGrBZcmp_arrrrrÚtest_object_assignùs&
" 
 
 z/TestFancyIndexingEquivalence.test_object_assigncCst d¡}| ¡}t dddg¡|dd<t dddg¡|dddg<t||t d¡dddf}t dddgg¡|dgddf<t||ddS)    Nr Ú2z-3z-1r+rrr )rr=rlrrrFrrrÚtest_cast_equivalences
 
 z2TestFancyIndexingEquivalence.test_cast_equivalenceN)rrrrrrrrrrýøs rýc@sPeZdZdZddZddZddZdd    Zd
dZdd Z    ddZ
ddZdS)ÚTestMultiIndexingAutomateda:
    These tests use code to mimic the C-Code indexing for selection.
 
    NOTE:
 
        * This still lacks tests for complex item setting.
        * If you change behavior of indexing, you might want to modify
          these tests to try more combinations.
        * Behavior was written to match numpy version 1.8. (though a
          first version matched 1.7.)
        * Only tuple indices are supported by the mimicking code.
          (and tested as of writing this)
        * Error types should match most of the time as long as there
          is only one error. For multiple errors, what gets raised
          will usually not be the same one. They are *not* tested.
 
    Update 2016-11-30: It is probably not worth maintaining this test
    indefinitely and it can be dropped if maintenance becomes a burden.
 
    cCsPt t ddddg¡¡ dddd¡|_t d¡|_dtdt dd    d    g¡t dd    gd    dgg¡t d    d    gd    d    ggg¡t    d
ddt    dddt    d ddt    dddtjdtj
dt dddg¡t dgdgdgg¡tjddgddggt d¡ ¡dtjddgtj dtjdgdtdt ddg¡g|_tdddgt dg¡dg|_t    dddg|_dS)Nr+r r rI)r+rr rIÚskiprTFéûÿÿÿrr`r%rPéþÿÿÿéýÿÿÿ)rr r r3rDérr#)rr=Úprodr?rÚemptyrGrNrrXrDr4Znewbyteorderr§r5ÚintÚcomplex_indicesÚsimple_indicesÚfill_indicesrrrrÚsetup_method>s.&
 
 
 
"íÿz'TestMultiIndexingAutomated.setup_methodc
Csæt|}g}d}d}d}d}d}d}    t|D]\}
}|dkr@q,t|tjr~|jtkr~d}|jdkrht||j7}||j7}q,|t    kr|    dkr|
}    q,tt|t
r¬|d7}q,t|tjsðztj|tjd}Wnt k
rätYnX|||
<n$|jjdkr|jjdkrtd    |jdkr$d}|d7}|d7}q,|j|dkrJt|dkrjd|krj| ¡|fS|    dk    rt
ddg|j|||    |    d<t|D]:\}}t|t
rÞtj| |j|¡}| d
|g¡qnP|dkr.| dtjdgtjdg¡| |jd|d|j|d¡}qt|tjr$|jtkr$|j|j|||jkrhtz*tjt |¡|j|||jd d} Wn4tk
rÆd}tjdg| ¡tjd} YnX|jdkr| |jd|t |j|||j¡f|j||jd¡}| }ntn2|jdkrV||j|ksR||j|krVt|jdkr||j|ks||j|krtt|dkrÂ|dddkrÂ||    krÂ|d |¡n|d7}| d|g¡q|dkr¾|s¾|dd}tt|j}g}| ddg¡d}d}|D]}|d7}|ddkr|d |dd¡||=|d8}t||t|ddD]}| |¡| |¡q||t|d7}q$|}|j||}d}|D]}|ddkrÀt|dkrìqÆ|j}|||t|dd}| |jd|t |¡  t!¡f|j|t|ddd¡}tj"|dd}|j#dkrÎ|rztt$|dd|D]@\}}|j#dkr¤qt %||k¡sÆt %||k¡rtqt|ddt|krPt &|¡dkr6ztj|dd|d d}Wntk
r2tdYnXntj|dd|dd}nt |j'| (¡|d}z0| |jd||j|j|dd¡}Wnt k
r°tYnX||j7}qÆ|j'|d|d}|d7}qÆ||fS)aêMimic multi dimensional indexing.
 
        Parameters
        ----------
        arr : ndarray
            Array to be indexed.
        indices : tuple of index objects
 
        Returns
        -------
        out : ndarray
            An array equivalent to the indexing operation (but always a copy).
            `arr[indices]` should be identical.
        no_copy : bool
            Whether the indexing operation requires a copy. If this is `True`,
            `np.may_share_memory(arr, arr[indices])` should be `True` (with
            some exceptions for scalars and possibly 0-d arrays).
 
        Notes
        -----
        While the function may mostly match the errors of normal indexing this
        is generally not the case.
        TrFNr r3rGrmz;arrays used as indices must be of integer (or boolean) typer8Únr°Úraise)ÚmoderPr\zinvalid index into 0-sizedÚwrapr;))ÚlistÚ    enumerater0rr7r4rWr,rrNrXrrDr]Úkindrlr=Úindicesr²Úappendr?Zravel_multi_indexZnonzeroÚ    ExceptionÚsumr    rÍrØÚinsertÚextendÚremoveZ    transposer>rÚ    broadcastr©ÚzipÚanyrÚtakeÚravel)rrBrZ
in_indicesÚno_copyZ    num_fancyZ    fancy_dimZerror_unless_broadcast_to_emptyr,Zellipsis_posrmrbZaxZ    flat_indxZnew_indicesZaxesZ
fancy_axesÚniZaiZ
orig_shapeZ
orig_slicerZ_indxÚ_sizeÚmirrrÚ_get_multi_index[s4
 
 
 
 
 
 
 
 
ÿ
&ÿ
 ÿþ""ÿþ
 
ÿþ"ÿ
ÿÿþ
 
 
z+TestMultiIndexingAutomated._get_multi_indexc
Csz| ||¡\}}Wnptk
r}zRtr4t |¡}tt||j|tt||j|dtrnt    |t |¡WY¢dSd}~XYnX| 
||||¡dS)aCheck 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.
        rN©r'rr
r¡r¢rr6rEryrÚ_compare_index_result©rrBrAÚ    mimic_getr#rðZ prev_refcountrrrÚ_check_multi_indexJs
z-TestMultiIndexingAutomated._check_multi_indexc
Csz| ||f¡\}}Wnptk
r}zRtr6t |¡}tt||j|tt||j|dtrpt    |t |¡WY¢dSd}~XYnX| 
||||¡dS)aWCheck 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`).
        rNr(r*rrrÚ_check_single_indexbs
z.TestMultiIndexingAutomated._check_single_indexcCsÖ| ¡}||}t|||jdkrl|jdkrltt ||¡|ktrl|r\tt     
|¡dntt     
|¡d| ¡}|d||<|jdkrdS|r²|jdkr²|d7}t||dS|j| ¡d7<t||dS)z4Compare mimicked result to indexing result.
        rr+rr_N) rlrr©r,rrZmay_share_memoryr
rr¡r¢Zflatr")rrBrAr+r#Zindexed_arrrGrrrr)zs&
 
 
z0TestMultiIndexingAutomated._compare_index_resultcCst d¡}t|t d¡dd|t d¡<t|d| |jtj|jtdf¡| |jtj|jtddf¡| |jtj|jtddf¡dS)Nr Tr r3rM)N.)rrrr,rZ
zeros_likerWr-rrrÚtest_booleans 
 
ÿÿÿz'TestMultiIndexingAutomated.test_booleanc    st ¡t ddt¡t ddtj¡dddD]b}|j|j|j|jg}|j||<t    |D]6}t
fdd|D}| |j|¡| |j |¡q^q4W5QRX| |jd¡| |j d¡tt|jjd    tt|jjd    d
tt|jjd
d
dgd
d
ftt|jjd
d
dgd
d
fd
dS)NÚerrorrÊcSst|to|dkS)Nr)r0rë)r~rrrÚisskip´sz8TestMultiIndexingAutomated.test_multidim.<locals>.isskip)rrr+c3s|]}|s|VqdSr)r)rÑrm©r0rrÚ    <genexpr>¼sz;TestMultiIndexingAutomated.test_multidim.<locals>.<genexpr>)rrrr)rrrrrrr )ÚwarningsÚcatch_warningsÚfilterwarningsÚDeprecationWarningrZVisibleDeprecationWarningrrr rr±r,rrGrrrEry)rZ
simple_posZtocheckrArr1rÚ test_multidim«s(
ÿ
z(TestMultiIndexingAutomated.test_multidimcCs&t d¡}|jD]}| ||¡qdS)Nr:)rr=rr-)rrrArrrÚtest_1dÉs
 
z"TestMultiIndexingAutomated.test_1dN)rrrÚ__doc__rr'r,r-r)r.r7r8rrrrr(sp rc@s8eZdZdZddZddZddZdd    Zd
dZdS) ÚTestFloatNonIntegerArgumentzÙ
    These test that ``TypeError`` is raised when you try to use
    non-integers as arguments to for indexing and slicing e.g. ``a[0.0:5]``
    and ``a[0.5]``, or other functions like ``array.reshape(1., -1)``.
 
    cCstt dggg¡}|t dg¡|ddg|ddddgf|dddddf|ddddddfdS)Nr rrr-rrrÚtest_valid_indexingÕsz/TestFloatNonIntegerArgument.test_valid_indexingcCs|t dggg¡}|dd|dd|dd|dd|ddd|ddd|ddd|ddddS)Nr rrr rr-rrrÚtest_valid_slicingßsz.TestFloatNonIntegerArgument.test_valid_slicingcCsjt dgg¡}tttj|dtttj|t d¡dftttj|dgdtttj|dgt d¡dS)Nr )r#r#rPr#rPr)rrrr'r?r!rr-rrrÚ test_non_integer_argument_errorsìs
z<TestFloatNonIntegerArgument.test_non_integer_argument_errorscCs4dd}tt|dgt d¡|dgt d¡dS)NcSs||Sr)r)rrGrrrÚmultöszRTestFloatNonIntegerArgument.test_non_integer_sequence_multiplication.<locals>.multr r+)rr'rrr1)rr>rrrÚ(test_non_integer_sequence_multiplicationôszDTestFloatNonIntegerArgument.test_non_integer_sequence_multiplicationcCsNt d¡}tttj|dtttj|dtttj|dtttj|ddS)N)r+r+r+çà?)r@r )r g@)gÉ?g333333ó?)rr5rr'Úmin)rrrrrÚtest_reduce_axis_float_indexýs
 
z8TestFloatNonIntegerArgument.test_reduce_axis_float_indexN)    rrrr9r;r<r=r?rBrrrrr:Îs
     r:c@s$eZdZddZddZddZdS)ÚTestBooleanIndexingcCszt dggg¡}tttj|dtttj|t d¡dftttjt d¡tt    tjtj
tttj|dgdfddS)Nr )TrPTrPrF)Úargs)rrrr'r?ruÚoperatorrAr    r6ZTrue_r!r-rrrÚ test_bool_as_int_argument_errorssz4TestBooleanIndexing.test_bool_as_int_argument_errorscsZt d¡djdkstdddgdddgdggfjdksDtttfd    d
dS)N)rr+r%)FT.)rrr+r%Trr rrþcsdddgdfS)NFrr .rrrrrrrzETestBooleanIndexing.test_boolean_indexing_weirdness.<locals>.<lambda>)rrVr²rÀrrrrrrÚtest_boolean_indexing_weirdnesss
(z3TestBooleanIndexing.test_boolean_indexing_weirdnesscsÀt d¡t dgdg¡ttdfddt dgddgg¡ttdfd    dt dgd
g¡ttdfddt d¡t dgdggg¡ttd fdddS)N)r+r+FrLztboolean index did not match indexed array along dimension 0; dimension is 3 but corresponding boolean dimension is 1csSr)rr)rÚidx1rrr#rzETestBooleanIndexing.test_boolean_indexing_fast_path.<locals>.<lambda>rKTcsSr)rr)rÚidx2rrr*rr:csSr)rr)rÚidx3rrr1r)r r rztboolean index did not match indexed array along dimension 1; dimension is 1 but corresponding boolean dimension is 2csSr)rr)rr~rrr9r)rrVrrrrr)rr~rHrIrJrÚtest_boolean_indexing_fast_paths,
ýýý
ýz3TestBooleanIndexing.test_boolean_indexing_fast_pathN)rrrrFrGrKrrrrrCsrCc@seZdZdZddZdS)ÚTestArrayToIndexDeprecationz7Creating an index from array not 0-D is an error.
 
    cCsRt dggg¡}tttjt dg¡tttj||dftttj|dg|dS)Nr rPr)rrrr'rErAr?r!r-rrrÚtest_array_to_index_error@sz5TestArrayToIndexDeprecation.test_array_to_index_errorN)rrrr9rMrrrrrL<srLc@seZdZdZddZdS)ÚTestNonIntegerArrayLikez¢Tests that array_likes only valid if can safely cast to integer.
 
    For instance, lists give IndexError when they cannot be safely cast to
    an integer.
 
    cCs>t d¡}tt|jddgtt|jddgf| g¡dS)Nr:r@gø?Ú1r)rr=rrrEr-rrrröPs
z"TestNonIntegerArrayLike.test_basicN©rrrr9rörrrrrNIsrNc@seZdZdZddZdS)ÚTestMultipleEllipsisErrorz/An index can only have a single ellipsis.
 
    csLt d¡ttfddttjtfdfttjtfdfdS)Nr:csdS)N)..rrrrrr`rz6TestMultipleEllipsisError.test_basic.<locals>.<lambda>rr+)rr=rrrErNrrrrrö^s
z$TestMultipleEllipsisError.test_basicNrPrrrrrQZsrQc@seZdZddZddZdS)ÚTestCApiAccesscCs²t td¡}tt|t d¡dtt|t d¡dtt|t d¡dtt|t d¡dtt|t d¡dt d¡}t|d||d|     dd    ¡}t|d
||d
dS)Nrrr:r{éõÿÿÿ©r:r:r%r réüÿÿÿ)
Ú    functoolsÚpartialrrrrrVr=rr?)rZ    subscriptrrrrÚtest_getitemfs
zTestCApiAccess.test_getitemcCsât td¡}tt|t d¡dtt|t d¡ddtt|t d¡ddtt|t d¡ddtt|t d¡ddtt|t d¡ddt d¡}||ddt    |ddk| 
d    d
¡}||ddt|dddgdS)Nr r:rrr{rSrTr%r rrP)rVrWrrr]rrVrr=rr?r)rrârrrrÚtest_setitemvs
zTestCApiAccess.test_setitemN)rrrrXrYrrrrrResrR)!r¡r3rVrErÅÚnumpyrZnumpy.core._multiarray_testsrÚ    itertoolsrZ numpy.testingrrrrrr    r
rrrßràrôrúrýrr:rCrLrNrQrRrrrrÚ<module>s8(I    C@0)77