i< TddlZddlZddlmZddlmZmZmZm Z m Z m Z m Z m Z mZddlmZddlmZejdgddZejdd d gd Zejdd d gejd d ejdge de e dggdZejd d ejdge de e dggdZejd d ejdge de e dggdZejdd d gejdddgdZdZdZdZdZ dZ!dZ"dZ#ejddd gd!Z$d"Z%d#Z&d$Z'd%Z(d&Z)d'Z*d(Z+ejdgd)d*Z,dS)+N)WARNING_CHECK_DISABLED) NA ArrowDtype DataFrameIntervalNaTSeries Timestampinterval_rangeoption_context) get_arraymethod)padnearestlinearcxtdddgi}|}d}|dkrt}d}tj||5||}dddn #1swxYwY|r4t jt|dt|dsJn3t jt|dt|drJd|j d <|r3t jt|dt|drJtj ||dS) Nar3DataFrame.interpolate with method=pad is deprecatedmatchrdrr) rcopy FutureWarningtmassert_produces_warning interpolatenp shares_memoryr ilocassert_frame_equal)using_copy_on_writerdfdf_origwarnmsgresults a/srv/django_bis/venv311/lib/python3.11/site-packages/pandas/tests/copy_view/test_interp_fillna.pytest_interpolate_no_opr,s} C!Q= ! !BggiiG D  ?C #D 4 4 4//v..///////////////P &# 6 6 "c8J8JKKKKKK#Ifc$:$:Ib#> ? ?B b#  E ?C #M = = =33 eT222333333333333333?w((+++++#E9R+=+=>>>>>??>>sA::A>A>ctdtjdgdd}t|d}d}t jt |5|ddd dddn #1swxYwY|r|j d sJtj |t|dsJdS Nrg@rMrrrrTinfer)rr;downcastr) rr!r?r rrrr r6r7r"rWs r+test_interpolate_downcastr]s !RVS)22 3 3B b#  E ?C #M = = =EE eTGDDDEEEEEEEEEEEEEEE,w((+++++ E9R#5#5 6 666 666s A22A69A6ctdtjdgdd}|}t |d}|dd}d}t jt|5|ddd dddn #1swxYwY|rX|j d sJtj |t |drJt j ||dSt j ||dSrZ) rr!r?rr rrrr r6r7r"r$)r%r&r'rTrBr)s r+1test_interpolate_downcast_reference_triggers_copyr_s] !RVS)22 3 3BggiiG b#  E aaa5D ?C #M = = =EE eTGDDDEEEEEEEEEEEEEEE(w((+++++#E9R+=+=>>>>> gt,,,,, b$'''''s+BBBctdtjgdd}|}|d}|r4tjt |dt |dsJn3tjt |dt |drJd|jd<tj ||dS)N?rrM@rNrrr rr!r?rfillnar"r r#rr$r%r&r'df2s r+ test_fillnarhs #rvQ// 0 0BggiiG ))C..CM "c 2 2Ic34G4GHHHHHH#Ib#$6$6 #s8K8KLLLLLCHTN'2&&&&&r3ctdtjgdd}|}|ddi}|rgtjt |dt |dsJtjt |dt |drJn3tjt |dt |drJd|jd<tj ||dS) NrarrMr Y@rNrrcrdrfs r+test_fillna_dictrks #rvQ// 0 0BggiiG ))S%L ! !CM "c 2 2Ic34G4GHHHHH#Ib#$6$6 #s8K8KLLLLLL#Ib#$6$6 #s8K8KLLLLLCHTN'2&&&&&r3r\Fc(tdtjgdd}t|d}t|d}d}t jt |5|dd | dddn #1swxYwYtjt|d|sJtjt|d|sJ|r8|j d sJ|j dsJdSdS) NrarrMrrNz.The 'downcast' keyword in fillna is deprecatedrrbT)r;r\r) rr!r?r rrrrer"r6r7)r%r\r&rTarr_br)s r+test_fillna_inplacernsY #rvQ// 0 0B b#  E b#  E :C #M = = =88 #th 777888888888888888 Ib#.. 6 666 6 Ib#.. 6 666 6,w((+++++w((+++++,,++sBBBc^tdtjgdd}|}t |d}t |d}|dd}t j|5|dddddn #1swxYwY|rtjt |d|rJtjt |d|sJ|j d sJ|j d sJt j ||nJtjt |d|sJtjt |d|sJtddgdd}t j ||dS) NrarrMrrNrbTrIr) rr!r?rr rr@rer"r6r7r$)r%rAr&r'rTrmrBexpecteds r+test_fillna_inplace_referencerqs #rvQ// 0 0BggiiG b#  E b#  E aaa5D 1 2 2%% #t $$$%%%%%%%%%%%%%%%;#Ib#$6$6>>>>> "c 2 2E:::::y**1-----w((+++++ dG,,,, "c 2 2E::::: "c 2 2E:::::Sz2233H"h'''''s1BBBcttdddd}tj|jd<|}|dd}t j|5|tddd dddn #1swxYwY|r^tj t|dj j t|dj j rJt j||dStj t|dj j t|dj j sJdS) Nr)startendrzinterval[float64, right])namerDr)leftrightT)valuer;)r r r!r?r#rrr@rerr"r rwvaluesassert_series_equal)r%rAserser_origrBs r+&test_fillna_interval_inplace_referencer~s QA&&&S8R   C&CHQKxxzzH qqq6D 1 2 2BB q222D AAABBBBBBBBBBBBBBB # c3   $ +YtS-A-A-F-M      tX..... c3   $ +YtS-A-A-F-M       s('BB"Bctdtjdg}|}|i}|r2tjt |t |sJn1tjt |t |rJd|jd<tj ||dS)Nrrrjr) r r!r?rrer"r r#rr{)r%r|r}r*s r+test_fillna_series_empty_argr2s !RVQ CxxzzH ZZ^^FG # &0A0ABBBBBB#IcNNIf4E4EFFFFFCHQK8V,,,,,r3ctdtjdg}t|}|idtjt||sJ|r|jdsJdSdS)NrrTrIr)r r!r?r rer"r6r7)r%r|r<s r+$test_fillna_series_empty_arg_inplacer@s !RVQ C C..CJJr4J   IcNNC 0 000 0-x))!,,,,,--,,r3ctdtdgdd|}|}|d}t jt |dt |drJ|rPt jt |dt |dsJ|jdrJnt|j j dtr4t jt |dt |dsJn3t jt |dt |drJtj||d|j d <|rkt jt |dt |drJ|jdsJ|jdsJtj||dS) NrrMrHrrrNrrc)rrrrer!r"r r6r7 isinstancedtypesr#rrr$)r%any_numeric_ea_and_arrow_dtyper&r'rgs r+!test_fillna_ea_noop_shares_memoryrJs !R!,,4R S S SBggiiG ))C..C "c 2 2Ic34G4GHHHH HM "c 2 2Ic34G4GHHHHH8--a000000 BIN1%z 2 2M "c 2 2Ic34G4GHHHHHH#Ib#$6$6 #s8K8KLLLLL'2&&&CHTN,#Ib#$6$6 #s8K8KLLLLLx))!,,,,,w((+++++'2&&&&&r3ctdtdgdd|}|}|dd}tj|5|dddddn #1swxYwYt |djts|r4tj t|dt|drJn3tj t|dt|dsJtj t|d t|d sJ|r8|j drJ|j drJtj|od |v5d|jd <dddn #1swxYwYt |djts|rtj||dStj||dS) NrrrMrHrTrIrrNpyarrowrc)rrrrr@rerrDrr!r"r r6r7r#r$)r%rArr&r'rBs r+)test_fillna_inplace_ea_noop_shares_memoryrfs !R!,,4R S S SBggiiG aaa5D 1 2 2%% #t $$$%%%%%%%%%%%%%%%"S'-,,J0CJ#Ib#$6$6 $8L8LMMMMMM "c 2 2IdC4H4HIIIII Ib#.. $0D0D E EEE E27,,Q/////9..q11111 Ny0NN   "S'-,,(0C( gt,,,,, b$'''''s$A33A7:A7 F##F'*F'cvtdtjdgdd}|}|rt j5|ddddddn #1swxYwYt j||t j5|dgdddddn #1swxYwYt j||dSt jd5tdd5|dgdddddn #1swxYwYdddn #1swxYwYt jd5tdd5||j d kdddddn #1swxYwYdddn #1swxYwYt jtstndd 5|dddddddS#1swxYwYdS) NrrrMrrTrImode.chained_assignmentrsinplace methodr) rr!r?rrraises_chained_assignment_errorrer$rr rrr)r%r&r's r+test_fillna_chained_assignmentrs !RVQa00 1 1BggiiG.  / 1 1 . . sGNN3N - - - . . . . . . . . . . . . . . . b'***  / 1 1 0 0 uI  S$  / / / 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 b'*****  ' - - 4 4 94@@ 4 4C5   d 333 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 ' - - 7 7 94@@ 7 724!8 ##C#666 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 '!7 AMMT"    . . sGNN3N - - -  . . . . . . . . . . . . . . . . . .s A33A7:A7&CCCEE7 EE E E EE!E9G 'F=1 G=G GG GGGH..H25H2)r r.r/ctdtjdgdd}|}|rt j5t |d|ddddn #1swxYwYt j||t j5t |dg|ddddn #1swxYwYt j||dSt jtstndd5t |d|ddddn #1swxYwYt jd5td d5t |dg|ddddn #1swxYwYdddn #1swxYwYt jd5td d5t ||ddk|ddddn #1swxYwYddddS#1swxYwYdS) NrrrMrTrIrrr) rr!r?rrrr1r$rrrr )r%r-r&r's r+#test_interpolate_chained_assignmentrs !RVQa00 1 1BggiiG=  / 1 1 1 1 "GBsGT " "4 0 0 0 0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 b'***  / 1 1 3 3 $GBuIt $ $T 2 2 2 2 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 b'*****  '!7 AMMT"    1 1 #GBsGT " "4 0 0 0 0  1 1 1 1 1 1 1 1 1 1 1 1 1 1 1  ' - - 7 7 94@@ 7 7(C5 4((6666 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 ' - - = = 94@@ = =.2c7Q;..t<<<< = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = =s !A66A:=A:)"CCC!E  EE)F?:"F( F?(F, ,F?/F, 0F??GGH>/+H& H>&H* *H>-H* .H>>II)-numpyr!pytest pandas.compatrpandasrrrrrr r r r pandas._testing_testingrpandas.tests.copy_view.utilr mark parametrizer,r2r?r8r=rCrJrQrUrXr]r_rhrkrnrqr~rrrrrrr3r+rs 000000                      111111#?#?#?@@''A@'.'7!344''54'&'7!344 a^ii 55sIIl...6!B!B!BCC==DC===r3