
    bMh                     F    d dl Zd dlZd dlmZ d dlmZ d Zd Z	d Z
d ZdS )    N)Seriesc                     t          t          j                            d                              d                    } | dk    }|                     | t          j                  }t          j        || 	                    |                     |                     |           }| 	                    |          }t          j        ||           |                     | |            }| 	                    ||            }t          j        ||           t          g d| j
                  }|                                  }|                    |d d                    }|	                    |d d                   }t          j        ||           |                    |d d          |           }|	                    |d d         |           }t          j        ||           d}t          j        t          |          5  | 	                    d	           d d d            n# 1 swxY w Y   t          j        t          |          5  | 	                    |d d         j        |             d d d            d S # 1 swxY w 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msgs         f/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pandas/tests/series/indexing/test_mask.py	test_maskr       s   ry$$Q''77::;;Aq5D	
$		B2qvvd||,,,	
$B
&&,,C2s###	
$		B
&&r

C2s###33317CCCD5577B	48)		B
''$rr(

C2s###	48)bS	!	!B
''$rr(RC
 
 C2s###
8C	z	-	-	-  	q			              	z	-	-	- $ $	tBQBx###$ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $s$   H##H'*H'	%I;;I?I?c                      t          g d          } |                     | dk    t          j                  }t          ddt          j        t          j        g          }t	          j        ||           d S )N)r
   r   r      r   r
   r   r   r   r   r   r   )serresultexpecteds      r   test_mask_castsr'   )   s_    


CXXcAgrv&&Fq!RVRV,--H68,,,,,    c                      t          ddg          } |                     ddg          }t          t          j        dg          }t	          j        ||           d S )Nr
   r   TFr#   )r$   resexps      r   test_mask_casts2r,   1   sR    
!Q..C
((D%=
!
!C
"&!

C3$$$$$r(   c                  @   t          t          j                            d                              d                    } | dk    }|                                 }|                    |d           t          j        |	                                | |                     t          j        ||                     |                     |                                 }|                    ||  d           t          j        ||                     ||                       d S )Nr   r   r   T)inplace)
r   r   r   r   r   copyr   r   r   dropna)r   r   r   s      r   test_mask_inplacer1   :   s    ry$$Q''77::;;Aq5D	
BGGD$G299;;4%1112qvvd||,,,	
BGGD1"dG###2qvvdQB//00000r(   )numpyr   r   pandasr   pandas._testing_testingr   r    r'   r,   r1    r(   r   <module>r7      s                    $ $ $B- - -% % %1 1 1 1 1r(   