o
    թZhT
                     @   sx   d dl Zd dlZd dlmZmZmZmZ ejdefddZ	ejdd Z
ejdd	 Zejd
d Zejdd ZdS )    N)	DataFrameIndexNaT
date_rangereturnc                   C   s2   t tjddttdtdtddddd	S )
zb
    Fixture for DataFrame of floats with DatetimeIndex

    Columns are ['A', 'B', 'C', 'D']
       )d      ABCDdtypez
2000-01-01r   B)periodsfreq)columnsindex)	r   nprandomdefault_rngstandard_normalr   listobjectr    r   r   R/var/www/html/lang_env/lib/python3.10/site-packages/pandas/tests/frame/conftest.pydatetime_frame   s
   r   c                  C   sJ   t tjddtdd tdD tdttdtdd} d	| d
< | S )z
    Fixture for DataFrame of floats and strings with index of unique strings

    Columns are ['A', 'B', 'C', 'D', 'foo'].
    r   )   r	   c                 S      g | ]}d | qS Zfoo_r   .0ir   r   r   
<listcomp>#       z&float_string_frame.<locals>.<listcomp>r   r   r
   )r   r   barZfoo)	r   r   r   r   r   r   ranger   r   dfr   r   r   float_string_frame   s   r'   c                  C   sP   t dd ttdg dD tdd tdD tdd	} | d
 d| d
< | S )z|
    Fixture for DataFrame of different float types with index of unique strings

    Columns are ['A', 'B', 'C', 'D'].
    c                 S   s(   i | ]\}}|t jd jd|dqS )r   r   r   )r   r   r   r   colr   r   r   r   
<dictcomp>2   s    z%mixed_float_frame.<locals>.<dictcomp>r
   )float32r+   r+   Zfloat64c                 S   r   r   r   r   r   r   r   r!   8   r"   z%mixed_float_frame.<locals>.<listcomp>r   r   r   CZfloat16)r   zipr   r   r$   r   Zastyper%   r   r   r   mixed_float_frame*   s   
r/   c                   C   s:   t dd ttdg dD tdd tdD tdd	S )
zz
    Fixture for DataFrame of different int types with index of unique strings

    Columns are ['A', 'B', 'C', 'D'].
    c                 S   s    i | ]\}}|t jd |dqS )r   r   )r   Zonesr(   r   r   r   r*   G   s    z#mixed_int_frame.<locals>.<dictcomp>r
   )Zint32Zuint64Zuint8Zint64c                 S   r   r   r   r   r   r   r   r!   K   r"   z#mixed_int_frame.<locals>.<listcomp>r   r   r,   )r   r.   r   r   r$   r   r   r   r   r   mixed_int_frame?   s   r0   c                  C   sD   t tdddtddddtddddd} t| jd< t| jd	< | S )
a  
    Fixture for DataFrame of date_range Series with different time zones

    Columns are ['A', 'B', 'C']; some entries are missing

               A                         B                         C
    0 2013-01-01 2013-01-01 00:00:00-05:00 2013-01-01 00:00:00+01:00
    1 2013-01-02                       NaT                       NaT
    2 2013-01-03 2013-01-03 00:00:00-05:00 2013-01-03 00:00:00+01:00
    Z20130101   )r   z
US/Eastern)r   tzZCET)Ar   r-   )   r4   )r4   r   )r   r   r   Zilocr%   r   r   r   timezone_frameO   s   


r5   )numpyr   ZpytestZpandasr   r   r   r   Zfixturer   r'   r/   r0   r5   r   r   r   r   <module>   s    


