U f/e @sddlZddlZddlZddlZddlmZddl m Z ej ddZ ej ddZ ej ddZejjd d d d dd d ddddg fd d d dd d ddddg fgd d gdddZddZddZejdddejdejjdddgddZddZdS) N) FloatingArraycCs:tjddgdtjgddgdtjgddgddS)NTF,booleanZdtype)pdarraynpnanr r O/tmp/pip-unpacked-wheel-tiezk1ph/pandas/tests/arrays/boolean/test_arithmetic.pydata s.r cCs(tjdgddgddgdddS)NTFrrrrr r r r left_arraysrcCstjdddgdddS)NTFrrrrr r r r right_arraysrz opname, expaddTFmul)idscCs2tt|}|||}tj|dd}t||dS)Nrr)getattroperatorrrtmassert_extension_array_equal)rropnameexpopresultexpectedr r r test_add_mul!s  rc Cs*d}tjt|d||W5QRXdS)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 Csd||}ttjdtjtjdtjtjtjtjtjg ddtdddddddddg }t||dS)Ng?gZfloat64rFT)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,c Cs|}t|}t||}d}tjt|d|dW5QRXd}tjt|d|tdW5QRX|dkrd}tjt|d|tjd|jdW5QRXdS) 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)rZSeriesrr!r"r#Z Timestampr/)r Zall_arithmetic_operatorsrsopsr$r r r test_error_invalid_valuesZs   r2)rZnumpyr r!ZpandasrZpandas._testingZ_testingrZ pandas.arraysrZfixturer rrmarkZ parametrizerr%r'paramZxfailr,r2r r r r sB