
    bMh                     ~    d dl mZ d dlZd dlZd dlmZ d dlZd dlm	Z	m
Z
mZmZmZ d dlmZ  G d d          Zd ZdS )    	timedeltaN)iNaT)CategoricalIndexNaTSeriesisnac                   n    e Zd Zd Zd Zd Zej                            d          d             Z	d Z
dS )	TestSeriesMissingDatac                 4   t          t          ddt          j        dg                    }t	          j        |j        j        t          ddg                     t	          j	        |j
        j        t          j        g dt          j                             d S )Nab)r      r   dtype)r	   r   npnantmassert_index_equalcat
categoriesr   assert_numpy_array_equalvaluescodesarrayint8)selfss     `/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pandas/tests/series/test_missing.pytest_categorical_nan_handlingz3TestSeriesMissingData.test_categorical_nan_handling   s    ;S"&#67788
ae.sCj0A0ABBB
#HNBH]]]"'BBB	
 	
 	
 	
 	
    c                    t          dt          j        t          j        t          j        dg          }d}t          j        t          |          5  t	          j	        dd          5  |
                                }|                                }d d d            n# 1 swxY w Y   d d d            n# 1 swxY w Y   t          g d          }t          ddgdd	g
          }t          j        ||           t          j        ||           d S )Nr   g      ?z"use_inf_as_na option is deprecatedmatchzmode.use_inf_as_naT)FTTTFr      )index)r	   r   infr   pdNAr   assert_produces_warningFutureWarningoption_contextr
   dropnaassert_series_equal)r   r    msgrdredes          r!   test_isna_for_infz'TestSeriesMissingData.test_isna_for_inf   so   C4552'SAAA 	  	 "#7>>    FFHHXXZZ                             	  	  	  	  	  	  	  	  	  	  	  	  	  	  	  33344S#Jq!f---
q!$$$
r2&&&&&s6   B2&)BB2B	B2"B	#B22B69B6c                 :   t          d t          d          D                       }|                                }t          j        |d<   t          |d                   sJ |d         j        t          k    sJ |d         |d<   t          |d                   rJ t          j	        t          d          5  t          |d<   d d d            n# 1 swxY w Y   t          |d                   rJ |j        t          j        k    sJ |d         t          k    sJ |d         |d<   t          |d                   rJ t          |d<   t          |d                   sJ |d         j        t          k    sJ |d         |d<   t          |d                   rJ t          j        t          d	                    }t          j        t          d
	                    }t          j        |||k    ||k     z  <   t          |                                          dk    sJ d S )Nc                 .    g | ]}t          |           S )daysr   ).0is     r!   
<listcomp>z>TestSeriesMissingData.test_timedelta64_nan.<locals>.<listcomp>)   s#    :::1YA&&&:::r#   
   r   zincompatible dtyper%   r         r9      )r	   rangecopyr   r   r
   _valuer   r   r,   r-   r   object_r   timedelta64r   sum)r   tdtd1td3td7s        r!   test_timedelta64_nanz*TestSeriesMissingData.test_timedelta64_nan(   s1   ::b		:::;; ggiiACF||1v}$$$$AAA<< '=QRRR 	 	CF	 	 	 	 	 	 	 	 	 	 	 	 	 	 	A<<yBJ&&&&1v~~~~AAA<<ACF||1v}$$$$AAA<< nYA...//nYA...//&(fBHc"#Bxx||~~""""""s   6CCCz:Chained inequality raises when trying to define 'selector')reasonc                 j    d|cxk    odk    nc }|dk    |dk    z  }t          j        ||           d S )Ng      g      ?)r   r0   )r   datetime_seriesselectorexpecteds       r!   test_logical_range_selectz/TestSeriesMissingData.test_logical_range_selectI   sR     ?1111c1111#t+30FG
x22222r#   c                    |                                 }|j                            d           |_        t          j        |d d d<   |                                }t          |          |                                k    sJ t          j	        ||dd d                    t          j	        ||t          j        |                              d S )Nr?   r   )rC   r(   
_with_freqr   r   r/   lencountr   r0   r*   notna)r   rO   tsresults       r!   
test_validz TestSeriesMissingData.test_validT   s    !!##8&&t,,&33Q36{{bhhjj((((
vr!$Q$x000
vr"(2,,'788888r#   N)__name__
__module____qualname__r"   r6   rL   pytestmarkxfailrR   rZ    r#   r!   r   r      s        
 
 

' 
' 
'# # #B [K   3 3 39 9 9 9 9r#   r   c                      t          ddgd          } | j        du sJ d| j        v sJ |                                 }|j        du sJ t	          |d          rJ t
          j        |j        d<   |j        d	u sJ d S )
Nr   r   float64r   Fhasnans_cacher   T)r   rd   re   	to_serieshasattrr   r   iloc)idxsers     r!    test_hasnans_uncached_for_seriesrk   _   s     Ai
(
(
(C;%
""""
--//C;%sH%%%%%6CHRL;$r#   )datetimer   numpyr   r^   pandas._libsr   pandasr*   r   r   r   r	   r
   pandas._testing_testingr   r   rk   ra   r#   r!   <module>rr      s                                           I9 I9 I9 I9 I9 I9 I9 I9X
 
 
 
 
r#   