o
    _~#g                     @   sb   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 ed G dd dZG dd dZdS )    N)Categorical	DataFrame
MultiIndexSeries
date_rangexarrayc                   @   s2   e Zd Zejdd Zdd Zdd Zdd Zd	S )
TestDataFrameToXArrayc                 C   s`   t tdttddtdddtjddd	d
g dttdtdddtd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   )self r$   /var/www/static.ux5.de/https/Moving-Object-Detection-with-OpenCV/env/lib/python3.10/site-packages/pandas/tests/generic/test_to_xarray.pydf   s   

zTestDataFrameToXArray.dfc                 C   s   |}t |dkrtd ddlm} |d d |_d|j_d|j_| }|j	d dks/J t |j
dks8J t |jdksAJ tt|j
 dg t||sTJ | }|d	 |s`tnd
|d	< d |j_t| | d S )Nr   z'Test doesn't make sense for empty indexDatasetr   foobarr
      r   string[pyarrow_numpy])lenpytestskipr   r(   indexnamecolumns	to_xarraysizescoords	data_varstmassert_almost_equalr   keys
isinstancecopyr"   objectassert_frame_equalto_dataframe)r#   
index_flatr&   using_infer_stringr0   r(   resultexpectedr$   r$   r%   test_to_xarray_index_types    s&   

z0TestDataFrameToXArray.test_to_xarray_index_typesc                 C   sH   ddl m} d|j_|dd  }|jd dksJ t||s"J d S )Nr   r'   r)   )r   r(   r0   r1   r3   r4   r:   )r#   r&   r(   rA   r$   r$   r%   test_to_xarray_empty<   s
   z*TestDataFrameToXArray.test_to_xarray_emptyc                 C   s   ddl m} tjdgtdgddgd|_| }|jd dks"J |jd dks+J t|j	d	ks4J t|j
d
ks=J tt|j	 ddg t||sQJ | }| }|d |satnd|d< d |j_t|| d S )Nr   r'   r   r   onetwonamesr
      r+   r   r,   )r   r(   r   from_productr   r0   r3   r4   r-   r5   r6   r7   r8   r   r9   r:   r>   r;   r"   r<   r2   r1   r=   )r#   r&   r@   r(   rA   rB   r$   r$   r%   test_to_xarray_with_multiindexD   s    
z4TestDataFrameToXArray.test_to_xarray_with_multiindexN)	__name__
__module____qualname__r.   fixturer&   rC   rD   rK   r$   r$   r$   r%   r      s    
r   c                   @   s$   e Zd Zdd Zdd Zdd ZdS )TestSeriesToXArrayc                 C   s   |}ddl m} ttt||dd}d|j_| }t| t|t|ks)J t|j	dks2J t
t|j	 dg t||sEJ t
| | d S )Nr   	DataArrayint64)r0   r   r)   r
   )r   rR   r   r   r-   r0   r1   r3   reprr5   r7   r8   r   r9   r:   assert_series_equal	to_series)r#   r?   r0   rR   serrA   r$   r$   r%   rC   [   s   z-TestSeriesToXArray.test_to_xarray_index_typesc                 C   st   ddl m} tg td}d|j_| }t|dksJ t|jdks%J t	
t|j dg t||s8J d S )Nr   rQ   r   r)   r
   )r   rR   r   r<   r0   r1   r3   r-   r5   r7   r8   r   r9   r:   )r#   rR   rW   rA   r$   r$   r%   rD   m   s   z'TestSeriesToXArray.test_to_xarray_emptyc                 C   s   ddl m} tjddgtdgddgd}ttd	d
|d}| }t|dks*J t	t
|j ddg t||s>J | }t|| d S )Nr   rQ   r   r   r   rE   rF   rG      rS   )r   r0   rI   )r   rR   r   rJ   r   r   r3   r-   r7   r8   r   r5   r9   r:   rV   rU   )r#   rR   mirW   rA   resr$   r$   r%   rK   x   s   z1TestSeriesToXArray.test_to_xarray_with_multiindexN)rL   rM   rN   rC   rD   rK   r$   r$   r$   r%   rP   Z   s    rP   )numpyr    r.   pandasr   r   r   r   r   pandas._testing_testingr7   importorskipr   rP   r$   r$   r$   r%   <module>   s    
J