o g @sddlZddlZddlZddlZddlmZddl m Z ej ddZ ej ddZ ej ddZejjd d gd fd gd fgd d gdddZddZddZejdddejdejjdddgddZddZdS) N) FloatingArraycCs:tjddgdtjgddgdtjgddgddS)NTF,booleandtype)pdarraynpnanr r i/home/ubuntu/cloudmapper/venv/lib/python3.10/site-packages/pandas/tests/arrays/boolean/test_arithmetic.pydata s.rcCs(tjdgddgddgdddS)NTFrrrr r r r r left_arrays(rcCstjgddddS)N)TFNrrrrr r r r right_arraysrz opname, expadd) TTNTFNNNNmul) TFNFFNNNN)idscCs2tt|}|||}tj|dd}t||dS)Nrr)getattroperatorrr tmassert_extension_array_equal)rropnameexpopresultexpectedr r r test_add_mul!s rcCs@d}tjt|d ||WddS1swYdS)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_sub0s  "r&c CsV||}ttjdtjtjdtjtjtjtjtjg ddtgd}t||dS)Ng?gfloat64r) FFTFFTTTT)rr r infr rr)rrrrr r r test_div9s" r)rfloordivmodpowz#TODO follow int8 behaviour? GH34686)reason)markscCs:tt|}|||}||d|d}t||dS)NInt8)rrastyperr)rrrrrrr r r test_op_int8Es r1cCs|}t|}t||}d}tjt|d |dWdn1s$wYd}tjt|d|tdWdn1sDwY|dvrsd}tjt|d|tjd|jdWddS1slwYdSdS) 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''r fooz]unsupported operand type\(s\) for|Concatenation operation is not implemented for NumPy arrays20180101)__mul____rmul__zounsupported operand type\(s\) for|can only concatenate str|not all arguments converted during string formatting)index)rSeriesrr"r#r$ Timestampr6)rall_arithmetic_operatorsrsopsr%r r r test_error_invalid_valuesZs(   "r<)rnumpyr r"pandasrpandas._testing_testingr pandas.arraysrfixturerrrmark parametrizerr&r)paramxfailr1r<r r r r sB