fume-manage-python.git

U  
ý°d«)ã    @sªddlZddlZddlmZmZmZmZmZm    Z    m
Z
ddlmZ ddlmZej ddddg¡dd    Zej d
ddg¡d dZej d
ddg¡ej ddejdge    dee    dgg¡ddZej ddejdge    dee    dgg¡ddZej ddejdge    dee    dgg¡ddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zej d(dd)g¡d*d+Zd,d-Z d.d/Z!d0d1Z"d2d3Z#d4d5Z$d6d7Z%dS)8éN)ÚNAÚ    DataFrameÚIntervalÚNaTÚSeriesÚ    TimestampÚinterval_range)Ú    get_arrayÚmethodÚpadZnearestÚlinearcCstdddgi}| ¡}|j|d}|rFt t|dt|d¡sbtnt t|dt|d¡rbtd|jd<|rt t|dt|d¡rtt     ||¡dS)NÚaéé©r
éd©rr)
rÚcopyÚinterpolateÚnpÚ shares_memoryr    ÚAssertionErrorÚilocÚtmÚassert_frame_equal)Úusing_copy_on_writer
ÚdfÚdf_origÚresult©rú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/copy_view/test_interp_fillna.pyÚtest_interpolate_no_ops
r!ÚfuncZffillZbfillcCstdddgi}| ¡}t||}|rFt t|dt|d¡sbtnt t|dt|d¡rbtd|jd<|rt t|dt|d¡rtt     ||¡dS)Nr rrrr)
rrÚgetattrrrr    rrrr)rr"rrrrrr Útest_interp_fill_functions$s
r$Úvalsrrz
2019-12-31z
2020-12-31cCsLtd|i}t||}t t|dt|d¡r4t|rH|j d¡sHtdS)Nr r)rr#rrr    rÚ_mgrÚ_has_no_reference)rr%r"rrrrr Útest_interpolate_triggers_copy8s
r(cCsRtd|i}t|d}|jdddt |t|d¡s:t|rN|j d¡sNtdS)Nr rT©r
Úinplacer)rr    rrrrr&r')rr%rÚarrrrr Ú-test_interpolate_inplace_no_reference_no_copyFs
r,cCs¢tddtjdgi}| ¡}t|d}|dd}|jddd|rt |t|d¡rZtt     ||¡|j
 d¡svt|j
 d¡stnt |t|d¡stdS)Nr rrrTr)r)rrÚnanrr    rrrrrr&r')rr%rrr+Úviewrrr Ú"test_interpolate_inplace_with_refsTs
r/cCs¦tdtjdgdd}| ¡}|jdd}|rLt t|dt|d¡shtnt t|dt|d¡rhttd|j    d<|rt t|dt|d¡rtt
 ||¡dS)    Nr Úcr©r ÚbZasfreqrz
2021-12-31r)rrr-rrrr    rrrrr)rrrrrrr Ú$test_interpolate_cleaned_fill_methodisr3cCsZtdddgdd}t|d}|jddd|rV|j d    ¡s@tt |t|d¡sVtdS)
Nr r2r0rr1rTr)r)rr    rr&r'rrr©rrÚarr_arrr Ú%test_interpolate_object_convert_no_op|s
r6cCs`ttddgtddd}t|d}|jddd|r\|j d    ¡sFtt     |t|d¡r\tdS)
Nrr©Zdtyper1r rTr)r)
rrÚobjectr    rr&r'rrrr4rrr Ú&test_interpolate_object_convert_copiess
r9cCs^tdtjdgdd}t|d}|jdddd|rD|j d    ¡sDtt |t|d¡sZtdS©
Nrg@r1r rTZinfer)r
r*Údowncastr)    rrr-r    rr&r'rrr4rrr Útest_interpolate_downcasts
r<cCstdtjdgdd}| ¡}t|d}|dd}|jdddd|r||j d    ¡sXtt     |t|d¡rntt
 ||¡nt
 ||¡dSr:)rrr-rr    rr&r'rrrr)rrrr5r.rrr Ú1test_interpolate_downcast_reference_triggers_copys
r=cCs~tdtjgdd}| ¡}| d¡}|rHt t|dt|d¡sdtnt t|dt|d¡rdtd|jd<t     
||¡dS)Nçø?rr1ç@r2r©rr©rrr-rÚfillnarr    rrrr©rrrÚdf2rrr Útest_fillnaªs
 
rEcCstdtjgdd}| ¡}| ddi¡}|rht t|dt|d¡sJtt t|dt|d¡rtnt t|dt|d¡rtd|jd<t     
||¡dS)    Nr>rr1r ç Y@r2rr@rArCrrr Útest_fillna_dict¸s
rGr;FcCstdtjgdd}t|d}t|d}|jdd|dt t|d|¡sNtt t|d|¡sdt|r|j d    ¡sxt|j d¡stdS)
Nr>rr1r r2r?T)r*r;r)    rrr-r    rBrrr&r')rr;rr5Úarr_brrr Útest_fillna_inplaceÇs
 
rIcCsötdtjgdd}| ¡}t|d}t|d}|dd}|jddd|r¨t t|d|¡rdtt t|d|¡szt|j     d    ¡st|j     d    ¡stt
 ||¡n,t t|d|¡s¾tt t|d|¡sÔttddgdd}t
 ||¡dS)
Nr>rr1r r2r?T©r*r)rrr-rr    rBrrr&r'rr)rrrr5rHr.Úexpectedrrr Útest_fillna_inplace_referenceÕs 
 
rLcCs¨ttddddd}tj|jd<| ¡}|dd}|jtddddd    |rt t    |dj
jt    |dj
j¡rrtt  ||¡n$t t    |dj
jt    |dj
j¡s¤tdS)
Nré)ÚstartÚendr )Únamer)ÚleftÚrightT)Úvaluer*)rrrr-rrrBrrr    rQÚvaluesrrÚassert_series_equal)rÚserÚser_origr.rrr Ú&test_fillna_interval_inplace_referenceêsÿÿrXcCsrtdtjdg}| ¡}| i¡}|r@t t|t|¡sXtnt t|t|¡rXtd|jd<t     
||¡dS)NrrrFr)rrr-rrBrr    rrrrU)rrVrWrrrr Útest_fillna_series_empty_argýs
 
rYcCsRtdtjdg}t|}|jiddt t||¡s:t|rN|j d¡sNtdS)NrrTrJr)    rrr-r    rBrrr&r')rrVr+rrr Ú$test_fillna_series_empty_arg_inplacesrZcCsútdtdgdd|d}| ¡}| d¡}t t|dt|d¡rFt|rxt t|dt|d¡sft|j     d¡rtnt t|dt|d¡rtt
 ||¡d|jd<|rêt t|dt|d¡rÊt|j     d¡sÚt|j     d¡sêtt
 ||¡dS)    Nrér1r7rr r2r@) rrrrBrrr    rr&r'rrr)rÚany_numeric_ea_and_arrow_dtyperrrDrrr Ú!test_fillna_ea_noop_shares_memorys
 
r]cCsÎtdtdgdd|d}| ¡}|dd}|jdddt t|dt|d¡rVt|rt t|d    t|d    ¡svt|j     d¡rt|j     d¡r´tnt t|d    t|d    ¡r´td|j
d
<t ||¡dS)Nrr[r1r7rTrJr r2r@) rrrrBrrr    rr&r'rrr)rr\rrr.rrr Ú)test_fillna_inplace_ea_noop_shares_memory.s
r^)&ÚnumpyrZpytestZpandasrrrrrrrZpandas._testingZ_testingrZpandas.tests.copy_view.utilr    ÚmarkZparametrizer!r$r-r(r,r/r3r6r9r<r=rErGrIrLrXrYrZr]r^rrrr Ú<module>sN$    
 
ÿ
ÿ
ÿ