U
    sVc                     @   s(   d dl Zd dlmZ dd Zdd ZdS )    Nc                  C   sD   t jddgdd} | | dk}t jt jdgdd}t|| d S N      ZInt64)Zdtype)pdSerieswhereZNAtmZassert_series_equal)Zserresultexpected r   M/tmp/pip-unpacked-wheel-xj8nt62q/pandas/tests/arrays/integer/test_indexing.py(test_array_setitem_nullable_boolean_mask   s    r   c                  C   sB   t jddgddj} d| | dk< t jddgdd}t| | d S r   )r   r   arrayr   Zassert_extension_array_equal)Zarrr
   r   r   r   test_array_setitem   s    r   )Zpandasr   Zpandas._testingZ_testingr   r   r   r   r   r   r   <module>   s   