a GGb @sddlZddlZddlZddlZddlmZddl m Z ej ddZ ej ddZ ej ddZejjd d gd fd gd fgd d gdddZddZddZejdddejdejjdddgddZddZdS) N) FloatingArraycCs:tjddgdtjgddgdtjgddgddS)NTF,booleanZdtype)pdarraynpnanr r Jlib/python3.9/site-packages/pandas/tests/arrays/boolean/test_arithmetic.pydata s.r cCs(tjdgddgddgdddS)NTFrrrrr r r r left_arraysrcCstjgddddS)N)TFNrrrrr r r r right_arraysrz opname, expadd) TTNTFNNNNmul) TFNFFNNNN)ZidscCs2tt|}|||}tj|dd}t||dS)Nrr)getattroperatorrrtmassert_extension_array_equal)rropnameZexpopresultexpectedr 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 TypeError)rrmsgr r r test_sub0sr#c CsV||}ttjdtjtjdtjtjtjtjtjg ddtgd}t||dS)Ng?gZfloat64r) FFTFFTTTT)rr rinfr rr)rrrrr r r test_div9s" r%rfloordivmodpowz#TODO follow int8 behaviour? GH34686)reason)ZmarkscCs:tt|}|||}||d|d}t||dS)NZInt8)rrZastyperr)rrrrrrr r r test_op_int8Es  r*cCs|}t|}t||}d}tjt|d|dWdn1sH0Yd}tjt|d|tdWdn1s0Y|dvrd}tjt|d$|tjd|jdWdn1s0YdS) Nzdid 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''rZfooz]unsupported operand type\(s\) for|Concatenation operation is not implemented for NumPy arraysZ20180101)__mul____rmul__zounsupported operand type\(s\) for|can only concatenate str|not all arguments converted during string formatting)index)rZSeriesrrr r!Z Timestampr-)r Zall_arithmetic_operatorsrsopsr"r r r test_error_invalid_valuesZs   &,r0)rZnumpyr rZpandasrZpandas._testingZ_testingrZ pandas.arraysrZfixturer rrZmarkZ parametrizerr#r%ZparamZxfailr*r0r r r r s@