<>i`tSSKrSSKrSSKrSSKJr \RRr SSS.r SSK J r SISjr"SS\5r\R""SS 9S 5r\R""SS 9S 5r\R""SS 9SJS j5r\R""SS 9SJS j5r\R""SS 9S5r\R""SS 9S5r\R""SS 9S5r\R""SS 9S5r\R""SS 9S5r\R""SS 9S5r\R""SS 9S5r\R""SS 9S5r\R""SS 9S5r\R""SS 9S5r\R""SS 9S5r SSS.Sjr!Sr"Sr#Sr$Sr%S r&S!r'S"r(S#r)S$r*S%r+S&r,S'r-S(r.S)r/S*r0S+r1S,r2S-r3S.r4S/r5S0r6S1r7S2r8S3r9S4r:S5r;S6rS9r?\RRS:5r@\RRS;5rAS<rBS=rC\RRS>5rD\RRS?5rE\RRS@5rF\RRSA5rGSBrHSCrI\RRSD5rJ\RRSE5rK\RRSF5rM\RRSG5rN\RRSH5rOg!\a SrGN8f=f!\a Sr GNf=f)KN)computesummary descriptioncFSSKJn U"[R"5U5$)Nr)_get_udf_context)pyarrow._computer padefault_memory_pool) batch_lengthr s U/var/www/html/land-tabula/venv/lib/python3.13/site-packages/pyarrow/tests/test_udf.pymock_udf_contextr)s1 B224l CCc\rSrSrSrg)MyError.N)__name__ __module__ __qualname____firstlineno____static_attributes__rrrrr.srrsession)scopecSnSn[n[R"UUUS[R"50[R"55 X4$)z, Register a unary aggregate function (mean) cV[R"[R"U55$N)r scalarnpnansum)ctxxargss rfunc"sum_agg_func_fixture..func7syy1&&rsum_udfr#) empty_udf_docpcregister_aggregate_functionr float64r% func_namefunc_docs rsum_agg_func_fixturer/2sN 'IH""4#,#+'*BJJL$%$&::< % ?rcSnSn[n[R"UUUS[R"50[R"55 X4$)Nc[S5e)NOops) RuntimeErrorr rlenr"r#s rr%(exception_agg_func_fixture..funcJs 6""ry=exception_len(x)r#r(r)r*r int64r,s rexception_agg_func_fixturer:HsN!%IH""4#,#+'*BHHJ$%$&88: % ?rcSnSn[n[R"UUUS[R"50[R"55 X4$)Nch[R"[U5[R"55$r)r rr4int32r5s rr%1wrong_output_dtype_agg_func_fixture..func^syyQ,,ry=wrong_output_dtype(x)r#r8rr%r-r.s r#wrong_output_dtype_agg_func_fixturerA\sN-*IH""4#,#+'*BHHJ$%$&88: % ?rcSnSn[n[R"UUUS[R"50[R"55 X4$)Nc[U5$r)r4r5s rr%0wrong_output_type_agg_func_fixture..funcqs 1v ry=wrong_output_type(x)r#r8r@s r"wrong_output_type_agg_func_fixturerFosN)IH""4#,#+'*BHHJ$%$&88: % ?rcSnSnSSS.n[R"UUU[R"5[R"5S.[R"55 X4$)z$ Register a binary scalar function. cD[R"SX/URS9$)Nmultiply memory_poolr) call_functionrK)r"mr#s rbinary_function,binary_func_fixture..binary_functions" QF,/OO= =rzy=mxzfind y from y = mxr)rNr#r)register_scalar_functionr r9)rOr- binary_docs rbinary_func_fixturerTs^ =I#!57J ) *&(hhj&(hhj!#!# ,  %%rc SnSSS.nSn[R"UUU[R"5[R"5[R"5S.[R"55 X4$)z% Register a ternary scalar function. c[R"SX/URS9n[R"SXC/URS9$)NrIrJaddrL)r"rNr#cmxs rternary_function.ternary_func_fixture..ternary_functions?   j1&*-//;w,/OO= =rzy=mx+czfind y from y = mx + cr)array1array2array3rQ)rZ ternary_docr-s rternary_func_fixturer`sh = '":.varargs_functions6C""53*/2@C rz z=ax+by+czfind z from z = ax + by + cr)r\r]rQ)rgr- varargs_docs rvarargs_func_fixturerjsa  I)"?K 0 ) +.0hhj.0hhj!"!# ,  &&rcvSnSSS.nSn[R"UUU0[R"55 X4$)z% Register a nullary scalar function. c[R"S/UR-[R"5URS9$N*typerKr arrayr r9rKcontexts r nullary_func*nullary_func_fixture..nullary_func3xxw333"((*$+$7$79 9rrandom functiongenerates a random valuertest_nullary_funcrQrur.r-s rnullary_func_fixturer|sJ 9 %1H$I ) ( " " ,  ""rctSnSSS.nSn[R"UUU0[R"55 U$)z Register a nullary scalar function with an ephemeral Python function. This stresses that the Python function object is properly kept alive by the registered function. c[R"S/UR-[R"5URS9$rmrqrss rru4ephemeral_nullary_func_fixture..nullary_funcrwrrxryrtest_ephemeral_nullary_funcrQr{s rephemeral_nullary_func_fixturersG9 %1H.I ) ( " " , rczSnSn0n[R"5nSSS.n[R"XUX#5 X4$)z] Register a scalar function which returns something that is neither a Arrow scalar or array. cg)Nrnrr"s rwrong_output_type9wrong_output_type_func_fixture..wrong_output_typesrtest_wrong_output_typezreturn wrong output typerrr r9r)rR)rr-in_typesout_typedocs rwrong_output_type_func_fixturersL )IHxxzH- C 1c (4  ''rcSnSnS[R"50n[R"5nSSS.n[R"XUX#5 X4$)ze Register a scalar function whose actual output DataType doesn't match the declared output DataType. c4[R"SUS/5$NrWr)rM)r"rrs rwrong_output_datatypeAwrong_output_datatype_func_fixture..wrong_output_datatypesqz22rtest_wrong_output_datatyperrzreturn wrong output datatyperr)r r9int16r)rR)rr-rrrs r"wrong_output_datatype_func_fixturersU 3,I$HxxzH1 C 5# (4 ++rczSnSn0n[R"5nSSS.n[R"XUX#5 X4$)z6 Register a scalar function with the wrong signature. cR[R"S[R"5S9$)Nrrp)r rr9rrrwrong_signature5wrong_signature_func_fixture..wrong_signature*syy,,rtest_wrong_signaturezUDF with wrong signaturerrr)rr-rrrs rwrong_signature_func_fixturer$sK -'IHxxzH- CC (4  %%rctSnSnSSS.n[R"XU0[R"55 X4$)z= Register a scalar function which raises a custom exception. c[S5e)Nerror raised by scalar UDF)rrs r raising_func*raising_func_fixture..raising_func>s233r test_raisezraising functionrrrQ)rr-rs rraising_func_fixturer9sA 4I% C  "BHHJ0  ""rcSnSn[n[R"XUS[R"50[R"55 X4$)z Register a vector function c[R"UR5R5R SS95$)NT)pct)r rr to_pandascopyrankr5s rpct_rank+unary_vector_func_fixture..pct_rankOs.xx **,11d1;<.pivot`sc ^^ ' 'q  ' I S S U88C38?KKM~~))(3CCEEr y=pivot(x)rrv1v2)r(r)rr r9r+utf8struct)rr-rs rstruct_vector_func_fixturer[s F I C #hhjrzz|"'')< C$tRZZ\&:T2::<.test_reg_functionsxx~rzfunc must be a callablematchtest_none_function)DataType expected, got test_output_functionz)in_types must be a dictionary of DataTypetest_input_functionrz?Already have a function registered with name: test_reg_function)r r9pytestraises TypeErrorr)rRKeyError)rrrr expected_exprs rtest_registration_errorsrs|$1 C"((*%HxxzH y ! ##$5$(#$, . " y(A B ##D*>$, . C @M y 6 ##$5$:C$( * 7 @M y 6 ##$5$93$, . 7 1 3S" (*M x} 5 ##$5$7b$, . 6 5A " ! C B 7 6 7 6 6 5s<F FF#F4G F F # F14 G GcUupSn[R"[US9 [R"US/5 SSS5 g!,(df  g=f)Nz9VarArgs function 'z=ax\+by\+c' needs at least 2 argumentsrrn)rr ValueErrorr)rM)rj_r- error_msgs r test_varargs_function_validationr s:'LALI z 3 RD) 4 3 3s A AcS[R"50n[R"5nSS0nSn[R"[SS9 [ R "USX U5 SSS5 SS 0n[R"[S S9 [ R "US X U5 SSS5 g!,(df  NN=f!,(df  g=f) Nrrdescc4[R"SUS/5$rr)r"rs r add_const/test_function_doc_validation..add_consts{33rz#Function doc must contain a summaryrtest_no_summaryrz test summaryz'Function doc must contain a description test_no_desc)r r9rrr r)rR)rrr.rs rtest_function_doc_validationrs"((*%HxxzH vH4 zB D ##I/@$,$, . D >H zF H ##I~$,$, . H H D D H Hs B/ C/ B= Cc[U/SSS9 g)NFrr)r)r|s rtest_nullary_functionrs.5'(*rcbUn[R"U/SS9nUR5S/:Xdeg)Nrrrn)r)rM to_pylist)rrrs rtest_ephemeral_functionrs4 )D   dBq 1F    " %% %rcUup[R"[SS9 [R"U/SS9 SSS5 g!,(df  g=f)NzUnexpected output type: intrrrrrrr)rM)rr r-s rrrs<1LA y: < Bq1 < < < > A cUupSn[R"[US9 [R"U[ R "SS/5/5 SSS5 g!,(df  g=f)NzDExpected output datatype int16, but function returned datatype int64rrr)rrrr)rMr rr)rr r-rs rrr&sL5LA/LA'M y 6 Bq1 7 6 6s A AcSnSnS[R"50n0nSSS.n[R"[SS9 [ R "XXBU5 SSS5 g!,(df  g=f) NcU$rrr"rfs ridentity1test_wrong_datatype_declaration..identity; rtest_wrong_datatype_declarationrrztest output valuez test outputrz%DataType expected, got rr r9rrrr)rRr#r-rrrs rr&r&:sl2I$HH&$ C yD F ##H$'8 = F F F A A-cSnSnSS0n[R"5nSSS.n[R"[SS9 [ R "XUX#5 SSS5 g!,(df  g=f) NcU$rrr"s rr#3test_wrong_input_type_declaration..identityLr%r!test_wrong_input_type_declarationrrztest invalid input typezinvalid input functionrrrr'r(s rr-r-Ksk4IHxxzH,/ C yH J ##H$, 8 J J Jr)c[R"[R"55nUup#[R"U[R "S/S-[R "5S9/US9nU[R "S/S-[R "5S9:XdeUR5S:XdeSnUR5S:Xdeg)NrrrJr@r)r proxy_memory_poolr r)rMrrr9bytes_allocated)r proxy_poolr r-ress rtest_scalar_udf_contextr5\s%%b&<&<&>?J%LA  9HHaS4ZbhhjAB'1 3C "((A3:BHHJ7 77 7  % % '8 33 3 C  % % '1 ,, ,rcUup[R"[SS9 [R"U/SS9 SSS5 g!,(df  g=f)Nrrrr)rrrr)rM)rr r-s rtest_raising_funcr7ks7'LA w&B C Bq1 D C CrcUup[R"U[R"S5/5nU[R"S5:Xdeg)Nr )r)rMr r)rrr-r4s rtest_scalar_inputr:qs<,H  9ryy}o 6C "))B-  rcUup[R"[R"55nUR5S:Xde[R"S/S-[R "5US9nUR5S:Xde[ R"X$/5 UR5S:XdeSnUR5S:Xdeg)Nrrr/ror0)r r1r r2rrr9r)rM)rrr-r3rs rtest_input_lifetimer<ws,H%%b&<&<&>?J  % % '1 ,, , !t"((**EA  % % '8 33 3Y$  % % '8 33 3 A  % % '1 ,, ,rc [U5VVs/sH/up#[R"[U5XRS9PM1 nnn[R R X@S9$s snnfNr)arraysschema) enumerater rrlistrprr)r@itersirr?s r_record_batch_from_itersrEs[#E*,*$!hhtAwVY^^4* , >> % %V % CC,s6A%c T[U[XS-5[US-US-55$)Nrrr9)rErange)r@ns r_record_batch_for_rangerIs/ #F$)!V$4$)!a%R$8 ::rc^^UU4SjnU$)Nc.>"UU4SjS5nU"5$)Nc*>\rSrSrSrUU4SjrSrg),make_udt_func..udf_func..UDTicSUlgrcallerselfs r__init__5make_udt_func..udf_func..UDT.__init__s " rc z>URcT"U5RSsUlnURU5nUR5$![ae TVs/sH"n[R"/UR S9PM$ Os snfnn[R RUTS9nUR5$f=fr>) rPsend StopIterationr rrrprrr)rRr"batchrr? batch_genr@s r__call__5make_udt_func..udf_func..UDT.__call__s6{{*+4S>+>+>( S KK,E ,,.. %6+13+1%!hhr ;+13F3NN66%f76E,,.. 6s8A B:)B&B:9B:rONrrrrrSrZrrYr@srUDTrMs # / /rr^r)r"r^rYr@s rudf_funcmake_udt_func..udf_funcs / /u rr)r@rYr_s`` r make_udt_funcras" Orc@^^[5m"U4SjS5mU4Sj$)zA short datasetc(>\rSrSrSrU4SjrSrg)%datasource1_direct..GeneratoricSUlg)NrrHrQs rrS.datasource1_direct..Generator.__init__s DFrc>URS:Xa[T//5nUR5$U=RS-sl[TUR5nUR5$)Nrr)rHrErIr)rRr"rXr@s rrZ.datasource1_direct..Generator.__call__s[vv{0R@((* *! /?((* *rrfNr\)r@sr Generatorrds  + +rrjc>T"5$rr)r"rjs r$datasource1_direct..sy{r)datasource1_schema)rjr@s@@rdatasource1_directros  !F + + #"rc<^[5mU4Sjn[TU5$)Nc3X># [SSS5Hn[TUS- 5v M g7f)Nrrr)rGrIr"rHr@s rrY(datasource1_generator..batch_gens)q!RA)&!a%8 8!s'*rnrar]s @rdatasource1_generatorrvs  !F9  ++rc<^[5mU4Sjn[TU5$)Nc3l># [SSS5Hn[TUS- 5v M [S5e7f)Nrrrrrdatasource1_exception)rGrIr3rss rrY(datasource1_exception..batch_gens7q!RA)&!a%8 8!233s14rur]s @rryrys  !F4  ++rc[R"S[R"54S[R"54/5$)Nr)r r@r=rrrrnrns, 99r288:&RXXZ(89 ::rcUS3SS.n0n[R"S[R"54S[R"54/5nXX#U4$)Nz UDTztest {func_name} UDTrr)r rr=)r%r-r.rrs rdatasource1_argsr}sQ'[-57HHyy2rxxz*R,<=>H H 88rc[5nU"5nURn[X#5n[R"U6 Sn[R "U5HnUS-nU[ X5:XaMe g)Nrr)rnrr}r)register_tabular_functioncall_tabular_functionrI) func_makerr@r%r-rrHitems r_test_datasource1_udtrsk  !F @gD@gI@mean_udfr)r rrr+r)rMr)unary_agg_func_fixturearrrexpecteds rtest_scalar_agg_basicrsF ((12::< @C   j3% 0FyyH   rc[R"/[R"55n[R"[R SS9 [ R"SU/5 SSS5 g!,(df  g=f)Nz empty inputsrr)r rrr+rr ArrowInvalidr)rM)remptys rtest_scalar_agg_emptyrsH HHR &E rn = eW- > = =s A00 A>c[R"/SQ[R"55n[R"[R SS9 [ R"SU/5 SSS5 g!,(df  g=f)Nrrr(2output datatyperr?r rrr9rrArrowTypeErrorr)rM)rArs r"test_scalar_agg_wrong_output_dtypersN ((' 4C r((0A B 2SE: C B B A22 Bc[R"/SQ[R"55n[R"[R SS9 [ R"SU/5 SSS5 g!,(df  g=f)Nr output typerrEr)rFrs r!test_scalar_agg_wrong_output_typer sL ((' 4C r(( > 1C59 ? > >rc[R"/SQ[R"55n[R"/SQ[R"55n[R "SX/5n[R "S5nX4:Xdeg)Nr)?@@@g@sum_meang@@)r rrr9r+r)rMr)varargs_agg_func_fixturearr1arr2rrs rtest_scalar_agg_varargsrsb 88("((* 5D 88-rzz| >$ 8#3#3D#9 99 9rc[R"/[R"55n[R"/[R"55n[R"X!/SS/S9nUR S5R S/5n[R"[R"/[R"55[R"/[R"55/SS/S9nXE:Xdeg)Nrrrrr)r rrr+r=rrr)rrrrrrs rtest_hash_agg_emptyr6s 88B %D 88B #D HHd\$ 9E ^^D ! + +-B,C DFxx"bhhj1288 BJJL4#')9": t&&(K'LM ? > >rc[R"/SQ[R"55n[R"/SQ[R"55n[R"X!/SS/S9n[ R "[SS9 URS5RS/5 SSS5 g!,(df  g=f) Nrrrrrr2r)rr7) r rrr9r=rrrr3rr)r:rrrs rtest_hash_agg_exceptionrUsz 88("((* 5D 88ORXXZ 0D HHd\$ 9E |6 2 t&&(G'HI 3 2 2s "B55 Cc:SnSn[R"[R"SU5[R"55n[R"[R R X!5[R"55n[R"XC/SS/S9nURS5RS/5nURS5RS/5RSS /5nURS5URS5:Xdeg ) z2Test hash aggregate udf with randomly sampled datai@Br/rrrr)rr')rsum value_sum_udfN) r rrr repeatr+randomchoicer=rrrrr)r/ value_num group_numrrrrrs rtest_hash_agg_randomr_sII 88BIIa+RZZ\ :D 88BII$$Y:BHHJ GD HHd\$ 9E ^^D ! + +-A,B CF~~d#-- *ND/+BC  >>$ 8#3#3D#9 99 9rc[R"/SQ[R"55n[R"SU/5nUS"SU5nX#:Xdeg)Nrrrr rrr+r)rMrrrrs rtest_vector_basicrrsJ ((12::< @C   ou 5F(+D#6H   rc[R"S/[R"55n[R"SU/5nUS"SU5nX#:Xdeg)Nrrrrrs rtest_vector_emptyrzsK ((A3 %C   ou 5F(+D#6H   rc>[R"/SQ[R"55n[R"/SQ[R"55n[R"/SQ5n[R "SXU/5nUS"SXU5nXE:Xdeg)N)rrrr)rrrr)rrrrrr)r rrr9r+r)rM)rrrrXrrs rtest_vector_structrs bhhj A bjjl A   A  lQ1I 6F)!,T1;H   r)r)r)Prnumpyr ImportErrorpyarrowr rr)markr pytestmarkr(pyarrow.datasetrrr3rfixturer/r:rArFrTr`rjr|rrrrrrrrrrrrrr rrrrrrr&r-r5r7r:r<rErIrarorvryrnr}rrrrrrrrrrrrrrrrpandasrrrrrrrs9&![[ r2  D  l i !*i !&i !$i !$i &!&(i '!'0i '!'2i #!#,i !0i (!((i ,!,(i &!&(i  #! # i  !  i !,*.'+C@ /.d*.>*& 2:2="8" -2  -D : *#$,,;9 :.15 ..; : 66::$  ONJJ::$  Y B Bs"L7M7MMMM