o
    թZh                     @   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pytestZraises
ValueErrorvalues)scondrsZrs2s2msg r   ]/var/www/html/lang_env/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)Zinplace)
r   r	   r
   r   r   copyr   r   r   Zdropna)r   r   r   r   r   r   test_mask_inplace:   s   r(   )numpyr	   r   Zpandasr   Zpandas._testingZ_testingr   r   r#   r&   r(   r   r   r   r   <module>   s    !	