mxiF ddlmZddlZddlmZddlZddlZddlZddl Z ddl m Z ddl m Z mZmZmZddlmZmZmZmZmZmZmZddlmZddlmZe j<dZd Z d Z!e jDjGd gd d Z$dZ%e jDjGdgddZ&dZ'e jDjGdgddZ(dZ)dZ*dZ+dZ,dZ-dZ.e jDj_eddZ0e jDjce dd Z2e jDj_ed!d"Z3d#Z4d$Z5d%Z6e jDjce d&d'Z7e jDjce d(d)Z8e jDjrd*Z:e jDjGd+gd,gd-gd.gd/gd0gd1gd2Z;y)3)StringION)ascii_uppercase)using_string_dtype) HAS_PYARROWIS64PYPYis_platform_arm)CategoricalIndex DataFrameIndex MultiIndexSeries date_rangeoption_context)Versioncxttjjdj dgdS)z&Dataframe with duplicate column names.)i)arbrcolumns)r nprandom default_rngstandard_normal[/home/htdocs/ttos/venv/lib/python3.12/site-packages/pandas/tests/frame/methods/test_info.pyduplicate_columns_framer s1  a 00;$ rct}t}|j||j}t j d}||k(sJy)Nbufzd RangeIndex: 0 entries Empty DataFrame )r rinfogetvaluetextwrapdedent)dfr#resultexpecteds rtest_info_emptyr+)sL B *CGGG \\^F H X  rc &d}tdtjjdj d|t i}t tjtdjtjjdj dd|t jd |d <|jt}|j| ||d d k(}t}|j| y) Ni int64rd)sizedtype abcdefghijr categoryr"d)r rrrintegersintrarraylisttakeastypeisnarr$)nr(r#df2s r'test_info_categorical_column_smoke_testr>8s A GRYY2215>>sRU>VW XB l#$)) II ! !! $ - -a!3 - G  fZ zN GGI *CGGG R ^s " #C *CHHHrfixture_func_name) int_frame float_framedatetime_framer float_string_framec|j|}t}|j||jj }t |dkDsJt}|j|dy)Nr"r2Fr#verbose)getfixturevaluerr$r% splitlineslen)r?requestframer#r)s rtest_info_smoke_testrLIsg  # #$5 6E *C JJ3J \\^ & & (F v;   *C JJ3J&rct}|jdgjd||jddgjd|tj|y)NArFrFr#Br")rreindexr$r )rAr#s rtest_info_smoke_test2rR^sf *C&++Es+Cc +00C0HKrz&num_columns, max_info_columns, verbose))r2r.T)r2 T)r2r2T)r2 F)r2Fczttjjdj d|f}t d|5t }|j||j}t }|j|||j}||k(sJ dddy#1swYyxYw)Nrdisplay.max_info_columnsr"rE) r rrrrrrr$r%) num_columnsmax_info_columnsrFrK io_defaultr) io_explicitr*s r#test_info_default_verbose_selectionr]hs bii++A.>>;?OP QE 24D E "Z  z "$$&j  {G 4'')!!! " " "s A!B11B:ct}d}d}ttjj dj d|f}|j d||j}d}||vsJ|j d||jd|j}t|dkDsJt|D]3\}}||cxkr ||zksnd ||z d } |j| r3Jy) NrWrTrOz- # Column Dtype --- ------ ----- r ) rr rrrrr$r%seek readlinesrI enumerate startswith) r#r/startrKresheaderlinesilineline_nrs r-test_info_verbose_check_header_separator_bodyrms *C D E bii++A.>>4yI JE JJtJ% ,,.C =F S== JJtJ%HHQK MMOE u:>>U#,4 A $ $!e)A&G??7+ ++,rz>size, header_exp, separator_exp, first_line_exp, last_line_exp))r$ # Column Non-Null Count Dtype $--- ------ -------------- ----- $ 0 0 3 non-null float64z$ 3 3 3 non-null float64)rSrnrorpz$ 10 10 3 non-null float64)ez% # Column Non-Null Count Dtype z%--- ------ -------------- ----- z% 0 0 3 non-null float64z% 100 100 3 non-null float64)r_z& # Column Non-Null Count Dtype z&--- ------ -------------- ----- z& 0 0 3 non-null float64z& 1000 1000 3 non-null float64)i'z' # Column Non-Null Count Dtype z'--- ------ -------------- ----- z' 0 0 3 non-null float64z' 10000 10000 3 non-null float64cjttjjdj d|f}t 5}|j dd||jj}ddddd}|^} } } } } | |k(sJ| |k(sJ| |k(sJ| |k(sJy#1swY3xYw)zETest header column, spacer, first line and last line in verbose mode.rr`T)rF show_countsr#N) r rrrrrr$r%rH)r/ header_exp separator_expfirst_line_exp last_line_exprKr# all_linestablerh separator first_linerest last_lines r%test_info_verbose_with_counts_spacingrsX bii++A.>>4yI JE 0s 4Ts ;LLN--/ 0 aOE6;3FIzD) Z    %% %  '' '  %% %00s 3B))B2ctdtddgdi}t}|j||j }t |j j}tjd|d}||k(sJy) NrrUri8r0r"a  RangeIndex: 2 entries, 0 to 1 Data columns (total 1 columns): # Column Non-Null Count Dtype --- ------ -------------- ----- 0 a 2 non-null int64 dtypes: int64(1) memory usage: z bytes ) r rrr$r%float memory_usagesumr&r')r(r#r)bytesr*s rtest_info_memoryrs CAd34 5B *CGGG \\^F "//#'') *E '  H X  rct}ttjj dj d}|j |t}|j |d|j}t|jdkDsJ|}tdd5t}|j ||j}||k(sJ dddy#1swYyxYw)Nr)rWrqr"rqr#max_colsr.rX) rr rrrrr$r%rIrHr)ior(r)r*s rtest_info_widers B 299((+;;HE FBGGGO BGGSG! [[]F v  " #c )) )H 2C 8" Z B!!! """s -4C++C4ct}tddggddg}|j||jj d}d|dk(sJd |d k(sJy) NrUg@rrr"Tz% 0 a 1 non-null int64 rWz% 1 a 1 non-null float64 )rr r$r%rH)rrKris r0test_info_duplicate_columns_shows_correct_dtypesrsg B 3xj3* 5E JJ2J KKM $ $T *E 3uQx ?? ? 3uQx ?? ?rc gd}i}d}t|D]G\}}tjjdj d|j |||<It |}t}|j||j}d}||vsJt|D]\}}d|dd |dd |dd |} | |vrJy) Nr-float64zdatetime64[ns]ztimedelta64[ns] complex128objectboolr2rr/r"zY # Column Non-Null Count Dtype --- ------ -------------- ----- rar4z z z non-null ) rdrrrr5r:r rr$r%) dtypesdatar<rjr0r(r#rgrhnames rtest_info_shows_column_dtypesrsF D Af%M5))''*33AA3>EEeLQM 4B *CGGG ,,.C 7  S==f%51Q%s1Q%wqe>%As{{rcttjjdj d}dD]u\}}t dd5t }|j|||j}t|jjd|k(sJ dddwdD]u\}}t dd 5t }|j|||j}t|jjd|k(sJ dddwd D]\}}t dd5t }|j|| |j}t|jjd|k(sJ dddt dd 5t }|j|| |j}t|jjd|k(sJ dddy#1swYxYw#1swYzxYw#1swYxYw#1swYxYw) Nr)r2rW))rWNrWF TrZrrE ))rNrrrW))rrW)rWrr) r rrrrrrr$r%rIstripsplit)r(len_rFr#rgrs rtest_info_max_colsrs  299((+;;GD EB<8 g . 2 8*C GGWG -,,.Csyy{((./47 77  8 88>8 g . 2 8*C GGWG -,,.Csyy{((./47 77  8 88, 8h . 2 8*C GGhG /,,.Csyy{((./47 77  8. 2 8*C GGhG /,,.Csyy{((./47 77  8 8 8 8 8 8 8 8 8 8 8s4 AHAH$AH10AH=H! $H. 1H: =I c gd}i}d}t|D]G\}}tjjdj d|j |||<It |}t}|j|d|jj}d|dvsJ|j|d |jj}d|dvsJ|j|d|jj}tjd |dsJ|jdddd fj|d|jj}tjd |drJgd }i}d }t|D]G\}}tjjdj d|j |||<It |}||_t ddgitdgt }|j|d|jj}tjd |dsJ|j|d|jj}tjd|dsJ|j#j%} t'||zdz|j(j*z} | | k(sJtj,|jj.dz} | tj,|j#k(sJ|j#j%|j#dj%k(sJt ddgdgj#dt ddgdgj(j*t dt1j2dgt5dgdg}|j(j*|j#d|j(j.j*|j#dj%} | dkDsJy)Nrr2rrT)r#rzmemory usage: Fzmemory usage: [^+]+\+rW)r-r-r-rr.rrUfoorindexdeepzmemory usage: [^+]+$rrNrrrrrr)rdrrrr5r:r rr$r%rHrematchilocrr rrrrIrnbytesr/valuesr from_productrange) rrr<rjr0r(r#rgdf_with_object_indexdf_sizeexp_sizesize_dfmems rtest_info_memory_usager@sF D Af%M5))''*33AA3>EEeLQM 4B *CGG$G' ,,. # # %C s2w && &GG%G( ,,. # # %C 3r7 ** *GG$G' ,,. # # %C 88,c"g 66 6GGArrENCd3 ,,. # # %Cxx0#b':: :4F D Af%M5))''*33AA3>EEeLQM 4BBJ$cA3ZueWF7ST#D9 ,,. # # %C 88,c"g 66 6#F; ,,. # # %C 88+SW 55 5oo##%G6{Q"RXX__4H h  ggbjj''(1,G bggboo/0 00 0 ??  "boo4o&@&D&D&F FF Fause,999E ause,2299  j--ueDk.BCcU BHHOOOO$OHHOO //t/ $ ( ( *C 7N7rz'on PyPy deep=True doesn't change result)reasonctddgitdgt}|jddj |jdj kDsJtdt dgti}|jdj |jj kDsJy NrrUrrrT)rrrr r rrrrr df_objects r$test_info_memory_usage_deep_not_pypyrs$cA3ZueWF7ST))4)@DDF  + +$ + 7 ; ; = > >3uF ;<=I  ! !t ! , 0 0 2Y5K5K5M5Q5Q5S SS Srz(on PyPy deep=True does not change resultctddgitdgt}|jddj |jdj k(sJtdt dgti}|jdj |jj k(sJyrrrs r test_info_memory_usage_deep_pypyrs$cA3ZueWF7ST))4)@DDF  , ,4 , 8 < < > ? ?3uF ;<=I  ! !t ! , 0 0 2i6L6L6N6R6R6T TT Trz PyPy getsizeof() fails by designc tdtjdgtdgdg}|j dj }|t j|z }t|dksJy) NrUrrrNrTrr.) r r rrrrsys getsizeofabs)r(rdiffs rtest_usage_via_getsizeofrsk  j--ueDk.BCcU B //t/ $ ( ( *C r" "D t9s??rc &t}tdtdgd}|j|d|j vsJt}tdtdt tdt }|j|d|j vsJt}tdtdt tdd }|j||rtrd|j vsJd|j vsJt}tdtdtjtd td g}|j|d|j vsJt}tdtdtjtd d d gg}|j||rtrd|j vsJyd|j vsJy) NrUab)rUrr`rrr"+ABCrstrr`rbar) rr r8r$r%r rrr rr)using_infer_stringr#r(s r test_info_memory_usage_qualifiedrs *C 1d4j :BGGG clln $$ $ *C 1d4jd5k0P QBGGG #,,.  *C 1d4jd5k0O PBGGGk#,,.(((clln$$$ *C  4:Z%<%test_info_memory_usage_bug_on_multiindex..memory_usages~~4~(,,..rr.20160101)periodsiddate)namesvaluerri)rIrr rr8rr rrrrunstackrr)rNMrr( unstackeds r(test_info_memory_usage_bug_on_multiindexrs/ A OA  # # o  :q ABVn E  "))''*::1q5AB% B 4 I 99  y//66 66 6  l95 55 5  "\"%5 5 << Index: 2 entries, A to B Data columns (total 2 columns): # Column Non-Null Count Dtype --- ------ -------------- ----- 0 1 2 non-null int64 1 2 2 non-null int64 dtypes: int64(2) memory usage: z50.0z48.0+z bytes )r rr$r%r&r'r)rr(r#r)r*s rtest_info_int_columnsrs 1v1a&)#s mark parametrizerLrRr]rmrrrrrrrskipifrxfailrrrrrrr single_cpurrrrrr sY "  -' " ' ',  "  ",,D$'P&Q'P&$,"$@88DRjD!JKTLTt8$NOUPUD!CDE"%J=2t8$EFG,%'0DE -F -$'   <