U e"e| @sddlZddlZddlZddlZddlmZej ddZ ej ddZ ej ddZ ej jdd d d dd d ddddg fd d d dd d ddddg fgd d gd ddZddZddZej ddddgddZddZdS)NcCs:tjddgdtjgddgdtjgddgddS)>Fixture returning boolean array with valid and missing values.TF,booleanZdtype)pdarraynpnanr r O/tmp/pip-unpacked-wheel-b4zrpxve/pandas/tests/arrays/boolean/test_arithmetic.pydata s.r cCs(tjdgddgddgdddS)rTFNrrrrr r r r left_arraysrcCstjdddgdddS)rTFNrrrrr 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 TypeErrorrrmsgr r r test_sub2sr&c Cs\d}tjt|dt|jt|jW5QRXtjt|d||W5QRXdS)N-operator '.*' not implemented for bool dtypesr)r!r"NotImplementedErrorrSeries_datar$r r r test_div;s "r+rfloordivmodpowc Csntt|}|dkr>d}tjt|d|||}W5QRXdS|||}||d|d}t||dS)Nr-r'rZInt8)rrr!r"r(Zastyperr)rrrrr%rrr r r test_op_int8Es  r/c Cs|}t|}t||}d}tjt|d|dW5QRXdddg}tjt|d|tdW5QRX|dkrddd d g}tjt|d|tjd|jd W5QRXdS) 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''rZfoo|z!unsupported operand type\(s\) forz;Concatenation operation is not implemented for NumPy arraysZ20180101)__mul____rmul__zcan only concatenate strz4not all arguments converted during string formatting)index) rr)rr!r"r#joinZ Timestampr3)r Zall_arithmetic_operatorsrsopsr%r r r test_error_invalid_values]s.  r7)rZnumpyr r!ZpandasrZpandas._testingZ_testingrZfixturer rrmarkZ parametrizerr&r+r/r7r r r r s: