U *if)@s0ddlmZddlZddlZddlZddlmZmZm Z m Z m Z ddl m ZddlmZejdddgdd Zd d Zd d ZddZddZddZddZejdee gejddddgejddddiidddgiigdd Zejd!dd"d#gid"d#gdd$gd%gd&d'Zd(d)Zejdddgd*d+Zejddddgejdddiddgigd,d-Zd.d/Zejd0d1d2dd3d4gffd5gd6d7Z d8d9Z!d:d;Z"ejdd?Z%ejd@ee#e dA&d$e$fdBe$fdCe$fdDe$fdEe$fd=e$fgdFdGZ'dHdIZ(dJdKZ)ejddLdMdNdMdOdMgdPdQZ*dRdSZ+ejdTdUd#gdVfej,dd%dWfgdXdYZ-ejdd#d"gd#dUggdZd[Z.ejdd"d#gdUd#ggd\d]Z/ejd^d_dMd`dMdadMdbdMgej0dcdddeZ1dS)f)chainN) Categorical DataFrameSeries date_rangenotna)SpecificationError result_typefooc Cs:|}d}tjt|d|jddd|dW5QRXdS)NzUinvalid value for result_type, must be one of {None, 'reduce', 'broadcast', 'expand'}matchcSs dddgS)Nr xrrV/home/mars/bis/venv/lib/python3.8/site-packages/pandas/tests/apply/test_invalid_arg.py$z(test_result_type_error..r axisr pytestraises ValueErrorapply)r int_frame_const_coldfmsgrrrtest_result_type_errors r c Cs\tdddgdddgddd ggd d d gd }d }tjt|d|dddW5QRXdS)Nr rr ac)indexz)No axis named 2 for object type DataFramer cSs|SNrrrrrr+rz/test_apply_invalid_axis_value..rrrrrrrrrrtest_apply_invalid_axis_value's*r-c Cs0tjtdd|jddddW5QRXdS)Nzna_action must be .*Got 'abc'r cSs tt|Sr*)lenstrrrrrr1rz1test_applymap_invalid_na_action..abcZ na_action)rrrZapplymap) float_framerrrtest_applymap_invalid_na_action.sr3c Cs@tddgddgd}d}tjt|d|W5QRXdS)Nrr rABz Must provider )rrr TypeErroraggr,rrrtest_agg_raises4sr9c CsBtdddg}d}tjt|d|jdddd W5QRXdS) Nr rrz)na_action must either be 'ignore' or Noner cSs|Sr*rrrrrrBrz8test_map_with_invalid_na_action_raises..Z____r1)rrrrmap)srrrr&test_map_with_invalid_na_action_raises=sr<c CsZttdtddd}t|dtdd}tjttjd|jd d d d W5QRXdS) NZABBABCDZDCBAT) categoriesZorderedXXZabcdefg)namer)r cSs|Sr*rrrrrrIrz0test_map_categorical_na_action..ignorer1) rlistrrrNotImplementedErrortmEMPTY_STRING_PATTERNr:valuesr;rrrtest_map_categorical_na_actionEsrGc CsRtddddd}t|dd}tjttjd|jd d d d W5QRXdS) Nz 2011-01-01z 2011-01-02H)freqz Asia/Tokyor>)r?r cSs|Sr*rrrrrrPrz/test_map_datetimetz_na_action..r@r1) rZ tz_localizerrrrBrCrDr:rErrrtest_map_datetimetz_na_actionLs rJboxmethodrr8 transformfuncr5r6sumc Cs>|ddgi}d}tjt|dt|||W5QRXdS)Nr5r nested renamer is not supportedr )rrrgetattr)rKrLrNobjr rrrtest_nested_renamerSsrSrenamerminmaxmean)r barc Cs>ttdddd}d}tjt|d||W5QRXdS)Nr#Zint64series)Zdtyper?rPr )rrangerrrr8)rTr;rrrrtest_series_nested_renamer^sr[c Cs\ttjdddddgtdddd}d }tjt|d |j d d d giW5QRXdS)N rr5r6Cz1/1/2000)Zperiods)columnsr)rPr r rOrW) rnprandomrandnrrrrr5r8)Ztsdfrrrrtest_apply_dict_depris  rbc CsPttddd}d}tjt|d"t||ddiddidW5QRXdS) Nr"r4rPr r rUrXrV)rrZrrrrQ)rLrrrrrtest_dict_nested_renaming_deprusrcc CsDtddgi}td}tjt|dt|||W5QRXdS)Nr5r zColumn(s) ['B'] do not existr )rreescaperrKeyErrorrQ)rLrNrRr rrrtest_missing_columns rgc CsBtddgi}d}tjt|d|tttdW5QRXdS)Nr'1z#Column\(s\) \[1, 'b'\] do not existr )r'r b)rrrrfrMintr/r,rrr'test_transform_mixed_column_name_dtypessrkz how, args)Z pct_changer nsmallestr'ri)tailr c CsPtddgddgd}d|d}tjt|d|j|d|d W5QRXdS) Nr rrr!)r'riz Operation z does not support axis=1r )rargsr+)howrnrrrrrtest_apply_str_axis_1_raisess rpc Cs6d}tjt|dtdgjdddW5QRXdS)Nz&No axis named 1 for object type Seriesr r rOr)rrrrrM)rrrrtest_transform_axis_1_raisessrrcCstdddddddddddg dddddddddddg dddddddddddg tjdtjdtjdd}tj|jd <d d }d d }d}tjt|d|j |ddW5QRXdS)Nr rXonetwoZdullZshiny )r5r6r]DEF)r!r]cSs&|ddr"|ddkr"d|d<|SNr]Zshinr5r r$rv) startswithrowrrrrMsz.test_apply_modify_traceback..transformcSs2t|dr.|ddr.|ddkr.d|d<|Sry)rrzr{rrr transform2s&z/test_apply_modify_traceback..transform2z,'float' object has no attribute 'startswith'r r rq) rr_r`rananlocrrAttributeErrorr)datarMr}rrrrtest_apply_modify_tracebacks`   / rzdf, func, expectedZcumprodc Cs0d}tj||d|j||dW5QRXdS)Nz0can't multiply sequence by non-int of type 'str'r rqrrr8)rrNexpectedrrrrr"test_agg_cython_table_raises_framesrzseries, func, expectedza b cprodZstdvarZmedianc Cs,d}tj||d||W5QRXdS)Nz?[Cc]ould not convert|can't multiply sequence by non-int of typer r)rYrNrrrrr#test_agg_cython_table_raises_seriessrc CsDtddgi}td}tjt|d|dtiW5QRXdS)Nr'zint() argument must be a stringr )rrdrerrr7r8rjr,rrrtest_agg_none_to_type s rc Cs>tddgi}d}tjt|d|dtiW5QRXdS)Nr'zTransform function failedr )rrrr7rMrjr,rrrtest_transform_none_to_typesrcCstddgddS)Nr r)r_arrayZreshaperrrrrrrcCsddgSNr rrrrrrrrcCs tddgSr)rrrrrrrc Cs6|}d}tjt|d|j|dddW5QRXdS)Nz2too many dims to broadcast|cannot broadcast resultr r broadcastrr)rrNrrrrrtest_apply_broadcast_errors rc CsLd}tjt|d0tjdd|jddg|dW5QRXW5QRXdS)Nz3cannot combine transform and aggregation operationsr r@allrVsqrtrqrrrr_Zerrstater8)rr2rrrrtest_transform_and_agg_err_agg+srz func, msgrz(cannot combine transform and aggregationz2cannot perform both aggregation and transformationc Cs@tjt|d(tjdd||W5QRXW5QRXdS)Nr r@rr) string_seriesrNrrrr!test_transform_and_agg_err_series3s rc Cs0d}tjt|d|j||dW5QRXdS)NFunction did not transformr rq)rrrrM)rr2rNrrrrtest_transform_wont_agg_frameDsrc Cs^d}|ddkrtnd}d}tjt|d*tj||d||W5QRXW5QRXdS)Nrrrz!invalid value encountered in sqrtr )RuntimeWarningrrrrCZassert_produces_warningrM)rrNrwarnZwarn_msgrrrtest_transform_wont_agg_seriesMs r op_wrappercCs|Sr*rrrrrr[rcCs|gSr*rrrrrr[rcCsd|iSNr5rrrrrr[rcCs d|giSrrrrrrr[rz(ignore:.*Select only valid:FutureWarningc CsR||}tddddgi}t||}d}tjt|d||W5QRXdS)Nr5r rrrr )rrCget_objrrrrM)Zall_reductionsZframe_or_seriesroprRrrrrtest_transform_reducer_raisesZs  r)2 itertoolsrrdnumpyr_rZpandasrrrrrZpandas._testingZ_testingrCZpandas.core.basermarkZ parametrizer r-r3r9r<rGrJrSr[rbrcrgrkrprrrZget_cython_table_paramsr7rsplitrrrrrrrrrfilterwarningsrrrrr s     "     A