U т╟-eН<у@sтddlZddlZddlZejjddДГZejjddДГZejjddДГZejjdd ДГZ ejj d dНdd ДГZejjddДГZejjddДГZ ejjddДГZejjddДГZejjddДГZejjddДГZejjddДГZdS)щNcCsrddlm}|аб}tаdtабб}tаdtабб}tа||gб}tаdtабб}|а|б}|а|б}|аб|j ksxt В|аd||gtабб}|а |||tабб} |а| |б} | абj tабks─t В|а|| gtабб}|jаdбdksьt Вtjdd d dgtабdН}tjdd d dgtабdН} tjdd d dgtабdН}tjj|| gddgdН}|а|б\}|а|бРsnt ВdS)Nr┌a┌b┌res┌greater_than·@expr_щ щ щщь щй┌typeщщй┌names)┌pyarrow.gandiva┌gandiva┌TreeExprBuilder┌pa┌field┌int32┌schema┌ make_field┌return_typer ┌AssertionError┌ make_function┌bool_┌make_if┌make_expression┌result┌make_projector┌default_memory_pool┌llvm_ir┌find┌arrayZRecordBatch┌from_arrays┌evaluate┌equals)r┌builderZfield_aZfield_br┌field_result┌node_a┌node_b┌ condition┌if_node┌expr┌ projectorrr┌eZinput_batch┌rйr3·[/var/www/html/Darija-Ai-Train/env/lib/python3.8/site-packages/pyarrow/tests/test_gandiva.py┌test_tree_exp_builders6 r5cCs▐ddlm}tjаtаddgбtаddgбgddgб}|аб}|а|jа dбб}|а|jа dбб}|а d||gtабб}tа d tабб}|а||б}|а |j|gtабб}|а|абdб\} tаdd gб} | а| бs┌tВdS)NrчЁ?ч@ч@g@rr┌add┌cg@)rrr┌Tabler&r%rrrrr┌float64rr!r"r'┌ to_batchesr(r)r┌tabler)r+r,┌sumr*r/r0r2r1r3r3r4┌ test_tableAs$ r@c CsЇddlm}tjаtаddДtdГDГбgdgб}|аб}|а|j а dбб}|аdtабб}|а d||gtабб}|а|б}|абjtабksТtВ|а|j |б}|jаdбd ks┤tВ|а|абdtабб}|абаtjtd ГtабdНбsЁtВdS)NrcSsg|]}d|СqS)r6r3)┌.0┌ir3r3r4┌ `sztest_filter..i'rg@П@┌ less_thanrriшr)rrrr;r&r%┌rangerrrr┌make_literalr<rr┌make_conditionr r r┌make_filterr#r$r'r=r"┌to_arrayr(┌uint32) rr>r)r+Zthousand┌condr-┌filterr r3r3r4┌test_filter\s rMc Cs*ddlm}tаddddddgб}tjа|gdgб}|аб}|а|jа dбб}|а |ddgtабб}|а|б}|а |j|б}|а|абdtабб}|абаtjd d gtабdНбs╕tВtаdd d d ddd ddd g б}tjа|аtаббgdgб}|а|jа dбб}|а |d dgtабб}|а|б}|а |j|б}|а|абdtабб}|абаtjd dd dgtабdНбРsttВtаdd d d ddd ddd g б}tjа|gdgб}|а|jа dбб}|а |d dgtабб}|а|б}|а |j|б}|а|абdtабб}|абаtjd dd dgtабdНбРs&tВdS)Nr┌gaZan┌ndZdiZiv┌varщщrщщrщ щщ)rrrr%r;r&rrrr┌make_in_expression┌stringrGrHr'r=r"rIr(rJr┌castr┌int64) r┌arrr>r)r+rKr-rLr r3r3r4┌test_in_exprss4 $ * r]z>Gandiva C++ did not have *real* binary, time and date support.)┌reasoncCs▄ddlm}tаddddddgб}tjа|gdgб}|аб}|а|jа dбб}|а |ddgtабб}|а|б}|а |j|б}|а|абdtабб}|абаtjd d gtабdНбs╕tВtjаdб} tjаd б} tjаdб}tа| | |gб}tjа|gdgб}|аб}|а|jа dбб}|а || gtаdбб}|а|б}|а |j|б}|а|абdtабб}t|абГd gkРsvtВ| аб}| аб} |аб}tа|| |gб}tjа|gdgб}|аб}|а|jа dбб}|а || gtаdбб}|а|б}|а |j|б}|а|абdtабб}t|абГd gkРs(tВ| аб}| аб}|аб}tа|||gб}tjа|gdgб}|аб}|а|jа dбб}|а ||gtабб}|а|б}|а |j|б}|а|абdtабб}t|абГd gkРs╪tВdS)NrsgasansndsdisivsvarrQrRrg╒╠ч╣,√╓Ag╒╠чп,√╓Ag╒╠ч╪+√╓A┌ms)rrrr%r;r&rrrrrX┌binaryrGrHr'r=r"rIr(rJr┌datetime┌utcfromtimestamp┌ timestamp┌list┌timeZtime64┌dateZdate32)rr\r>r)r+rKr-rLr Z datetime_1Z datetime_2Z datetime_3Ztime_1Ztime_2Ztime_3Zdate_1Zdate_2Zdate_3r3r3r4┌test_in_expr_todoШs\ $ rgcCsLddlm}tjаtаdddddddgбtаd d ddd ddgбgddgб}|аб}|а|jа dбб}|а|jа dбб}|а dtабб}|а dtабб}|аd||gtа бб}|аd||gtа бб}|аd||gtа бб} |а|а||gб| gб} |а| б}|а|j|б}|а|абdtабб} | абаtjdddgtабdНбРsHtВdS)Nrr6g?@gG@r8gАL@gF@g6@g@gАF@gB@g@R@g└T@g7@gS@rrgI@g&@rDrrRrr)rrrr;r&r%rrrrrFr<rr┌make_or┌make_andrGrHr'r=r"rIr(rJr)rr>r)r+r,ZfiftyZelevenZcond_1Zcond_2Zcond_3rKr-rLr r3r3r4┌test_booleanтs,■¤ rjc Cs╠ddlm}|аб}|аdtабб|аdtабб|аdtабб|аdtабб|аdtа бб|аdtа бб|аdtабб|аdtабб|аd tа бб|аd tабб|аdtабб|аdtабб|аd tабб|аddб|аddб|аddб|аddб|аddб|аddб|аddб|аddб|аd dб|аd dб|аddб|аddб|аd dбtаtбП|аdtа ббW5QRXtаtбП|аddбW5QRXdS)NrTrQrRrSrTrrVщg @g"@Zhellosworld┌bool┌uint8┌uint16rJ┌uint64┌int8┌int16rr[┌float32r<rYr`)rrrrFrrrmrnrJrorprqrr[rrr<rYr`┌pytest┌raises┌ TypeError)rr)r3r3r4┌ test_literals¤s@rvc CsЁddlm}ddddg}tj|tабdН}tjj|gdgdН}|аб}|а|j а dбб}|аd tабб}|аd ||gtа бб}tа dtа бб}|а||б} |а|j | gtабб} | а|абdб\}tjdd d d gtа бdН}|а|бsьtВdS)NrZpark┌sparklezbright spark and fireZsparkrrrz%spark%┌likerFT)rrrr%rYr;r&rrrrrFrrrr!r"r'r=r(r) r┌elements┌datar>r)r+┌regexrxr*r/r0r2rr3r3r4┌ test_regex%s$ r|cCs\ddlm}|аб}t|dабГtjks.tВt|dабГt ksFtВt |ddГsXtВdS)Nr┌name)rrZ"get_registered_function_signaturesr rrZDataTyperZparam_typesrd┌hasattr)rZ signaturesr3r3r4┌'test_get_registered_function_signatures=s rcCsЬddlm}tаб}tаddddddgtабб}tаdddd dd gtабб}tаddd dddgtабб}tjа|||gdddgб}tаdtабб}|а б}|а |jаdбб}|а |jаdбб} |а |jаdбб} |аd|| gtа бб}|а|б}|аd| | gtа бб} |а| | | tабб}|а||б}|а|j|б}|а|j|g|dб}|а|абd|б}|а|абd|б\}tаdddgtабб}|а|бРsШtВdS)Nrrr r rщщrrrSrQщщщiы rrr:rrrDZUINT32)rrrr"r%rr;r&rrrrrrrGrrrHr!r'r=r(r)rZmpoolZarray0Zarray1Zarray2r>r*r)r+r,Znode_cZgreater_than_functionZfilter_conditionZproject_conditionr.r/rLr0Zselection_vectorr2┌expr3r3r4┌test_filter_projectGsT rЖcCsddlm}|аб}t|аdtаббГаdбs2tВt|аdtа ббГdksNtВt|а tаdtабббГdksptВt|а tаdtабббГd ksТtВ|а tаd tа ббб}|аd|gtа бб}t|Гdks╠tВ|а tаdtа ббб}|а||gб}t|Гd kРstВdS)Nrr7z(const double) 2 raw(rRz(const int64) 2┌xz (double) x┌yz (string) y┌z┌notzbool not((bool) z)zbool not((bool) z) && (bool) y)rrr┌strrFrr<┌ startswithrr[rrrYrrri)rr)Zfield_zZ func_nodeZfield_yZand_noder3r3r4┌test_to_stringvs ""rНc Cs┤ddlm}|аб}tаdtабб}tа|gб}|аdtабб}t а tбП|аdбW5QRXt а tбП|а |dddбW5QRXt а tбП|а|dgбW5QRXt а tбП|аd|gбW5QRXt а tбП|аddddgtаббW5QRXt а tбП|аd|бW5QRXt а tбП|аdбW5QRXt а tбП|аd|dgtаббW5QRXt а tбП|а|dgбW5QRXt а tбП|а|dбW5QRXdS)NrrЗTrQrRrSrD)rrrrrrrrFrrsrtrurrrirhrXrrGrr!rH)rr)Zfield_xrZliteral_truer3r3r4┌test_rejects_noneКs2" rО)rarsZpyarrowr┌markrr5r@rMr]┌skiprgrjrvr|rrЖrНrОr3r3r3r4┌s4 ( $ H ' .