a
    PfT
                     @   sx   d dl Zd dlZd dlmZmZmZmZ ejed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   T/var/www/ai-form-bot/venv/lib/python3.9/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   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   Z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   s   g | ]}d | qS 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   s   g | ]}d | qS 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    r.   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+   )   r2   )r2   r   )r   r   r   Zilocr$   r   r   r   timezone_frameO   s    


r3   )numpyr   ZpytestZpandasr   r   r   r   Zfixturer   r%   r-   r.   r3   r   r   r   r   <module>   s   


