a Pf@sddlZddlZddlZddlZddlmZej ddZ ej ddZ ej ddZ ej jdd gd fd gd fgd d gd ddZddZddZej dgdddZddZdS)NcCs:tjddgdtjgddgdtjgddgddS)>Fixture returning boolean array with valid and missing values.TF,booleanZdtype)pdarraynpnanr r d/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/arrays/boolean/test_arithmetic.pydata s.r cCs(tjdgddgddgdddS)rTFNrrrrr r r r left_arraysrcCstjgddddS)r)TFNrrrrr r r r right_arraysrz opname, expadd) TTNTFNNNNmul) TFNFFNNNN)ZidscCs2tt|}|||}tj|dd}t||dS)Nrr)getattroperatorrrtmassert_extension_array_equal)rropnameexpopresultexpectedr r r test_add_mul#s  rcCs>d}tjt|d||Wdn1s00YdS)Nznumpy boolean subtract, the `-` operator, is (?:deprecated|not supported), use the bitwise_xor, the `\^` operator, or the logical_xor function instead\.match)pytestraises TypeErrorrrmsgr r r test_sub2sr%cCsd}tjt|d(t|jt|jWdn1s@0Ytjt|d||Wdn1sv0YdS)N-operator '.*' not implemented for bool dtypesr)r r!NotImplementedErrorrSeries_datar#r r r test_div;s 6r*r)floordivmodpowcCstt|}|dkrRd}tjt|d|||}Wdn1sD0YdS|||}||d|d}t||dS)Nr,r&rZInt8)rrr r!r'Zastyperr)rrrrr$rrr r r test_op_int8Es ( r.c Cs|rddl}t|jjtf}nt}|}t|}t||}d}tj t|d|dWdn1sh0Yd gd}tj ||d|t dWdn1s0Y|dvrd gd }tj ||d$|tjd|j d Wdn1s0YdS) Nrzdid not contain a loop with signature matching types|BooleanArray cannot perform the operation|not supported for the input types, and the inputs could not be safely coerced to any supported types according to the casting rule ''safe''rZfoo|)!unsupported operand type\(s\) forz;Concatenation operation is not implemented for NumPy arrays has no kernelZ20180101)__mul____rmul__)r0zcan only concatenate strz4not all arguments converted during string formattingr1znot implemented)index) Zpyarrowr"libZArrowNotImplementedErrorr'rr(rr r!join Timestampr4) r Zall_arithmetic_operatorsZusing_infer_stringpaerrrsopsr$r r r test_error_invalid_values]s,  &,  r<)rnumpyr r ZpandasrZpandas._testingZ_testingrZfixturer rrmarkZ parametrizerr%r*r.r<r r r r s4