o
    믪g                     @   sL   d dl Zd dlZd dlmZ d dlmZ dd Zdd Z	dd Z
d	d
 ZdS )    N)Seriesc                  C   s  t tjdd} | dk}| | tj}t|| 	| | | }| 	|}t|| | | |  }| 	||  }t|| t g d| j
d}|   }||d d  }|	|d d }t|| ||d d  | }|	|d d | }t|| d}tjt|d | 	d	 W d    n1 sw   Y  tjt|d | 	|d d j|   W d    d S 1 sw   Y  d S )
N      r   )TFFTF)index   z,Array conditional must be same shape as self)match   )r   nprandomdefault_rngstandard_normalwherenantmassert_series_equalmaskr   abspytestraises
ValueErrorvalues)scondrsrs2s2msg r   b/var/www/html/chatdoc2/venv/lib/python3.10/site-packages/pandas/tests/series/indexing/test_mask.py	test_mask   s2   

"r   c                  C   sB   t g d} | | dktj}t ddtjtjg}t|| d S )N)r   r   r      r   r   r   r   r	   r   r   r   )serresultexpectedr   r   r   test_mask_casts)   s   r%   c                  C   s8   t ddg} | ddg}t tjdg}t|| d S )Nr   r   TFr!   )r"   resexpr   r   r   test_mask_casts21   s   r(   c                  C   s   t tjdd} | dk}|  }|j|dd t|	 | |   t|| | |  }|j||  dd t|| ||   d S )Nr   r   r   T)inplace)
r   r	   r
   r   r   copyr   r   r   dropna)r   r   r   r   r   r   test_mask_inplace:   s   r,   )numpyr	   r   pandasr   pandas._testing_testingr   r   r%   r(   r,   r   r   r   r   <module>   s    !	