
    Pi                        d dl Zd dlZd dlmZmZmZmZmZ d dl	m
Z d dlmZ  ej        d          Ze7 eej                   ed          k     rej                            d          Z G d d          Z G d	 d
          ZdS )    N)Categorical	DataFrame
MultiIndexSeries
date_range)Versionxarray2025.1.0z6ignore:Converting non-nanosecond precision:UserWarningc                   @    e Zd Zej        d             Zd Zd Zd ZdS )TestDataFrameToXArrayc                 n   t          t          d          t          t          dd                    t          j        dd                              d          t          j        ddd	
          g dt          t          d                    t          dd          t          ddd          d          S )Nabcd            u1g      @g       @float64dtype)TFTF20130101   )periodsz
US/Eastern)r   tz)abcdefgh)r   listrangenparangeastyper   r   )selfs    w/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pandas/tests/generic/test_to_xarray.pydfzTestDataFrameToXArray.df   s    &\\%1++&&Yq!__++D11YsCy999/// f..
A666
A,GGG	 	
 
 	
    c                    |}t          |          dk    rt          j        d           t          t          j                  t          d          k     rt          j        d           |d d         |_        d|j        _        d|j        _        |	                                }|j
        d         dk    sJ t          |j                  dk    sJ t          |j                  d	k    sJ t          j        t          |j                                                  dg           t#          |t          j                  sJ |                                }d |j        _        t          j        |                                |           d S )
Nr   z'Test doesn't make sense for empty indexz2025.9.0z7Xarray bug https://github.com/pydata/xarray/issues/9661r   foobarr      )lenpytestskipr   r	   __version__indexnamecolumns	to_xarraysizescoords	data_varstmassert_almost_equalr#   keys
isinstanceDatasetcopyassert_frame_equalto_dataframe)r(   
index_flatr*   requestr4   resultexpecteds          r)   test_to_xarray_index_typesz0TestDataFrameToXArray.test_to_xarray_index_types&   sU   u::??KABBB6%&&)<)<<<KQRRR!9
|E"a''''6=!!Q&&&&6#$$))))
tFM$6$6$8$899E7CCC&&.11111
 7799 $
f1133X>>>>>r+   c                     d|j         _        |dd                                         }|j        d         dk    sJ t	          |t
          j                  sJ d S )Nr-   r   )r4   r5   r7   r8   r>   r	   r?   )r(   r*   rE   s      r)   test_to_xarray_emptyz*TestDataFrameToXArray.test_to_xarray_empty?   sZ    AaC""$$|E"a''''&&.1111111r+   c                    t          j        dgt          d          gddg          |_        |                                }|j        d         dk    sJ |j        d         dk    sJ t          |j                  dk    sJ t          |j                  dk    sJ t          j
        t          |j                                                  ddg           t          |t          j                  sJ |                                }|                                }|d	                             |st&          nd
          |d	<   t)          t          j                  t)          d          k     r<|d                             d          |d<   |d                             d          |d<   d |j        _        t          j        ||           d S )Nr   r   onetwonamesr      r/   r    strr
   r!   zM8[ns]r"   zM8[ns, US/Eastern])r   from_productr$   r4   r7   r8   r0   r9   r:   r;   r<   r#   r=   r>   r	   r?   rB   r@   r'   objectr   r3   r6   r5   rA   )r(   r*   using_infer_stringrE   rF   s        r)   test_to_xarray_with_multiindexz4TestDataFrameToXArray.test_to_xarray_with_multiindexE   s   *SE588+<UENSSS|E"a''''|E"a''''6=!!Q&&&&6#$$))))
tFM$6$6$8$899E5>JJJ&&.11111$$&&7799 ,,,7FF%
 
 6%&&)<)<<<$SM00::HSM$SM001EFFHSM $
fh/////r+   N)	__name__
__module____qualname__r1   fixturer*   rG   rI   rT    r+   r)   r   r      s\        ^
 
 ^
? ? ?22 2 20 0 0 0 0r+   r   c                        e Zd Zd Zd Zd ZdS )TestSeriesToXArrayc                 2   |}t          t          t          |                    |d          }d|j        _        |                                }t          |           t          |          t          |          k    sJ t          |j                  dk    sJ t          j	        t          |j                                                  dg           t          |t          j                  sJ t          j        |                                |           d S )Nint64)r4   r   r-   r   )r   r$   r0   r4   r5   r7   reprr9   r;   r<   r#   r=   r>   r	   	DataArrayassert_series_equal	to_series)r(   rC   rD   r4   serrE   s         r)   rG   z-TestSeriesToXArray.test_to_xarray_index_types]   s    U3u::&&e7CCC	V6{{c%jj((((6=!!Q&&&&
tFM$6$6$8$899E7CCC&&"233333 	v//11377777r+   c                 |   t          g t                    }d|j        _        |                                }t          |          dk    sJ t          |j                  dk    sJ t          j        t          |j        
                                          dg           t          |t          j                  sJ d S )Nr   r-   r   r   )r   rR   r4   r5   r7   r0   r9   r;   r<   r#   r=   r>   r	   r_   )r(   rb   rE   s      r)   rI   z'TestSeriesToXArray.test_to_xarray_emptym   s    Rv&&&	6{{a6=!!Q&&&&
tFM$6$6$8$899E7CCC&&"23333333r+   c                    t          j        ddgt          d          gddg          }t          t          d          d|	          }|                                }t          |          d
k    sJ t          j        t          |j	        
                                          ddg           t          |t          j                  sJ |                                }t          j        ||           d S )Nr   r   r   rK   rL   rM      r]   )r   r4   rO   )r   rQ   r$   r   r7   r0   r;   r<   r#   r9   r=   r>   r	   r_   ra   r`   )r(   mirb   rE   ress        r)   rT   z1TestSeriesToXArray.test_to_xarray_with_multiindexv   s    $sCj%((%;E5>RRRU1XXWB7776{{a
tFM$6$6$8$899E5>JJJ&&"233333  
sC(((((r+   N)rU   rV   rW   rG   rI   rT   rY   r+   r)   r[   r[   \   sA        8 8 8 4 4 4) ) ) ) )r+   r[   )numpyr%   r1   pandasr   r   r   r   r   pandas._testing_testingr;   pandas.util.versionr   importorskipr	   r3   markfilterwarnings
pytestmarkr   r[   rY   r+   r)   <module>rq      s<                           ' ' ' ' ' '		X	&	&	''&"455
8K8KKK++@ J
C0 C0 C0 C0 C0 C0 C0 C0L") ") ") ") ") ") ") ") ") ")r+   