o
    Eh8
                     @   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   s.   t tjddttdtdddddS )	zb
    Fixture for DataFrame of floats with DatetimeIndex

    Columns are ['A', 'B', 'C', 'D']
       )d      ABCDz
2000-01-01r   B)periodsZfreq)columnsindex)r   nprandomdefault_rngstandard_normalr   listr    r   r   ;lib/python3.10/site-packages/pandas/tests/frame/conftest.pydatetime_frame   s
   
r   c                  C   sF   t tjddtdd tdD tdt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   dtyper
   )r   r   ZbarZfoo)	r   r   r   r   r   r   rangeobjectr   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   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   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   ZtzZCET)Ar   r*   )   r0   )r0   r   )r   r   r   Zilocr#   r   r   r   timezone_frameO   s   


r1   )Znumpyr   ZpytestZpandasr   r   r   r   Zfixturer   r$   r,   r-   r1   r   r   r   r   <module>   s    


