3 UdB@s:ddlZddlZddlmZddljZddZddZ dS)N)Seriesc Csttjjd}|dk}|j|tj}tj||j||j|}|j|}tj|||j|| }|j|| }tj||tdddddg|j d}|j }|j|dd}|j|dd}tj|||j|dd| }|j|dd| }tj||d}t j t |d|jd WdQRXt j t |d|j|ddj| WdQRXtd d dd g}|j|d ktj}td d tjtjg}tj||td d g}|jddg}ttjd g} tj|| dS) NrTF)indexz,Array conditional must be same shape as self)match)rnprandomrandnwherenantmassert_series_equalmaskrabspytestZraises ValueErrorvalues) scondrsZrs2s2msgresultexpectedresexprZ/var/www/html/virt/lib64/python3.6/site-packages/pandas/tests/series/indexing/test_mask.py test_masks>       "  r!cCsttjjd}|dk}|j}|j|ddtj|j||tj||j||j}|j|| ddtj||j|| dS)NrrT)Zinplace) rr r r copyrrrZdropna)rrrrrr test_mask_inplace6sr#) Znumpyr rZpandasrZpandas._testingZ_testingrr!r#rrrr s   .