U °╒/eH<у@s▓ddlZddlZddlmZddlmZddlZddlZddl m Zddlm Z ddlmZddlmZeаdбdd ДZd dДZdd ДZddДddДddДddДddДddДddДddДddДddДddДddДddДddДddДddДddДd dДd!dДd"dДd#dДd$dДd%dДd&dДd'dДd(dДd)dДgZejаd*eбd+d,ДГZd-d.ДZejаd*eбejаd/бd0d1ДГГZejаd*eбejаd/бd2d3ДГГZd4d5ДZd6d7ДZd8d9ДZ d:d;ДZ!ejаdd?d@dAdBdCdDdEdFdGg бdHdIДГГZ"ejаdd?d@dAdBdCdDdEdFdGg бdJdKДГГZ#ejаd>dLdMgбdNdOДГZ$dPdQДZ%dRdSДZ&dTdUДZ'dVdWДZ(ejаdXdYdZgбd[d\ДГZ)d]d^ДZ*d_d`ДZ+ejаdadbdcddgбdedfДГZ,dgdhДZ-didjДZ.dS)kщN)┌deepcopy)┌product)┌ _numpy_123)┌ assert_eq)┌tokenizez dask.array.macCs╝tjjdddgdddgddН}tjjdddgdddgddН}tjjdddgdddН}t|Гt|ГkshtВt|Гt|Гks|tВt|Гt|ГksРtВt|Гt|ГksдtВt|Гt|Гks╕tВdS) NщщщTFщ й┌mask┌ fill_valuer)┌np┌ma┌masked_arrayr┌AssertionError)┌m┌m2Zm3йr·@/tmp/pip-unpacked-wheel-dbjnr7gq/dask/array/tests/test_masked.py┌test_tokenize_masked_arraysrcCs<tjjdddgdddgddН}tj|ddd Н}t||ГdS) Nrrr TFr r)rй┌chunks┌asarray)rrr┌da┌ from_arrayr)r┌dmrrr┌test_from_array_masked_arraysrcCsжtjjddgddgdН}tj||jddН}|аб}i}t||dН}tjаddgddgб}t ||Гt ||Гt |абtjjГsВtВt ||Гt |абtjjГsвtВdS) NrrrйrFr)┌memoT) rrrrr┌shape┌copyr┌masked_wherer┌ isinstance┌computer)┌t┌x┌yr┌y2┌xxrrr┌test_copy_deepcopy"s r*cCs|SйNrйr&rrr┌4єr-cCs tа|бSr+)r┌expm1r,rrrr-5r.cCsd|SйNrrr,rrrr-6r.cCs|dSr0rr,rrrr-7r.cCs|dSr0rr,rrrr-8r.cCs||Sr+rr,rrrr-9r.cCs||Sr+rr,rrrr-:r.cCs|dS)Nrrr,rrrr-;r.cCs|ddЕdfSйNrrr,rrrr-<r.cCs|ddЕdddЕfS)Nrr rr,rrrr-=r.cCs|jSr+)┌Tr,rrrr->r.cCstа|dбS)N)rrr)rZ transposer,rrrr-?r.cCs|абSr+й┌sumr,rrrr-@r.cCs|аtа|jdббSйNщ )┌dotr┌aranger r,rrrr-Ar.cCs|аtа|jdббSr5)r7rZeyer r,rrrr-Br.cCs$tj|tа|jddЕбddgdНS)Nr)rrйZaxes)r┌ tensordotrZonesr r,rrrr-Cr.cCs|jddНSйNrй┌axisr3r,rrrr-Dr.cCs|jddНSr;)┌maxr,rrrr-Er.cCs|jddНS)Nйrrr<r3r,rrrr-Fr.cCs|аtjбSr+)┌astyperZ complex128r,rrrr-Gr.cCs|аddДбS)NcSs|dSr0rr,rrrr-Hr.z..)┌ map_blocksr,rrrr-Hr.cCs |аdбSr1)┌roundr,rrrr-Ir.cCs$|а|jd|jd|jdfбS)Nrrr)┌reshaper r,rrrr-Jr.cCst|ГSr+)┌absr,rrrr-Kr.cCs|dkS)Nчр?rr,rrrr-Lr.cCs |аdбS)N)щrFrFйZrechunkr,rrrr-Mr.cCs |аdбS)NйrrrrGr,rrrr-Nr.┌funccCsrtjjdddН}d||dk<tjа|dб}||Г}||Г}t|tjа|dбГ|jrn|аб}t|t jj ГsntВdS)Nйrr rFйrrrйrrчЪЩЩЩЩЩ┘?)r┌randomr┌masked_equalr┌filledr r$r#rrr)rIr&r'r)┌yy┌zzrrr┌ test_basicRsrSc Cs▄tjjdddН}d||dk<tjjdddН}d||dk<tjа|dб}tjа|dб}ttj||dd Нtjаtj||dd НdбГttj||d d Нtjаtj||d d НdбГttj||dd Нtjаtj||dd НdбГdS)NrJrKrLrrM)rFr rrH)rrr9йrr)r?)rr)rrNrrOrr:rP)r&r'r)rQrrr┌test_tensordotcs$■■■rUzignore::numpy.ComplexWarningcCsВtjjdddН}tjjdddН}d||dk<tjа|dб}tj||gddН}tj||gddН}||Г}||Г}t||dddНdS) NrJrKrLrrMr<FйZ check_metaZ check_type)rrNrrO┌concatenater)rIr&r'rQ┌d┌s┌dd┌ssrrr┌test_mixed_concatenatezsr\cCsRtjjdddН}d||dk<ddД}|а|б}||Г}||Г}t||ddd НdS) N)rFr rFrKrLrrMcSstабdkrtjа|dбS|S)NrEr)rNrrrOr,rrrr-Сr.z#test_mixed_random..FrV)rrNrAr)rIrX┌fnrYrZr[rrr┌test_mixed_randomЛs r^cCsvtjjdddН}d||dk<tjа|dб}tjdddН}tj||gdd Н}|jd ksXtВ|аб}t |t jjГsrtВdS)Nйr r йщrarLrrM)r r)rarrr<)r щ)rrNrrO┌zerosrWr rr$r#rr)r'r&┌zrRrrr┌test_mixed_output_typeЪsrec Csьtаdddddgdбаdб}tаdddddgdб}tj|dd Н}tj|d d Н}tjа||б}t||g||gГD]\}}ttjа||б|Гqrttjа|dбtjа|dбГttjа |dбtjа |dбГttjа |dбtjа |dбГttjа|dбtjа|dбГttjа|dбtjа|dбГttjа |dбtjа |dбГttjаd|бtjаd|бГttjа|dk|бtjа|dk|бГtаtбП$tjа|dkddЕdf|бW5QRXttjа|ddбtjа|ddбГttjа|ddбtjа|ddбГttjа|dбtjа|dбГtаtбПtjа||бW5QRXtаtбПtjа||бW5QRX|аdб}tj|d <|d<tj|dd Н}ttjа|бtjа|бГtjа|dб}tjа|dб}ttjj|ddНtjj|ddНГdS)Nщ■ r6rrrщr_rarLrFF┌f8йrr)щraйr )r┌arrayrCrrr┌masked_greaterrrZmasked_greater_equalZmasked_lessZmasked_less_equalrOZmasked_not_equalr"┌pytest┌raises┌ IndexErrorZ masked_insideZmasked_outsideZ masked_values┌ ValueErrorr@┌nanZmasked_invalidZfix_invalid) r&r'┌dxZdy┌sol┌a┌bZmyZdmyrrr┌test_creation_functionsзs>&("" rwcCsДtаdddddgdбаdб}tj|dd Н}tjа|dб}tjа|dб}ttjа|бtjа|бГttjа|d бtjа|d бГdS)Nrfr6rrrrgr_rarLщ√ ) rrlrCrrrrOrrP)r&rs┌mx┌mdxrrr┌test_filled┘sr{cCs||аб}|tjjkr"||ksxtВnVt|Гt|Гks6tВt|dГrjtjа|j |j бt jа|б}tjа|б}t||ddНdS)NrT)Z equal_nan) r$rr┌maskedr┌type┌hasattrZtestingZassert_equalrrrPr)rurv┌resrrr┌assert_eq_maфs rА┌dtype)┌i8rh┌ reductionr4┌prodZmean┌varZstd┌minr>┌any┌allcCs tjаdбаddбdа|б}tj|ddН}tjа|dб}tjа|dб}t t|Г}t t|Г}t ||Г||ГГt ||ddН||ddНГt ||d d dН||d dНГt ||dd dН||ddНГt ||dd d dН||dd dНГt ||dd dН||ddНГt ||dd d dН||dd dНГdS)Nщ*rbr йrFrFrLrarr<TrF)┌keepdims┌split_everyйrЛr)r=rМ)r=rЛrМ)r=rЛr)rrN┌RandomState┌randr@rrrrm┌getattrrА)rБrГr&rsryrz┌dfuncrIrrr┌test_reductionsёs& ■■rТcCsLtjjddgddН}tj|ddН}tt|Г}tt|Г}t||Г||ГГdS)NrrTrF)r)rrrrrrРrА)rБrГr&rsrСrIrrr┌test_reductions_allmasked s rУZargminZargmaxcCsвtjаdб}tj|ddН}tjа|dб}tjа|dб}tt|Г}tt|Г}t||Г||ГГt||dГ||dГГt||dГ||dГГt||dГ||dГГdS)N)r r r )r rFrarLrMrrr)rrNrrrrmrРrА)rГr&rsry┌dmxrСrIrrr┌test_arg_reductionss rХcCsВtjаdбаdddб}tj|ddН}tjа|dб}tjа|dб}dD]8}t|j |d Н|j |d НГt|j |d Н|j |d НГqDdS) Nrrgщщ )щrarFrLrM)rrrr<)rrNrОrПrrrrmrАZcumsumZcumprod)r&rsryrФr=rrr┌test_cumulative+srЩcCsвtjаdб}tj|ddН}tjа|dб}tjа|dб}ttjа|бtjа|бГttjа|бtjа|бГttjа|бtjа|бГttjа|бtjа|бГdS)Nr_йr rFrLrM) rrNrrrrmr┌getmaskarrayZgetdataйr&rsryrФrrr┌test_accessors6srЭc Csftjаdбаdб}tj|ddН}tjtаdбddН}ddd|fg}t||g|ГD]░\}\}}ttjj ||d Нtjj ||d НГttjj ||d k|dНtjj ||d k|dНГttjj ||d k|dНtjj ||d k|dНГttjj ||dd Нtjj ||dd НГqPt аtбПtjj ||d НW5QRXt аtjj бП&tjj ||ddЕddЕfdНW5QRXdS)Nr_Zf4rЪrLrr)NN)rErErkrMrrh)r rБr r)rrNr@rrrlrrrrrnrorqZ MaskError)r&rs┌f1Zfill_values┌dataZdf┌frrr┌test_masked_arrayBs2 ■■■rбc Cs╥tjаdddб}tj|ddН}tjа|dб}tjа|dб}tjа|dбtjа|dбt||Гtjа|dбtjа|dбt||Гt а tбПtjа|dбW5QRXt а tбПtjа||бW5QRXdS) Nrr r_rЪrLr iЎ g@М╡xпD) rrN┌randintrrrrmZset_fill_valuerАrnro┌ TypeErrorrqrЬrrr┌test_set_fill_value`s rдrЛFTcCs╛tаddgddgddggб}tаdбаdб}tjj||dН}tjj||ddН}tаdd gб}tj|dd Н}tjj||d|dН}t rЮt |tjj||d|dНГn|s║t |tjj||dd НГdS)NTFrШйr rrrйrЯrrg╨?gш?rLr)┌weightsr=rЛ)rзr=)rrlr8rCrrrrZaveragerr)rЛrrЯru┌d_arзZ d_weightsZda_avgrrr┌&test_average_weights_with_masked_arrayusrйcCs~tаddgddgddggб}tаdбаdб}tjj||dН}tj|ddН}||}||}t||Гt|а бtjj ГsztВdS)NTFrШrеr)rrrL)rrlr8rCrrrrr#r$rr)rr&r|rsrrtrrr┌!test_arithmetic_results_in_maskedЗs rкcCs6tаdбаdб}|ddk|ddkB}tjа||б}tj|ddН}dD].}tjj||d Н}tjj||d Н}t||ГqHtjj|d dН}tjj|d dН}t||Гtjа dб}tj|ddН}ttjа|бtjа|бГtаdбаdб}tj|ddН}dD]8}tjj||d Н}tjj||d Н}t||t jdkd Нq°dS)Nщx)щr r rrF)rr rL)Nrrr<TrН┌win32)Zcheck_dtype)rr8rCrr"rr┌countrZ masked_all┌sys┌platform)rЯrr&rsr=rrtrrr┌ test_countУs( r▒┌funcnameZ ones_likeZ zeros_likeZ empty_likec Csкtаddgddgddggб}tаdбаdб}tjj||dН}tjj||ddН}ttj|Г}ttjj|Г}||Г}||Г}d|krЬt tjа |бtjа |бГn t ||ГdS) NTFrШrеrrrж┌empty)rrlr8rCrrrrР┌corerrЫ) r▓rrЯrurиZda_funcZnp_funcrrtrrr┌test_like_funcs░sr╡c Cs°tаdбаdб}tаdddgdddgdddggб}tjj||dН}tjj||ddН}|dk|dkf||f|d k|d kf|d k|d kffD]j\}}tjа|б}tjа|б}t|t |ГГs║t Вt|Гt|Гks╬t Вtt|ГГD]}t ||||Гq┌qИdS)Nщ )r r TFrrrжrFrfr)rr8rCrlrrrZnonzeror#r}r┌len┌ranger) rЯrrurи┌c1┌c2rtr┌irrr┌test_nonzero├s"№r╝c CsvtjjdddН}tjjddgdddgdН}d|d <|d <tjj||ddН}tjj||d Н}tjjdddНаtj б}tj |ddН}tjа|б}tjа|б}tt |ГГD]}t||||Гqж|dk|dkf||fdd dtjtjfddtjtjff D]|\}} d|ddЕdf|ddЕdff|ddЕ|ddЕffD]4\} }tjа||| б}tjа| ||б} t|| ГРq:qЇdS)Nr )щщ)┌sizeFTrE)rur┐┌prr`rК)rFra)rrrr╜)rFrLrarT)TTri)FFr╛)rrNrв┌choicerrrrlr@Zuint8r┌wherer╕r╖rZTrue_ZFalse_)r&rrXr'┌ertrr╗r╣r║Zb1Zb2Zw1Zw2rrr┌ test_where┘s2 ўFr─)/rNrпr!r┌ itertoolsrZnumpyrrnZ dask.arrayrlrZdask.array.numpy_compatrZdask.array.utilsrZ dask.baserZimportorskiprrr*Z functions┌markZparametrizerSrU┌filterwarningsr\r^rerwr{rАrТrУrХrЩrЭrбrдrйrкr▒r╡r╝r─rrrr┌sа х 2