gL ifJddlZddlZddlmZddlmZddlmZddl Z ddl m Z m Z m Z mZmZddlmZddlmZmZmZmZmZmZddlmZddlmZGd d Zej>jAd ejBdd gejD ejFdfejBddgejFdfe jHddge jHdfe jJddgdedfe jLddgdee jNjPjPjSdfe jTjWgdedfe jJddgedfe jXd ged!fgd"Z-ej>jAd#ejBgd$gd%Z.d&Z/ej>jAd'e jHddgd(fej`d)d*gd+ d,fe jBdejbgd- d.fejVdd gd/fedd gd0fej`ejBd d1gd d,fej`ejBd2d3gd4 ed d,fgd5Z2d6Z3ej>jAd7ejBd d1gejD ejBd d1gejD d8fe jHddgejBddge4 d9fe jNjjjmd)d*gd:ejBe jnd)d:e jnd*d:gd9fe jBdejbgd- ejBdejbgd9fejVgdejBe jpdd e jpd d1ge4 d9fedd gejBdd gejD d9fej`ejBd)d*gd4 ejBd)d*gd4 d8fej`ejBd;dded?dgd9fej`ejBdd@gdA jwd!ejBdd@gd! d8fe jHedBd1dCDejBedBdCedEdCgd9fg dFZjAdGd8d9gej>jAd#ejBgd$d ejBgdHe4 gdIZ=ej>jAdGd8d9gdJZ>ej>jAdKgdLdMdgdNfed)ed)e j~gded)ejdOgdPzfgdQZAej>jAdRgdSgdTeBddUdVejbdWgfgdSgdTeBejbdUdVejbdWgfdUdVejbdWggdXeCdgdYfed)ed)e j~gdedZfded[fd ed)fgded)ejdOgdPzfgd\ZDd]ZEej>jAd^gd$gdLd_ejBgd$ejBd d1ejbgd_e jBgd$e jBgdLd_gej>jAd`eBejbfe4dfgdaZFej>jAdbde jBgdLiejBdUgdVgejbggeB fgd$gd$d_ejBd d gd1d1gdPdPggeB fgdcZGddZHGdedfZIy)gN) HAS_PYARROW)np_version_gt2)DatetimeTZDtype)CategoricalIndexSeries Timedelta Timestamp date_range) DatetimeArray IntervalArrayNumpyExtensionArray PeriodArray SparseArrayTimedeltaArray)StringArrayNumpySemantics)ArrowStringArrayNumpySemanticsceZdZdefdefdefdefdefdefdefdefd efd efd efd efd efdefgZejjdeejjdddddggddZ ejjdde dfdedfde dfdedfgejjddddd ggdd!Z ejjded"Zejjdedefdefgzd#Zejjdd$d%d&d'ggdd(Zd)Zd*Zd+Zd,Zy-).TestToIterableint8int16int32int64uint8uint16uint32uint64float16float32float64datetime64[ns]zdatetime64[ns, US/Eastern]ztimedelta64[ns]z dtype, rdtypemethodc"|jSNtolistxs g/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/pandas/tests/base/test_conversion.pyzTestToIterable.6 ahhjc"|jSr#to_listr&s r(r)zTestToIterable.7 aiikr+ct|Sr#listr&s r(r)zTestToIterable.8 d1gr+c4t|jSr#r2__iter__r&s r(r)zTestToIterable.9d1::<(r+)r%r.r2iter)idsc|}|dk(rJt|tjr0tjt d5|dg|dddy|dg|}||d}t ||sJy#1swYyxYwNrzfloat16 indexes are not matchdtyper) issubclasspdIndexpytestraisesNotImplementedError isinstance)selfindex_or_seriesr!r@rdtypetypsresults r( test_iterablezTestToIterable.test_iterable2s{ I *S"((";2:TU &QCu% &  5 !1&&)))  & s A77Bzdtype, rdtype, objobjectar>categoryc"|jSr#r$r&s r(r)zTestToIterable.Vr*r+c"|jSr#r-r&s r(r)zTestToIterable.Wr/r+ct|Sr#r1r&s r(r)zTestToIterable.Xr3r+c4t|jSr#r5r&s r(r)zTestToIterable.Yr7r+cP|}||g|}||d}t||sJy)Nr?r)rG) rHrIr!r@rJobjrKrLrMs r(!test_iterable_object_and_categoryz0TestToIterable.test_iterable_object_and_categoryJs42 U #1&&)))r+ctdg|}tt|j\}}t ||sJtt|j\}}t ||sJy)Nr>r?)rnextr8itemsrG)rHr@rJrL_rMs r(test_iterable_itemsz"TestToIterable.test_iterable_itemshsb A3e $aggi) 6&&)))aggi) 6&&)))r+c6|}|dk(rJt|tjr0tjt d5|dg|dddy|dg|}|j td}t|ts|f}||vsJy#1swYyxYwr;) rArBrCrDrErFmaptyperGtuple)rHrIr@rJrKrLrMs r(test_iterable_mapz TestToIterable.test_iterable_mapts  I *S"((";2:TU &QCu% &  5 !tQ&%(YF & s BBc"|jSr#r$r&s r(r)zTestToIterable.r*r+c"|jSr#r-r&s r(r)zTestToIterable.r/r+ct|Sr#r1r&s r(r)zTestToIterable.r3r+c4t|jSr#r5r&s r(r)zTestToIterable.r7r+c|ttdtdg}||d}t|tsJy)Nz 1999-12-31z 2000-12-31r)rr rG)rHr!irMs r(test_categorial_datetimelikez+TestToIterable.test_categorial_datetimelikes: i 5y7NO P1&),,,r+c:tdtdg}t|jj|}|jd|dk(sJt ||D]=\}}t |tsJ|jJ||k(sJ|j|k(r=Jy)N 2011-01-01 2011-01-02 datetime64[] r rdtas_unitr@ziprGtzunitrHrtvalsserresexps r(test_iter_box_dt64z!TestToIterable.test_iter_box_dt64s,'<)@ATloo%%d+yyk$q1111C $HCc9- --66> !>#: :88t# ##  $r+c\tddtddg}t|jj|}|jd|dk(sJt ||D]J\}}t |tsJ|j|jk(sJ||k(sJ|j|k(rJJy)Nrk US/Easternrsrlrmz , US/Eastern]rorus r(test_iter_box_dt64tzz#TestToIterable.test_iter_box_dt64tzs l| 4 l| 4 Tloo%%d+yyk$}====C $HCc9- --66SVV# ###: :88t# ##  $r+ctdtdg}t|jj|}|jd|dk(sJt ||D]/\}}t |tsJ||k(sJ|j|k(r/Jy)Nz1 daysz2 daysz timedelta64[rn)rrrprqr@rrrGrtrus r(test_iter_box_timedelta64z(TestToIterable.test_iter_box_timedelta64s(#Yx%89Tloo%%d+yyl4&2222C $HCc9- --#: :88t# ## $r+c(tjddtjddg}t|}|jdk(sJt ||D]9\}}t |tjsJ|j dk(sJ||k(r9Jy)NrkMfreqrlz Period[M]ME)rBPeriodrr@rrrGr)rHrvrLrxrys r(test_iter_box_periodz#TestToIterable.test_iter_box_periods ,S1299\PS3TU 4Lww+%%%At  HCc299- --88t# ###: : r+N)__name__ __module__ __qualname__intfloatr rdtypesrDmark parametrizerNrOrXr]rbrirzr~rrr+r(rrs   # # # # 3 3 3 E E E 9% %y1 I&F" [[_f5 [[ !  (  2  * 6 * [[ vs # sA   % a   [[ !  (  2  * & * [[_f5 *6 * [[Hc?Z4E"FF     [[ !  (  2 - - $ $$r+rzarr, expected_type, dtyper>r?rrPbrOrQ20172018 US/Centralr}zdatetime64[ns, US/Central]iiYrzY-DEC)rr>intervalr ld( zm8[ns]c|r|dk(rtrtnt}t|j}t j |j}t||usJt|t|usJtj||y)NrO) rrrr_valuesrBrCr`tm assert_equal)arr expected_typer@using_infer_stringl_valuesr_valuess r(test_values_consistentrsv<ex/.9 *?X c{""Hxx}$$H >] ** * >T(^ ++ +OOHh'r+rr>rctt|}|j}t|}tj||yr#)rarrayr rassert_extension_array_equal)rrwrMexpecteds r(test_numpy_arrayrs. +C YYF"3'H##FH5r+c,t|}|j}tj|jdk(rt |t sJytj|jdk(rt |tsJyt |tsJy)Nr?rm) rrnpr@kindrGr rr )any_numpy_dtyperwrMs r(test_numpy_array_all_dtypesrst  'C YYF xx %%,&-000 / " ' '3 .&.111&"5666r+z arr, attr_codes20002001z period[D]_ndarrayInt64_data_left_sparse_valuesrz2000-01-01T12:00:00z2000-01-02T12:00:00M8[ns]cp|}||dj}|rt||}t||}||usJy)NFcopy)rgetattr)rattrrIrequestboxrMs r( test_arrayrsB2 C 5 ! ' 'F c4 & S==r+ctjjdgddgg}d}tjt |5|j dddy#1swYyxYw)NArPrz&MultiIndex has no single backing arrayr<)rB MultiIndex from_productrDrE ValueErrorr)idxmsgs r(test_array_multiindex_raisesr#sP -- $ $sec3Z%8 9C 2C z - s  AA!zarr, expected, zero_copyTFDz2000-01-01T06:00:00z2000-01-02T06:00:00UTC 2000-01-01 2000-01-02l qa i8z 2016-01-01z US/Pacificperiodsrsz 2016-01-02c|}tjd5||}dddj}tj||t j |}tj||t j |d}t j |d}t j||rJtsy|s>d} tjt| 5t j |ddddyt j |d} t j |d} t j| | sJy#1swY/xYw#1swYyxYw)NTrzGStarting with NumPy 2.0, the behavior of the 'copy' keyword has changedr<F) rassert_produces_warningto_numpyassert_numpy_array_equalrasarrayrmay_share_memoryr FutureWarning) rr zero_copyindex_or_series_or_arrayrthingrM result_cp1 result_cp2rresult_nocopy1result_nocopy2s r( test_to_numpyr*s(x #C # #D )C^^ F1 ZZ F1%d+J%d+J"":z:: :  W  ' ' S A ( HHU ' ( (%e4%e4"">>BBB;, ( (s E&EEE$ as_series)rPrcctj|d}|rt|jd}|j }|rF|j t k(r3|j jdk(rtj||dusJtj||dusJ|j d}|rF|j t k(r3|j jdk(rtj||dusJtj||dusJ|j d}tj||dusJy)NFrpyarrowT) rBrCrvaluesrr@rOstorager shares_memory)rrrrWrMs r(test_to_numpy_copyrs ((3U #CSZZe,\\^Fcii61cii6G6G96TV,555V,444 \\u\ %Fcii61cii6G6G96TV,555V,444\\t\ $F  C (E 11 1r+cd}tjddg|}|r t|}|j}t j t d|t d|gt}tj|||jd}tj|||jd}t j dd gd}tj||y) Nr|rrr}r?rOrz 2000-01-01T05z 2001-01-01T05) rB DatetimeIndexrrrrr rOrr)rrtrsrWrMrs r(test_to_numpy_dtypers B  FF+ 3CSk\\^Fxx 6b !9V#;rNr)?@z2000-01-01T00:00:00.000000000rc||}|j||}tj|}tj||yNr@na_value)rrrrr)rIrr@rrrWrMs r("test_to_numpy_na_value_numpy_dtypers> & !C \\\ 9Fxx!H1r+z+data, multiindex, dtype, na_value, expected)r>rN))rrP)rr)r>r)r>rrrg@))rPr)rPr>)rPr)rr)r>rrr20212022ctjj|}t||}|j ||}t j |}tj||y)N)indexr) rBr from_tuplesrrrrrr)data multiindexr@rrrseriesrMs r((test_to_numpy_multiindex_series_na_valuersUH MM % %j 1E D &F __58_ rrr?)rB DataFramerrrrr)rr@rdfrMrs r( test_to_numpy_dataframe_na_valuersW d B [[ux[ 8Fxx!Q!Q!X7uEH1r+zdata, expectedctj|}|jttj }t j||yr)rBrrrrnanrr)rrrrMs r($test_to_numpy_dataframe_single_blockrs8 d B [[urvv[ 6F1r+c8tjtjddtjg}tjtjddtjg}|j dt j||y)Nrrr)r)rBrrrrrrassert_frame_equal)rMrs r(.test_to_numpy_dataframe_single_block_no_mutater+sd \\"((Cbff#56 7F||BHHc3%789H OOSO!&(+r+c`eZdZejj dddgdZdZdZy) TestAsArrayrsNrcBttdd|}tjd5t j |t }dddt jtd|td|g}tj|y#1swYMxYw)Nrrrr?rr}r) rr rrrrrOrr r)rHrsrwrMrs r(test_asarray_object_dt64z$TestAsArray.test_asarray_object_dt644sZb9:  ' ' - 3ZZ62F 388 | +Y|-K L  ##FH5 3 3s BBcttdd}tjddgd}tj|}t j ||y)Nrr)rrrrr?rr rrrrr)rHrwrrMs r(test_asarray_tz_naivez!TestAsArray.test_asarray_tz_naiveAsEZ2388\<8IC ##FH5r+cd}ttdd|}tjddgd}tj|d }t j ||tj|d}t j ||y) Nrrrrz 2000-01-01T06z 2000-01-02T06rr?r r)rHrsrwrrMs r(test_asarray_tz_awarez!TestAsArray.test_asarray_tz_awareIsr Zb9:88_o>hOC'78 ##FH5Cx0 ##FH5r+) rrrrDrrr rrrr+r(r r 3s4 [[TD,#78 69 66 6r+r )JnumpyrrD pandas.compatrpandas.compat.numpyrpandas.core.dtypes.dtypesrpandasrBrrrr r pandas._testing_testingrpandas.core.arraysr r r rrrpandas.core.arrays.string_rpandas.core.arrays.string_arrowrrrrrrndarray Categoricalr PeriodIndexcorer PeriodDtype IntervalIndex from_breaksTimedeltaIndexrrr_from_sequencerrrrOarrays period_arrayrInterval tz_localize tz_convertviewrrrNaT datetime64rrrrrrrrr rr+r(r-s  %.5AJaaH 1a& )2::w? 3* rzz84 c #R^^Z@ B  ff-, ?  ( BNND$F )M ( (662B()S T BHHff%X 6   )M ( (/1FGxX [  Z % BHHl|<l|<   *N ) )!]+48==hG  BHHa'x 8    BNN:lA,O P BHHl|<l|<   [7:v Cw:v CFtUm4 HBHHYg .PV0W X252.tUm4252*' y!_5 v  & 1266 :  f  R]]: ; ?*eRVV_vtn,MN2O2("((<( ) BHHsecURVVH-U ; ) , BHHq!fq!fq!f-U ;  2 2,!6!6r+