a ▒<bl3у@sжddlZddlZddlZddlZddlZddlmZddlm Z ddlmZm Z mZddlmZeejаddбedГdd НZeejаd dбedГdd НZeedабedаd бedаdбedаdбdЬГZeedабedаd бedаdбedаdбdЬГZeejjdd ddНedГdd НZeejjdd ddНedГdd НZedjабZedjабZ edjабZ!edjабZ"ej#j$ej%ddНGddДdГГZ&dS)щN)┌ set_option)┌ DataFrame┌Index┌Series)┌expressionsщ'щ┌ABCDZfloat64)┌columns┌dtypeщd┌A┌BZfloat32┌CZint64┌DZint32)r rrrщ)rrй┌size)щerznot using numexpr)┌reasonc@sдeZdZddДZddДZeeedЬddДГZe j аdee eejjd d eаeбdНeeeegбe j аdd dgбe j аdgdвбddДГГГZe j аdee eejjd d eаeбdНeeeegбe j аdd dgбddДГГZddДZe j аdgdвбe j аdeefeefgбddДГГZe j аdgdвбe j аdeefeefgбddДГГZe j аdd dgбe j аdeeeegбd d!ДГГZe j аd"gd#вбd$d%ДГZ e j аd"gd&вбd'd(ДГZ!e j аd)e"gd*вgd*вggd+вd,Нe"ddgddggd-d.gd,Нfe"gd/вgd0вgd1вggd+вd,Нe"ddgddgddggd-d.gd,Нfgбd2d3ДГZ#e j аddбe j аd4d5бd6d7ДГГZ$e j аd8gd9вбe j аd:e"e%e&gбe j аd;dd?ДГГГZ'd@S)A┌TestExpressionscCstj|_dSйN)┌expr┌ _MIN_ELEMENTSй┌self┌methodйr·m/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/pandas/tests/test_expressions.py┌setup_method2szTestExpressions.setup_methodcCs|jt_dSr)rrrrrr┌teardown_method5szTestExpressions.teardown_method)┌flex┌opnamecsZ|rЗfddД}И|_n ttИГ}tddГ|||Г}tddГtаб|||Г}||fS)Ncst|ИГ|ГSr)┌getattr)┌x┌yйr"rr┌;єz)TestExpressions.call_op..·compute.use_numexprFT)┌__name__r#┌operatorrr┌get_test_result)┌df┌otherr!r"┌op┌expected┌resultrr&r┌call_op8s zTestExpressions.call_opr-rщrr!TF┌arith)┌add┌sub┌mul┌mod┌truediv┌floordivcCs┤dt_|а||||б\}}|dkr_r(z6TestExpressions.test_run_arithmetic..r;)rrr2┌allZdtypes┌values┌tm┌assert_equal┌range┌lenr ┌ilocrr<)rr-r!r4r1r0┌irrr┌test_run_arithmeticIs0z#TestExpressions.test_run_arithmeticc Csn|j}tddГ|абd}tddГdt_tаdб|а||||б\}}tаб}|s^JdГВtа ||бdS)z▓ tests solely that the result is the same whether or not numexpr is enabled. Need to test whether the function does the correct thing elsewhere. r)FrTrz Did not use numexpr as expected.N) r*r┌copyrrZ set_test_moder2r,rArB) rr-r!Z comparison_opr4r.r1r0Zused_numexprrrr┌test_run_binaryhs zTestExpressions.test_run_binarycCsptjаdб}tjаdб}tаtjd||dб}|r4JВtаtjd||dб}|rPJВtаtjd||dб}|slJВdS)NiABr┌evaluate·+)┌np┌random┌randnr┌_can_use_numexprr+r5)r┌arrayZarray2r1rrr┌test_invalidРszTestExpressions.test_invalidz opname,op_str))r5rK)r6·-)r7┌*)r9·/)┌pow·**z left,rightcsNЗЗЗЗfddД}tddГ|ГtddГtаdб|Гtаб|ГdS)NcsЮИdkrdSttИГ}tабПFd}tаd|tбtj|ИИddН}tj|ИИddН}WdГn1sj0Ytа ||бtа |ИИИdб}|rЪJВdS)NrUz(invalid value encountered in true_divide┌ignoreTй┌use_numexprFrJ)r#r+┌warnings┌catch_warnings┌filterwarnings┌RuntimeWarningrrJrA┌assert_numpy_array_equalrO)r/┌msgr1r0й┌left┌op_strr"┌rightrr┌testitиs 0z/TestExpressions.test_binary_ops..testitr)FTrйrrZset_numexpr_threadsйrr"rbrarcrdrr`r┌test_binary_opsаs zTestExpressions.test_binary_ops))┌gt·>)┌lt·<)┌gez>=)┌lez<=)┌eqz==)┌nez!=csNЗЗЗЗfddД}tddГ|ГtddГtаdб|Гtаб|ГdS)NcshИd}Иd}ttИГ}tj|И|ddН}tj|И|ddН}tа||бtа|ИИ|dб}|rdJВdS)NrTrXFrJ)r#r+rrJrAr^rO)Zf12Zf22r/r1r0r`rrrd╥s z3TestExpressions.test_comparison_ops..testitr)FTrrerfrr`r┌test_comparison_ops├s z#TestExpressions.test_comparison_ops┌condcsJЗЗfddД}tddГ|ГtddГtаdб|Гtаб|ГdS)NcsXtjИjtjdН}|аИбtа|ИjИjdб}tа|ИjИjdб}tа ||бdS)N)rr) rL┌empty┌shapeZbool_┌fillr┌wherer@rAr^)┌cr1r0йrqr-rrrdъs z*TestExpressions.test_where..testitr)FTrre)rrqr-rdrrwr┌ test_whereчs zTestExpressions.test_wherez op_str,opname))rTr9)z//r:)rVrUcCsиttjаdбdktjаdбdkdЬГ}d|ЫdЭ}tt|Г}tа|б}tj t |dНП|||ГWdГn1st0Ytj t |dНП||j|jГWdГn1s░0Ytj t |dНП||jdГWdГn1sъ0Ytj t |dНП|d|jГWdГn1Рs&0Ytj t |dНП|d|ГWdГn1Рs`0Ytj t |dНП||dГWdГn1РsЪ0YdS) Nщ чр?й┌a┌bz operator 'z!' not implemented for bool dtypes)┌matchTF) rrLrM┌randr#r+┌re┌escape┌pytestZraises┌NotImplementedErrorr|r})rrbr"r-r_r;┌err_msgrrr┌!test_bool_ops_raise_on_arithmetic∙s & (,*,*z1TestExpressions.test_bool_ops_raise_on_arithmetic))rKr5)rSr7)rRr6c Csjd}ttjа|бdktjа|бdkdЬГ}ddddЬ}dd d dЬ}tt|Г}tt|||Г}|dkrjdStjd ddНРП┌tабП0|||Г} |||Г} tа | | бWdГn1s║0YtабП8||j |jГ} ||j |jГ} tа| | бWdГn1Рs0YtабП4||j d Г} ||j d Г} tа| | бWdГn1РsZ0YtабП4|d|j Г} |d|j Г} tа| | бWdГn1Рsи0YtабП0|d|Г} |d|Г} tа | | бWdГn1РsЄ0YtабП0||d Г} ||d Г} tа | | бWdГn1Рs<0YWdГn1Рs\0YdS)Nryrzr{·|·&·^)rKrSrR┌or_┌and_┌xor)rЖrЗrИrRTщ)Zmin_elementsF) rrLrMrr#r+rArYZassert_produces_warning┌assert_frame_equalr|r}Zassert_series_equal)rrbr"┌nr-┌subsZ sub_funcsr;Zfe┌r┌errr┌ test_bool_ops_warn_on_arithmeticsB& * , , , , z0TestExpressions.test_bool_ops_warn_on_arithmeticztest_input,expected)rrr3┌aa)r|r}rvr)r r|r)rщr3rУ)rrr3rУ)rrrZbbcCs>|jddЕddgfа|jddЕddgfб}tа||бdS)Nr|r)┌locrorArН)rZ test_inputr0r1rrr┌test_bool_ops_column_name_dtypeEs.z/TestExpressions.test_bool_ops_column_name_dtype┌axis)rrcCs~t}|dkr |jdddЕf}n|jddЕdf}dt_t||Г}tddГ|||dН}tddГ|||dН}tа||бdS)Nrrr)F)rЧT)┌_framerErrr#rrArН)rrЧr4r-r.Zop_funcr0r1rrr┌test_frame_series_axis_s z&TestExpressions.test_frame_series_axisr/)┌__mod__┌__rmod__┌__floordiv__┌ __rfloordiv__┌box┌scalarщ√ rМc Cs└dt_tаddб}||Г}t||Г}||Г}tddГ||Г}tddГtа||бt|ГD]^\} } |t kr||j | df}n|| }ztt| Г|Г|Г}WntyмYq\0||ks\JВq\dS)Nri╬ щ2r)FT) rrrLZaranger#rrArB┌ enumeraterrE┌int┌ZeroDivisionError)rr/rЮrЯ┌data┌objrr1r0rF┌elemZ scalar_resultrrr┌,test_python_semantics_with_numexpr_installedws$ zsH № №