U md| @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 d/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/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)ZidscCs2tt|}|||}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"join Timestampr2)r Zall_arithmetic_operatorsrsZopsr$r r r test_error_invalid_values]s.  r6)rnumpyr r ZpandasrZpandas._testingZ_testingrZfixturer rrmarkZ parametrizerr%r*r.r6r r r r s: