d`(ddlmZddlZddlmZddlZddlZddl Z ddl m Z ddl Z ddlmZddlmZmZGddZdd Zdd Zdd ZddZeddZe jdgdddZddZdddZddZdS)) annotationsN)dedent) defchararray) formatting) requires_daskrequires_netCDF4ceZdZddZddZddZddZddZddZdd Z dd Z dd Z dd Z dd Z ddZdZddZejdddZddZddZddZddZeddZdS)TestFormattingreturnNonec<dtdftddffddtdfdtddffddtdfdtddffdtd tdftd dtdffd dtd tdfdtd dtdffd dtd tdfdtd dtdffd dtdtdfdtddtdffdtd tdtdftd dtdtdffddtdtdfdtddtdffdtd tdtdftd dtdtdffg }|D]D\}}}tj|dd}||ksJtj|dd}||ksJEdS)N) i)rr)r)rr)rr)rrr)rrr)rrr)rrr)rrrF)from_endT)slicer_get_indexer_at_least_n_items)selfcasesshapestart_expected end_expectedactuals <=eRRWXXXF!V++++=eRRVWWWF6)))))  * *cptjdddd}dD]@}tj||}|jd|}||ksJAtjtd5tj|dddddS#1swxYwYdS Ndrrr)rr r&zat least one item)matchr) nparangereshaper first_n_itemsflatallpytestraises ValueErrorrarraynr expecteds r!test_first_n_itemsz!TestFormatting.test_first_n_items0s #&&r1a00& . .A-eQ77Fz"1"~H&++-- - - - - ]:-A B B B / /  $UA . . . / / / / / / / / / / / / / / / / / /sB++B/2B/crtjdddd}dD]A}tj||}|j| d}||ksJBtjtd5tj |dddddS#1swxYwYdSr%) r*r+r,r last_n_itemsr.r/r0r1r2r-r3s r!test_last_n_itemsz TestFormatting.test_last_n_items:s #&&r1a00& . .A,UA66Fz1"##H&++-- - - - - ]:-A B B B / /  $UA . . . / / / / / / / / / / / / / / / / / /s B,,B03B0ctjd}d}tjdg}|D]1}tj||}||ksJ2dS)Nr&))rr)rr&)rrrrc)r*r+r4r last_itemr,)rr4r,r6rresults r!test_last_itemzTestFormatting.test_last_itemDsl #48RD>> & &A)%--*:*:;;FX%%%%% & &r#ctjddftjddftjddftjddftjddftjd d ftjddfd d d dtjddftjddftjddfg}|D]!\}}tj|}||ksJ"dS)Nz 2000-01-01T12z2000-01-01T12:00:00z 2000-01-01NaTz10 days 1 hourz10 days 01:00:00z-3 daysz-3 days +00:00:00z3 hoursz0 days 03:00:00)fooz'foo')sfoozb'foo')r1)g?z1.0g8EGr?z1.123g1Ղ-?z1.011gs~86@z22.22) pd Timestamp Timedeltar*float16float32float64r format_itemrritemr6r s r!test_format_itemzTestFormatting.test_format_itemNs \/ * *,A B \, ' ' 6 \% % ( \* + +-? @ \) $ $&9 : \) $ $&7 8 \% % (     Z   ) Z " "G , Z " "G ,  $ & &ND(+D11Fv%%%%% & &r#ctjdtjddzdftjdtjddzdftjdtjdd zd ftjgd d ftjgd dfgddfg}|D]4\}}dt j|}||ksJ5dS)NrrDz0 days 1 days 2 days 3 daysrhz#00:00:00 03:00:00 06:00:00 09:00:00msz100:00:00 00:00:00.500000 00:00:01 00:00:01.500000)rB0s1srBzNaT 00:00:00 00:00:01 NaT)z 1 day 1 hourz1 dayz0 hoursz/1 days 01:00:00 1 days 00:00:00 0 days 00:00:00rrrz1 2 3 )r*r+ timedelta64rE to_timedeltajoinr format_itemsrLs r!test_format_itemsz TestFormatting.test_format_itemscs  Yq\\BN1c22 24Q R ! r~a5555   ! r~c4888C _777 8 8:U V D D DEEA YY  "$ & &ND(XXj5d;;<z9TestFormatting.test_format_array_flat..s@@@m@@@r#z'hello world hello...)rformat_array_flatr*r+rZrangeasarray)rr r6long_strs r!test_format_array_flatz%TestFormatting.test_format_array_flatys -binna@@6!!!!-binna@@6!!!!-binnbAA6!!!!-binnbAA"6!!!!-binnbAA$6!!!!-bi.>.>CC 6!!!!-bi.>.>CC!6!!!!-billA>>6!!!!-binnbAA 6!!!!-billA>>6!!!!-billA>>6!!!!-billA>>6!!!!-billA>>6!!!!-billA>>6!!!!HH@@U3ZZ@@@AAB-bj(.D.DbII*6!!!!!!r#crtjdddksJtjdddksJdS)N abcdefghijzabcde...ßr)r pretty_printrs r!test_pretty_printz TestFormatting.test_pretty_printsD&|Q77:EEEE&tQ//4777777r#c<tjdddksJdS)Nrtr)rmaybe_truncatervs r!test_maybe_truncatez"TestFormatting.test_maybe_truncates&(r22d::::::r#cd}tjt5tj|ddddS#1swxYwYdS)Nz2021-12-06 17:00:00 00)r0r1r2rformat_timestamp)rr6s r!+test_format_timestamp_invalid_pandas_formatz:TestFormatting.test_format_timestamp_invalid_pandas_formats+ ]: & & 2 2  ' 1 1 1 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2s>AAcddlm}|ddd}d}tj|}||ksJ|ddd}d}tj|}||ksJdS) Nr)datetimeirbrz 1300-12-01iz 2300-12-01)rrr|)rrdater6r?s r!#test_format_timestamp_out_of_boundsz2TestFormatting.test_format_timestamp_out_of_boundss%%%%%%xb!$$,T22!!!!xb!$$,T22!!!!!!r#c.tjdd}tjdd}tjdd}tjdd}|dksJt|dksJ|dsJd |vsJd |vsJdS) Nkeyz Short stringaVery long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string Very long string z z z key: Short stringPr^  )rsummarize_attrlenendswith)rshortlongnewlinestabss r!test_attribute_reprz"TestFormatting.test_attribute_reprs)%@@(0IJJ,UH==(99/////4yyB}}U#####8####4r#cddlm}Gdd|}ddg}||}d}tj||d}||vsJd|vsJd |_tj||d}||vsJ|d |vsJdS) Nr)IndexceZdZdZdZdS)3TestFormatting.test_index_repr..CustomIndexc||_dSNnames)rrs r!__init__z.CustomIndex.__init__s " r#cd|jdS)NzCustomIndex(coords=)rrvs r!__repr__z.CustomIndex.__repr__s:TZ::::r#N)__name__ __module__ __qualname__rrrgr#r! CustomIndexrs2 # # # ; ; ; ; ;r#rxyr) col_widthc>dd|jdS)Nz CustomIndex[z, ])rZr)r max_widths r!z0TestFormatting.test_index_repr..s $K499TZ3H3H$K$K$Kr#(r)xarray.core.indexesrrsummarize_index _repr_inline_)rrr coord_namesindexnamenormalinlines r!test_index_reprzTestFormatting.test_index_reprs------ ; ; ; ; ;% ; ; ;Cj  K((+D%2FFFv~~~~&&&& L K !+D%2FFFv~~~~""R"00F::::::r#c tjtjgdgdgddtjddgdtjgdddd d d  }tjtjd dgddtjddgddtjd dgdfdddi }tjdkrdnd}t d||fz}tj||d} ||ksJn,#t$r|| ddksJYnwxYwtj dtjgddddi}tj dtjgdgdgd}t d}tj||d} ||ksJdS#t$r || ddksJYdSwxYw)NrVrrint64dtyperrabU1mdescunits description)dimscoordsattrsrrrcrlabelrkglittle<>a Left and right DataArray objects are not identical Differing dimensions: (x: 2, y: 3) != (x: 2) Differing values: L array([[1, 2, 3], [4, 5, 6]], dtype=int64) R array([1, 2], dtype=int64) Differing coordinates: L * x (x) %cU1 'a' 'b' R * x (x) %cU1 'a' 'c' Coordinates only on the left object: * y (y) int64 1 2 3 Coordinates only on the right object: label (x) int64 1 2 Differing attributes: L units: m R units: kg Attributes only on the left object: description: desc identicalz , dtype=int64rctitlez test Variablea Left and right Variable objects are not equal Differing dimensions: (x: 3) != (x: 2, y: 3) Differing values: L array([1, 2, 3], dtype=int64) R array([[1, 2, 3], [4, 5, 6]], dtype=int64)equals) xr DataArrayr*r4sys byteorderrrdiff_array_reprAssertionErrorreplaceVariable)rda_ada_brr6r vavbs r!test_diff_array_reprz#TestFormatting.test_diff_array_reprs| Hiii+7 ; ; ;XsCj555Xiiiw777 77   | HaV7 + + +XsCj555rxAg>>>?D/   =H44CC#  !,)$- %  4+D$ DD CX%%%%% C C CX--orBBBBBBBB C[ )))7333g5O  [RXyyy))).DG%T%T%T U U /   +BH== CX%%%%%% C C CX--orBBBBBBBBB Cs$D&D76D7=G&G10G1errorcdtjddgi}ddi}td}t j||d}||ksJdtjddgi}td}t j||d}||ksJdtjgd i}td }t j||d}||ksJdS) Nattrrrzb Differing attributes: L attr: [0 1] R attr: 1 rrzh Differing attributes: L attr: [0 1] R attr: [-3 5] )rrrzh Differing attributes: L attr: [0 1] R attr: [0 1 2] )r*r4rstriprdiff_attrs_repr)rattrs_aattrs_br6r attrs_cs r!test_diff_attrs_repr_with_arrayz.TestFormatting.test_diff_attrs_repr_with_arrayEs628QF++,1+   %'' +GWhGG6!!!!28RG,,-   %'' +GWhGG6!!!!28III../   %'' +GWhGG6!!!!!!r#c dtjdtjgdgdgdfdtjddgdfd tjd d gd tjgdddd dd}tjddtjddgdfidtjd dgd ddifdtjddgdfdddi}tjdkrdnd}t d||fz}tj||d}||ksJdS)NrrVrrrrrr)var1var2rrrrrr) data_varsrrrrrrsourcerrrrrrra Left and right Dataset objects are not identical Differing dimensions: (x: 2, y: 3) != (x: 2) Differing coordinates: L * x (x) %cU1 'a' 'b' R * x (x) %cU1 'a' 'c' source: 0 Coordinates only on the left object: * y (y) int64 1 2 3 Coordinates only on the right object: label (x) int64 1 2 Differing data variables: L var1 (x, y) int64 1 2 3 4 5 6 R var1 (x) int64 1 2 Data variables only on the left object: var2 (x) int64 3 4 Differing attributes: L units: m R units: kg Attributes only on the left object: description: descr) rDatasetr*r4rrrrdiff_dataset_repr)rds_ads_brr6r s r!test_diff_dataset_reprz%TestFormatting.test_diff_dataset_reprksz#RXyyy))).DG%T%T%TUbh1vW===> XsCj555Xiiiw777 77    zRXq!fG%D%D%DEF28S#Jd;;;h]KrxAg>>>?D/    =H44CC#  !,)$- %  4-dD+FF!!!!!!r#ctjgdgdd}tjdgd|d<|d}tj|}t d}||ksJt ||ksJt||ksJ|}||ksJtjd 5tj|d}t d }||ksJ ddddS#1swxYwYdS) NrV)rCbar)rrtestr)rrzk array([0]) Dimensions without coordinates: testFdisplay_expand_datazn 0 Dimensions without coordinates: test) rrrr array_reprrreprstr set_options)rdsds_12r r6s r!test_array_reprzTestFormatting.test_array_reprs ZyyyCC D D D\1#F3336 6 &u-- 0  !!!!E{{h&&&&5zzX%%%%!!!! ^ 6 6 6 & &*2f:66F4HX%%%%% & & & & & & & & & & & & & & & & & &s42C44C8;C8ctjdddg}tj|tjd5tj|ddddS#1swxYwYdS)NrrrFr)rrrrr)rvars r!test_array_repr_variablez'TestFormatting.test_array_repr_variablesk#1v&&c""" ^ 6 6 6 ' '  !# & & & ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' 'sA##A'*A'cVtjdddg}||jd<tj|tjddgdg}||jd<tj|||jd<||jd<tj|tj|dS)Nrrrr)rrrrrr)rrdas r!test_array_repr_recursivez(TestFormatting.test_array_repr_recursivesk#1v&& #c""" \1a&u - - - b!!! # c"""b!!!!!r#c tjtjd}t |dt |ksJt |ddksJt |ddksJddlm}tj|d}t |dt |ksJtjt5}t |ddddn #1swxYwYdt|j vsJtjdd g}tjt5}t |ddddn #1swxYwYd t|j vsJdS) Nrrcdrdz.2fz0.00z#unsupported format string passed tog?g?z#Using format_spec is only supported) rrr*r4formatr dask.arrayr0r1 TypeErrorrvalueNotImplementedError)rrrexcinfos r!test_array_scalar_formatz'TestFormatting.test_array_scalar_formatsl28A;;''c2$s))++++c33&&&&c5!!V++++ l288A;;''c2$s))++++ ]9 % %  3                  4GM8J8JJJJJlC:&& ]. / / 7 3                  4GM8J8JJJJJJJs$C<<DDE++E/2E/Nr r )rrrr"r7r:r@rNr\rprwrzr}rrrrr0markfilterwarningsrrrrrrr rgr#r!r r s****>////////&&&&&&&&*&&&&,>">">">"@8888;;;;2222 " " " "     ;;;2MCMCMCMC^ [((#"#"#")(#"J2"2"2"2"h!&!&!&!&F''''""""$KKK]KKKr#r r r cGdd}|tjddgd}tjd|}d}t j|d}|||ksJdS) NcdeZdZdZdZdZed dZedZedZ d S) @test_inline_variable_array_repr_custom_repr..CustomArrayc"||_||_dSr)rr)rrrs r!rzItest_inline_variable_array_repr_custom_repr..CustomArray.__init__sDJDIIIr#cfd|jd|j}t||kr d|jd}|S)N(z) z) ...)rrr)rwidth formatteds r!rzNtest_inline_variable_array_repr_custom_repr..CustomArray._repr_inline_sD5DI5555I9~~%%0 000  r#ctSr)NotImplemented)rargskwargss r!__array_function__zStest_inline_variable_array_repr_custom_repr..CustomArray.__array_function__s! !r#r tuple[int, ...]c|jjSr)rrrvs r!rzFtest_inline_variable_array_repr_custom_repr..CustomArray.shape :# #r#c|jjSr)rrrvs r!rzFtest_inline_variable_array_repr_custom_repr..CustomArray.dtyperr#c|jjSr)rndimrvs r!rzEtest_inline_variable_array_repr_custom_repr..CustomArray.ndim s :? "r#N)r r) rrrrrrpropertyrrrrgr#r! CustomArrayrs       " " "  $ $ $  $  $ $  $  # #  # # #r#r!rrrrrr)r*r4rrrinline_variable_array_reprr)r!rvariablerr s r!+test_inline_variable_array_repr_custom_reprr$s########6 K"b**C 0 0E{3&&HI  28r J J JF U((33 3 3 3 3 3 3r#c"tj}tjd5t t tjddksJ dddn #1swxYwYtj|ksJdS)Nr) thresholdrRr()r*get_printoptionsrset_numpy_optionsrrr+)original_optionss r!test_set_numpy_optionsr*s*,,  % 3 3 3//4 #''((3.....///////////////   $4 4 4 4 4 4 4s5A++A/2A/c tjdtjddtjdddtjddddtjdddg}|D]4}tj|d dz}|d ksJ5tjd}d tj|vsJtjd 5d tj|vsJ ddddS#1swxYwYdS) NrRrrrr`rr&rrr^)display_values_threshold) r*randomrandnrshort_array_reprcountr+rr)rr4 num_linesarray2s r!test_short_array_reprr4!sr  B 2r"" 2r1%% Q""  E/66< [300 values with dtype=int64] Dimensions without coordinates: testz 0 1 2 3 4 5 6 7 8 9 10 11 12 ... 288 289 290 291 292 293 294 295 296 297 298 299 Dimensions without coordinates: test) rrr*r+r to_netcdfopen_dataarrayrrrcomputer)tmp_path arr_to_storearrr r6 arr_loadeds r!test_repr_file_collapsedrC>s< #RX > > >VLLLL8i/ BBB 8i/ 0 0"C!:::""c 0  !!!![[]] $$&& 0  !!!!!-"""""""""""""""""""""""""""""""""s7&Dz&test__mapping_repr..es   aQ   r#c<i|]}|tjddgS)rr)r*r4)rh_cs r!rLz&test__mapping_repr..fs( / / /rb"(Aq6"" / / /r#rr)rdatarr)display_max_rowsrcg|]}|v| SrgrgrhvrFs r!rjz&test__mapping_repr..u888a1r#cg|]}|v| SrgrgrRs r!rjz&test__mapping_repr..|rTr#cg|]}d|v| S)rHrg)rhrSs r!rjz&test__mapping_repr..s666A1r#F)rPdisplay_expand_coordsdisplay_expand_data_varsdisplay_expand_attrsz Dimensions:rrmax_rowsz rz) Coordinates: (z) Data variables: (z) Attributes: (r)raddr*r+astyperdictzipitemsrrr4rrrr dataset_reprsplitrmindata_vars_reprr coords_reprr_calculate_col_width variablesrudim_summary_limitedr)rPn_varsn_attrrrrrrrrSrNrsummary len_summarydata_vars_print_sizer r dims_start dims_valuesr6rFs @r!test__mapping_reprrp\s  IBIa$8$8$?$?$D$DEEA")Av"6"6"="=c"B"BCCA")Av"6"6"="=c"B"BCCA  1   E / /Q / / /FIQ ''  2|1a&!!Q%::    ! I  BBH )9 : : :%%)"--33D998888g888'll "#3[AA22222+BL99??EE8888g888'll f$$$$(3399$??66g666'll f$$$$$'%%%%%%%%%%%%%%%* )#!&"    "" (,,3BLAA ,]IFF  4 )a-2B        (##!!!!!'""""""""""""""""""s&C=JJJ3BMM M ctjddgdfi}||jd<tj|tjddgdfi}||jd<||jd<tj|dS)NrrrVrrr)rrrrra)rds2s r!test__mapping_repr_recursiverss S3 *+ , ,BBHTNB *cC+, - -CBHTNCIdOC     r#r& n_elementsintc"d}t|}d}ddtj|DD}tjj|t|d}||z}||ksJdS)Na Dimensions without coordinates: dim_0: 3, dim_1: 3, dim_2: 3, dim_3: 3, dim_4: 3, dim_5: 3, dim_6: 3, dim_7: 3, dim_8: 3, dim_9: 3, dim_10: 3, dim_11: 3, dim_12: 3, dim_13: 3, dim_14: 3, dim_15: 3, dim_16: 3, dim_17: 3, dim_18: 3, dim_19: 3, dim_20: 3, dim_21: 3, dim_22: 3, dim_23: 3, ... dim_76: 3, dim_77: 3, dim_78: 3, dim_79: 3, dim_80: 3, dim_81: 3, dim_82: 3, dim_83: 3, dim_84: 3, dim_85: 3, dim_86: 3, dim_87: 3, dim_88: 3, dim_89: 3, dim_90: 3, dim_91: 3, dim_92: 3, dim_93: 3, dim_94: 3, dim_95: 3, dim_96: 3, dim_97: 3, dim_98: 3, dim_99: 3z Dimensions without coordinates: c"g|] \}}|d| S)z: rg)rhrKrSs r!rjz+test__element_formatter..s311  r#ci|]}d|d S)dim_rrgrJs r!rLz+test__element_formatter..s N N N1Q N N Nr#rbrZ) rr*r+r`rcorer_element_formatterr)rtr6introelementsvaluesr s r!test__element_formatterrs RHhH .E N N *8M8M N N N T T V VHW  2 2CJJ3FV^F v      r#cddlm}Gdd|}|tjddg}t j|}||dS)Nr)LazilyIndexedArrayceZdZddZdS)Etest_lazy_array_wont_compute..LazilyIndexedArrayNotComputableNc td)Nz%Computing this array is not possible.)r)rrs r! __array__zOtest_lazy_array_wont_compute..LazilyIndexedArrayNotComputable.__array__s%&MNN Nr#r)rrrrrgr#r!LazilyIndexedArrayNotComputablers. O O O O O Or#rrr)xarray.core.indexingrr*r4rrr _repr_html_)rrrArs r!test_lazy_array_wont_computers777777OOOOO*<OOO * )"(Aq6*:*: ; ;C ,s  CLLNNNOOr#r )r&)rtrur r ) __future__rrtextwraprnumpyr*pandasrEr0 numpy.corerxarrayr xarray.corer xarray.testsrrr r$r*r4r8rCr  parametrizerprsrrrgr#r!rs""""""  ######""""""88888888aKaKaKaKaKaKaKaKH"4"4"4"4J5555<<<<,"""":&;;;:":":" :"z ! ! ! !8      r#