gL iddlmZddlmZddlZddlZddlZddlm Z m Z m Z m Z ddl mZGddZGddZy) )deepcopy) methodcallerN) DataFrame MultiIndexSeries date_rangeceZdZejj dddgdZejj dddgdZdZdZ dZ d Z d Z d Z y ) TestDataFramefunc_set_axis_name rename_axisc>tddgddgg}t|d|}|jjJ|jjdk(sJt|dd|}|jjJ|jjdk(sJy)Nfoocolsaxis)rrindexnamecolumns)selfr dfresults e/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/pandas/tests/generic/test_frame.pytest_set_axis_namez TestDataFrame.test_set_axis_names AA' (*dE*2.xx}}$$$||  E)))3dF3B7zz&&&~~""f,,,c Bttjdtjt dDcgc]}d|fc}tjt dDcgc]}d|fc}}ddg}t |||}|jj|k(sJ|jjddgk(sJt ||d |}|jjddgk(sJ|jjddgk(sJycc}wcc}w) N)rraBcAxyzC)rrL1L2rr) rnpemptyr from_tupleslistrrnamesr)rr xr level_namesrs rtest_set_axis_name_miz#TestDataFrame.test_set_axis_name_mis  HHV ((DK)Hq3()HI**d5k+JS!H+JK  Tl 0dK04||!![000~~##d|3338dKa8<~~##d|333||!!dD\111*I+Js D# Dcd}tdgg}tdgg}tjt|5|j sJ dddtjt|5|j rJ dddtddgg}d}t j t|5t |dddtjt|5t j t|5|j ddddddy#1swYxYw#1swYxYw#1swYzxYw#1swY6xYw#1swYyxYw)NzPDataFrame.bool is now deprecated and will be removed in future version of pandasTFmatchz+The truth value of a DataFrame is ambiguous)rtmassert_produces_warning FutureWarningboolpytestraises ValueError)rmsg_warnrdf1msg_errs rtest_nonzero_single_elementz)TestDataFrame.test_nonzero_single_element0s. * x % "  ' ' X F 779 9  ' ' X F "xxz !>z "' (? ]]:W 5  H  ' ' X F z9        " "       sGD#/D/6 D;%EEE#D,/D8;EE EEcFtgdgdtjjdj dtjjdj dd}|j dj }tj||y)N)rbarrr>rr>rr)oner?twothreer@r@r?rAr)r"Br$Dr") rr'random default_rngstandard_normalgroupbysumr2assert_metadata_equivalentrrrs r'test_metadata_propagation_indiv_groupbyz5TestDataFrame.test_metadata_propagation_indiv_groupbyGsy MQYY**1-==a@YY**1-==a@   C$$& %%b&1rcttjjdj dt ddd}|j d}tj||y) Nr)r20130101rNs)periodsfreq)r1min) rr'rErFrGrresampler2rJrKs r(test_metadata_propagation_indiv_resamplez6TestDataFrame.test_metadata_propagation_indiv_resampleTsU  II ! !! $ 4 4Y ?ZC@ V$ %%b&1rcdd}|j5}|jtddg|jtd|ttjj dj dddd d g }ttjj dj dddd d g }dgt_d|_d|_|j|d gd gd}|jdk(sJttjj dj dddtd }d|_tj||g}|jdk(sJ dddy#1swYyxYw)Nc |jD]}|dk(rN|j|j}}t||ddzt||dz}tj |||V|dk(rYdj |jDcgc]}t||dst||c}}tj |||tj ||t||d|Scc}w)Nmerge|concat+) _metadataleftrightgetattrobject __setattr__joinobjs) rothermethodkwargsrr^r_valueos rfinalizez?TestDataFrame.test_metadata_propagation_indiv..finalizeas MW$"'**ekk%D#D$3c9GE4QS>Fctdgdi}d|_gd|d<d|_|jdk(sJtj|dt gddy)Nr,rrrr)rry)r)rrr2assert_series_equalr)rrs rtest_set_attributez TestDataFrame.test_set_attributesUY' (3ttqyy r#wys(CDrcdtggdg}t|}tj||y)Nr")datarr)rrr2assert_frame_equal)r empty_frameempty_frame_copys rtest_deepcopy_emptyz!TestDataFrame.test_deepcopy_emptys0 RrC5A #K0 . rs:!  M=M=b/(/(r