U
    *if                    @   s  d dl mZmZ d dlmZmZmZ d dlZd dlZd dlZd dl	Z	d dl
Zd dlmZ d dlm  mZ d dlZd dlZd dlmZ d dlm  mZ d dlmZ d dlmZmZmZmZ d dlZ d dlm!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/ d dl0m1Z2 d dl3m4Z4m5Z5m6Z6m7Z7 d d	l8m9Z9 d
ddgZ:ddddddddgZ;G dd dZ<G dd dZ=G dd dZ>G dd dZ?dd Z@G dd  d ZAdS )!    )OrderedDictabc)datedatetime	timedeltaN)np_version_under1p19)is_integer_dtype)DatetimeTZDtypeIntervalDtypePandasDtypePeriodDtype)CategoricalCategoricalIndex	DataFrameDatetimeIndexIndexInterval
MultiIndexPeriod
RangeIndexSeries	Timedelta	Timestampcut
date_rangeisna)DatetimeArrayIntervalArrayPeriodArraySparseArray)
Int64IndexZfloat16Zfloat32float64uint8Zuint16Zuint32uint64Zint8Zint16int32int64c                   @   s  e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Ze	j
dddgdd Zdd Ze	j
ddd dd dd dd dd dd dd dd dd dd d d d!d d"d gd#d$ Ze	j
d%g ged&ed'fg g ged(ed'fd)d* d+D ged&ed'fgd,d- Zd.d/ Zd0d1 Zd2d3 Zd4d5 Zd6d7 Zejd8d9 Zejd:d; Zd<d= Ze	j
jed>d?d@dA ZdBdC ZdDdE ZdFdG Z dHdI Z!dJdK Z"dLdM Z#e	j
dNe$j%dOge&dPe$%d(dQ gdRge$j%dSge&dPe$%dTgd(dQ  d( ggdUdV Z'dWdX Z(dYdZ Z)d[d\ Z*d]d^ Z+d_d` Z,dadb Z-dcdd Z.dedf Z/e	j
dgd(e$j0dhdigdjdk Z1e	j
dld(e$j0dhe2dmgdndo Z3e	j
dle$j0dhe2dmgdpdq Z4drds Z5dtdu Z6dvdw Z7dxdy Z8dzd{ Z9d|d} Z:d~d Z;dd Z<dd Z=dd Z>dd Z?dd Z@dd ZAdd ZBdd ZCdd ZDdd ZEdd ZFdd ZGdd ZHdd ZIe	j
deJdeKdfeLd'ddeMddfeNdddeOddfgdd ZPdd ZQdd ZRdd ZSdd ZTdd ZUdd ZVdd ZWe	j
jXdd ZYdd ZZe	j
ddhe[e\dddge&e$j]fdhdhddgde$^dfdhe[e\dddge_e$^dfi dhddgdhe$j]fdd&ie[e\de[de_e$^dfgdd Z`e	j
ddddhgdeajbfdddhgdeajcfd&d(dhgdeajdfdddgdeajeffdd΄ ZfddЄ Zgdd҄ ZhddԄ Ziddք Zjdd؄ Zkddڄ Zldd܄ Zmddބ Zndd Zodd Zpdd Zqe	j
deNdggdeNdigdeNdgideNdigdd Zrdd Zsdd Ztdd Zudd Zvdd Zwdd Zxdd Zydd Zzdd Z{e	j
de|e}gdd Z~e	j
de|e}gdd  Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Ze	j
dd+g fd+g fdd+d+gfdg g gfg g fg g gfd&gd(gfd&gd(ggfd&d(dgdddgfd&d(dgdddggfgdd Zdd Zdd Zdd Zdd Zdd Zd d! Zd"d# Zd$d% Zd&d' Ze	j
d(d)d*d+d,d-d.d/gd0d1 Zd2d3 Zd4d5 Zd6d7 Zd8d9 Zd:d; Zd<d= Zd>d? Zd@dA ZdBdC ZdDdE ZdFdG ZdHdI ZdJdK ZdLdM Ze	j
dNe$%dhdhdhdhe dhge$%dhdhe dhge$dOgdhgge$dOgeajggdhge$dOggdhgeajggeajge$dOggeajgdhgggdPdQ Ze	j
dRdSdTdUdVge	j
dWdXdYdZd[d\d]d^d_gd`da Ze	j
dRdSdTdUdVge	j
dWdbdcdddedfdgdhgdidj Zdkdl Zdmdn Zdodp Zejdqdr Ze	j
dsed&d(dgdddggd&e$j0gdted&d(dgdddggdudve$j0gdwed'd&d(dgddddxgge$j0dudve$j0gdwedyd&d(dzgddddxgge$j0dudve$j0gdwedyd&d(dzgddddxgge$j0d&d(d(gdwgd{d| Zd}d~ Zdd Zdd Zdd Zdd Zdd Zdd Zdd Ze	j
dWejej ej ej ej ej ej dd Ze	j
dWejej ej dd Zdd Ze	j
je	j
dWdhddgdd ZŐdd Ze	j
dee[ded&e$j0e$j0e$j0geeLd'd&eLd&dgeeajːddddgdd Z̐dd Z͐dd Zΐdd Zϐdd ZАdd Zѐdd ZҐdd ZӐdd Ze	j
dd&d(gddge[eՐdd(dde[eՐdd(dddeLd'ddggdd Z֐dd Ze	j
dddgdd Zؐdd Zِdd ZڐddÄ Ze	j
dd&gd(gge$%d&gd(ggfe$%d&gd(ggd&gd(ggfe$%d&gd(gge$%d&gd(ggfgdŐdƄ ZܐdǐdȄ ZdhS (  TestDataFrameConstructorsc                 C   sr   t dddd}|jdd}t|}t|d d df |d d df d	}t|| |snt|jjdksnt	d S )
N
2016-01-01   
US/Pacificperiodstz      r      r   r/   )
r   _datareshaper   tmassert_frame_equallen_mgrblocksAssertionError)selfusing_array_managerdtiZdtadfexpected r>   W/home/mars/bis/venv/lib/python3.8/site-packages/pandas/tests/frame/test_constructors.py&test_constructor_from_2d_datetimearrayI   s    &z@TestDataFrameConstructors.test_constructor_from_2d_datetimearrayc                 C   sl   t dd}td|idgd}td|gi}t|| t|dgd}t|gdgd}t|| d S )Nz2019-11-03 01:00:00-0700zAmerica/Los_Angelesdtr   indexr/   )rA   value)r   Z
tz_convertr   r3   r4   )r9   rA   r<   r=   r>   r>   r?   )test_constructor_dict_with_tzaware_scalarT   s    zCTestDataFrameConstructors.test_constructor_dict_with_tzaware_scalarc              	   C   s   t dt jgddgg}tt t|dd}W 5 Q R X |jj|jksLt	t
|jd s^t	tt t|d ddd}W 5 Q R X |jd }t|| d S )	Nr/   r.   r-   i8dtyper0   r   rH   name)nparraynanr3   assert_produces_warningFutureWarningr   valuesrH   r8   r   ilocr   assert_series_equal)r9   arrr<   serr=   r>   r>   r?   -test_construct_ndarray_with_nas_and_int_dtypea   s    
zGTestDataFrameConstructors.test_construct_ndarray_with_nas_and_int_dtypec                 C   s0   t t t g}|d jtdks,td S )Nr   M8[ns])r   r   nowrH   rK   r8   r9   r<   r>   r>   r?   %test_construct_from_list_of_datetimeso   s    z?TestDataFrameConstructors.test_construct_from_list_of_datetimesc                 C   s\   t ddgdd}|d}ttj| dddd}|j|jksDtt|}t	|| d S )Nz2013-1-1 13:00z2013-1-2 14:00BrJ   r)   objectrG   )
r   tz_localizer   rK   rL   tolistrH   r8   r3   rR   )r9   Znaiveidxr=   resultr>   r>   r?   +test_constructor_from_tzaware_datetimeindexs   s    
zETestDataFrameConstructors.test_constructor_from_tzaware_datetimeindexc              	   C   sh   t dd}t j|gtd}|tkr0|dd}dddg}tjt	|d ||d	d W 5 Q R X d S )
NNaTnsrG   r/   |z+Could not convert object to NumPy timedeltaz=Invalid type for timedelta scalar: <class 'numpy.datetime64'>matchm8[ns])
rK   
datetime64rL   r\   r   r2   joinpytestraises
ValueError)r9   frame_or_seriesZnatrS   msgr>   r>   r?   ,test_array_of_dt64_nat_with_td64dtype_raises   s    zFTestDataFrameConstructors.test_array_of_dt64_nat_with_td64dtype_raiseskindmMc                 C   s|  |dkrd}t }nd}t}tjddd|dd}|tkrN|d d d	f }||td}|jj	d	 j
tksptt|jj	d	  d	 |st|||td}|jj	d	 j
tkstt|jj	d	  d	 |st|||ttd}|jj	d	 j
tkstt|jj	d	  d	 |st|tkrxd
d |D }||td}|jj	d	 j
tksXtt|jj	d	  d	 |sxtd S )Nrr   rV   rg   r(   rF   rG   r-   r.   r   c                 S   s   g | ]}t |qS r>   r   .0xr>   r>   r?   
<listcomp>   s     zXTestDataFrameConstructors.test_datetimelike_values_with_object_dtype.<locals>.<listcomp>)r   r   rK   arangeviewr2   r   r\   r6   arraysrH   r8   
isinstanceravelr   r   )r9   rp   rm   rH   Zscalar_typerS   objZsersr>   r>   r?   *test_datetimelike_values_with_object_dtype   s,     
zDTestDataFrameConstructors.test_datetimelike_values_with_object_dtypec                 C   st   t tddd}t tddd}t|dgd}tg dgd}t|| t|dgd}tg dgd}t|| d S )N   r/   r[   r   columns)r   ranger   r3   r4   )r9   rv   yr`   r=   r>   r>   r?   )test_series_with_name_not_matching_column   s    zCTestDataFrameConstructors.test_series_with_name_not_matching_columnconstructorc                   C   s   t  S Nr   r>   r>   r>   r?   <lambda>       z"TestDataFrameConstructors.<lambda>c                   C   s   t d S r   r   r>   r>   r>   r?   r      r   c                   C   s   t i S r   r   r>   r>   r>   r?   r      r   c                   C   s   t dS )Nr>   r   r>   r>   r>   r?   r      r   c                   C   s   t g S r   r   r>   r>   r>   r?   r      r   c                   C   s   t dd dD S )Nc                 s   s   | ]
}|V  qd S r   r>   ru   _r>   r>   r?   	<genexpr>   s     5TestDataFrameConstructors.<lambda>.<locals>.<genexpr>r>   r   r>   r>   r>   r?   r      r   c                   C   s   t tdS Nr   r   r   r>   r>   r>   r?   r      r   c                   C   s
   t d dS Ndatar   r>   r>   r>   r?   r      r   c                   C   s
   t i dS r   r   r>   r>   r>   r?   r      r   c                   C   s
   t ddS )Nr>   r   r   r>   r>   r>   r?   r      r   c                   C   s
   t g dS r   r   r>   r>   r>   r?   r      r   c                   C   s   t dd dD dS )Nc                 s   s   | ]
}|V  qd S r   r>   r   r>   r>   r?   r      s     r   r>   r   r   r>   r>   r>   r?   r      r   c                   C   s   t tddS )Nr   r   r   r>   r>   r>   r?   r      r   c                 C   s@   t  }| }t|jdkstt|jdks0tt|| d S r   )r   r5   rC   r8   r   r3   r4   )r9   r   r=   r`   r>   r>   r?   test_empty_constructor   s
    z0TestDataFrameConstructors.test_empty_constructorz)emptylike,expected_index,expected_columnsr/   r   r.   c                 c   s   | ]
}|V  qd S r   r>   r   r>   r>   r?   r      s     z#TestDataFrameConstructors.<genexpr>r>   c                 C   s$   t ||d}t |}t|| d S )NrC   r   r   r3   r4   )r9   Z	emptylikeZexpected_indexZexpected_columnsr=   r`   r>   r>   r?   test_emptylike_constructor   s    	z4TestDataFrameConstructors.test_emptylike_constructorc                 C   s8   t  \}}t||d}t|}|d jtjks4td S )NrB   foo)r3   ZgetMixedTypeDictr   rH   rK   object_r8   )r9   Zfloat_string_framerC   r   Zindexed_frameZunindexed_framer>   r>   r?   test_constructor_mixed   s    z0TestDataFrameConstructors.test_constructor_mixedc              	   C   s   d}t jt|d tddddgitjd}W 5 Q R X |d jtksHttt	d}t	d
 |d< d	}tjt|d t	d|d
< W 5 Q R X t	d
 |d< d S )NBeither all columns will be cast to that dtype, or a TypeError willre   abcrG   )   r.   r   z5Expected a 1D array, got an array with shape \(4, 2\)testZfoo2)r3   rN   rO   r   rK   r!   rH   r\   r8   onesr^   rj   rk   rl   )r9   rn   r   r<   r>   r>   r?   test_constructor_cast_failure   s    "z7TestDataFrameConstructors.test_constructor_cast_failurec                 C   sD   t dgdgdgd}t |tdd}d|d< |d d	 dks@td S )
N      ?       @      @)col1col2col3T)rH   copyg      i@r   r   )r   floatr8   )r9   Zorig_dfZnew_dfr>   r>   r?   test_constructor_dtype_copy  s    z5TestDataFrameConstructors.test_constructor_dtype_copyc                 C   sB   t ddgg}t ||d jd}d|d d< |jd dks>td S )Nr/   r.   r   rG   c   r   r   )r   rH   rP   r8   )r9   r<   should_be_viewr>   r>   r?   ,test_constructor_dtype_nocast_view_dataframe
  s    zFTestDataFrameConstructors.test_constructor_dtype_nocast_view_dataframec                 C   s|   t ddgddggdd}|sPt |j|d jd}d|d d< |jd	 dksxtn(t |j|d jd}|jjd jjsxtd S )
Nr/   r.   r-   r   r%   rG   r   a   r   )r   rP   rH   r8   r6   rz   flagsc_contiguous)r9   r:   r<   r   df2r>   r>   r?   +test_constructor_dtype_nocast_view_2d_array  s    zETestDataFrameConstructors.test_constructor_dtype_nocast_view_2d_arrayc                 C   s0   t jddgdd}t|}t |j|s,td S )Nr   r   r\   rG   rK   rL   r   shares_memoryrP   r8   r9   rS   r<   r>   r>   r?   "test_1d_object_array_does_not_copy  s    z<TestDataFrameConstructors.test_1d_object_array_does_not_copyc                 C   s8   t jddgddggdd}t|}t |j|s4td S )Nr   r   r   dr\   rG   r   r   r>   r>   r?   "test_2d_object_array_does_not_copy#  s    z<TestDataFrameConstructors.test_2d_object_array_does_not_copyc                 C   s@   t ddgd dggtd}|jd d ks*t|jd dks<td S )Nr/   2r   rG   r/   r   r0   )r   r\   locr8   rX   r>   r>   r?    test_constructor_dtype_list_data*  s    z:TestDataFrameConstructors.test_constructor_dtype_list_datazNumPy change.)reasonc              	   C   s   t  }td}tjtdd t |g W 5 Q R X tjtdd t |g W 5 Q R X t dddgi}tjtdd t ||g W 5 Q R X d S )Nr   zshape=\(1, 0, 0\)re   Ar/   r.   zshape=\(2, 2, 1\))r   rK   emptyrj   rk   rl   )r9   r   r   r>   r>   r?   "test_constructor_list_of_2d_raises/  s    
z<TestDataFrameConstructors.test_constructor_list_of_2d_raisesc                 C   sR   ddd}ddd}|d}|| |ddddd	}|| |d
}|| d S )Nc                 S   s   | dkrt }dd |D }n| dkr6t}dd |D }t||D ]\}}|j|ks@tq@|d krdi }|dd t||D  t|S )Nintc                 S   s"   g | ]}t jt jd |dqS )
   rG   )rK   rL   randomrandru   r   r>   r>   r?   rw   C  s     zjTestDataFrameConstructors.test_constructor_mixed_dtypes.<locals>._make_mixed_dtypes_df.<locals>.<listcomp>r   c                 S   s&   g | ]}t jt jjd d d|dqS )r   )sizerG   )rK   rL   r   randintr   r>   r>   r?   rw   F  s    c                 S   s   i | ]\}}||qS r>   r>   )ru   r   r   r>   r>   r?   
<dictcomp>N  s      zjTestDataFrameConstructors.test_constructor_mixed_dtypes.<locals>._make_mixed_dtypes_df.<locals>.<dictcomp>)MIXED_INT_DTYPESMIXED_FLOAT_DTYPESziprH   r8   updater   )typaddtypesrz   r   r   r>   r>   r?   _make_mixed_dtypes_df?  s    zVTestDataFrameConstructors.test_constructor_mixed_dtypes.<locals>._make_mixed_dtypes_dfc                 S   s8   |d krt t }|D ]}|| kr| j| |kstqd S r   )r   r   r   r8   )r<   r   r   r>   r>   r?   _check_mixed_dtypesQ  s
    zTTestDataFrameConstructors.test_constructor_mixed_dtypes.<locals>._check_mixed_dtypesr   r/   r   barr   rZ   Cr   )N)Nr>   )r9   r   r   r<   r>   r>   r?   test_constructor_mixed_dtypes>  s    

z7TestDataFrameConstructors.test_constructor_mixed_dtypesc                 C   s^   t jdt j}t jdt j}t||d}|j|jjksHt	|j|j
jksZt	d S )Nr   r   r   )rK   r   r   astypeZ	complex64
complex128r   rH   r   r8   r   r9   r   r   r<   r>   r>   r?   test_constructor_complex_dtypesd  s
    z9TestDataFrameConstructors.test_constructor_complex_dtypesc                 C   sx   t ddd gi|d}| }t dddgi}t|| |jd d ksJtt ddtjgi|d}t|jd sttd S )Nr   rv   rG   FTr   )	r   r   r3   r4   rQ   r8   rK   rM   isnan)r9   string_dtyper<   r`   r=   r>   r>   r?   $test_constructor_dtype_str_na_valuesm  s    z>TestDataFrameConstructors.test_constructor_dtype_str_na_valuesc           	      C   s   |j dd}t|jjd d d |j_|j}t|}t|jt	|jj t||d}t|jt	|jj t|j| t
t|d d d }t||ddgd}t||djddgd}t|| d S )NFrB   r   rZ   r   r   )Z
to_recordslistrH   namesrC   r   r3   assert_index_equalr   r   rK   rx   r5   reindexr4   )	r9   float_frameZrecrC   r<   r   rngdf3r=   r>   r>   r?   test_constructor_recx  s    z.TestDataFrameConstructors.test_constructor_recc                 C   s8   t tjdtdtjdtdd}|jjtjks4td S )Nr   rG   r0   )	r   rK   r   boolzerosrP   rH   bool_r8   rX   r>   r>   r?   test_constructor_bool  s    "z/TestDataFrameConstructors.test_constructor_boolc                 C   s   t jdd tddD t jd}td|i}|d jt jks@tddd	d
dg}ddg}t jt|f|d}||d d < t|}|d jt jkstd S )Nc                 S   s   g | ]}d | qS )            r>   ru   ir>   r>   r?   rw     s     zMTestDataFrameConstructors.test_constructor_overflow_int64.<locals>.<listcomp>r/   r   rG   r   )l   Y2<   )l   FJLW]L*    )l   M:M^ -   )l   8Ma i  )l   r*R3_PvN r   )uidu8)Zscorer   r   )	rK   rL   r   r#   r   rH   r8   r   r5   )r9   rP   r`   Zdata_scoresrH   r   Z	df_crawlsr>   r>   r?   test_constructor_overflow_int64  s     z9TestDataFrameConstructors.test_constructor_overflow_int64rP   r   rG   A   l           l        l        c                 C   s:   |d }t |}|d jtks"t|d d |ks6td S r   )r   rH   r\   r8   )r9   rP   rD   r`   r>   r>   r?   test_constructor_int_overflow  s    z7TestDataFrameConstructors.test_constructor_int_overflowc                 C   s`   dd l }d}tt|}|| dd |D }ttt|dgg| }|t|jks\td S )Nr   d   c                 S   s   g | ]}d |dqS )r   r   r>   r   r>   r>   r?   rw     s     zJTestDataFrameConstructors.test_constructor_ordereddict.<locals>.<listcomp>)	r   r   r   shuffler   r   r   r   r8   )r9   r   Znitemsnumsr=   r<   r>   r>   r?   test_constructor_ordereddict  s    
z6TestDataFrameConstructors.test_constructor_ordereddictc              	   C   s2  t jdd}t jdddd  }t||d}t|dks>tt|dksNtt |d |d tt	tj
gd |jg|jdd}t ||d  t||ddd	d
gd}t|t|kstd|kstt|d	  sttt dkstd}tjt|d  tddddddgd W 5 Q R X d S )N   Znperr   r   r      r   r   rC   rJ   r   Zcol4r   r   <Mixing dicts with non-Series may lead to ambiguous ordering.re   r   r   r   r   r   rZ   )r3   makeTimeSeriesr   r5   r8   rR   renamer   rK   ZconcatenaterM   rP   rC   r   allrj   rk   rl   )r9   datetime_seriesdatetime_series_shortframeexprn   r>   r>   r?   test_constructor_dict  s.    z/TestDataFrameConstructors.test_constructor_dictc                 C   s,   t ddddi}t|jtddg d S )Nr   r/   r.   1r   r  r   )r   r3   r   rC   r   r9   r  r>   r>   r?   test_constructor_dict_length1  s    z7TestDataFrameConstructors.test_constructor_dict_length1c                 C   s,   t dddg}ti |d}|j|ks(td S )Nr   r/   r.   rB   )r   r   rC   r8   r9   r_   r  r>   r>   r?    test_constructor_dict_with_index  s    z:TestDataFrameConstructors.test_constructor_dict_with_indexc                 C   sN   t dddg}ti ||d}|j|ks*t|j|ks8tt|jdksJtd S )Nr   r/   r.   r   r-   )r   r   rC   r8   r   r5   _seriesr  r>   r>   r?   ,test_constructor_dict_with_index_and_columns  s
    zFTestDataFrameConstructors.test_constructor_dict_with_index_and_columnsc                 C   s0   t g g dddgd}tj|jtddd d S )Nr  r   rZ   r   r   Texact)r   r3   r   rC   r   r  r>   r>   r?   $test_constructor_dict_of_empty_lists  s    z>TestDataFrameConstructors.test_constructor_dict_of_empty_listsc                 C   s^   t dd idgd}t dd gidgd}|ddd ks:t|ddd ksNtt|| d S )Nr   r   rB   )r   
_get_valuer8   r3   r4   )r9   Z
frame_noneZframe_none_listr>   r>   r?   test_constructor_dict_with_none  s
    z9TestDataFrameConstructors.test_constructor_dict_with_nonec              	   C   sZ   d}t jt|d tddi W 5 Q R X t jt|d tddidgd W 5 Q R X d S )N2If using all scalar values, you must pass an indexre   r   gffffff?r   rj   rk   rl   r   )r9   rn   r>   r>   r?   test_constructor_dict_errors  s
    z6TestDataFrameConstructors.test_constructor_dict_errorsscalarNDc                 C   s.   t d|idgd}t dgd}t|| d S )Nr   r   r   r   )r9   r  r`   r=   r>   r>   r?   %test_constructor_invalid_items_unused  s    z?TestDataFrameConstructors.test_constructor_invalid_items_unusedrD   rM   c                    s   d|dg d|gddgddgddgg fdd	t dD }t|djddd
}ttjddddd d}t|| t|djddd
}t|| t| d}t|| d S )Nr/   r-   r   r   r   r.   r   c                    s"   i | ]} | t | d qS rB   rs   ru   r   colsr_   rP   r>   r?   r     s      zKTestDataFrameConstructors.test_constructor_dict_nan_key.<locals>.<dictcomp>Zaxisr(   r%   rG   r   rB   )r   r   sort_valuesrK   rx   r2   r3   r4   r9   rD   r   r`   r=   r>   r  r?   test_constructor_dict_nan_key  s    
  z7TestDataFrameConstructors.test_constructor_dict_nan_keyc                    s   t d|dfd|fg t d|f|dfgddgdd	gdd
gg fddtdD }t|djd|fdd}ttjddddd d}t|| t|djd|fdd}t|| t| d}t|| d S )N)            r   r.   r   r-   r/   r   r   c                    s"   i | ]} | t | d qS r  rs   r  r  r>   r?   r   2  s      zQTestDataFrameConstructors.test_constructor_dict_nan_tuple_key.<locals>.<dictcomp>r!  r(   r%   rG   r   rB   )	r   r   r   r"  rK   rx   r2   r3   r4   r#  r>   r  r?   #test_constructor_dict_nan_tuple_key,  s      z=TestDataFrameConstructors.test_constructor_dict_nan_tuple_keyc                 C   sL   t jdd}t jdd}||d}t|d}t|tdd}t || d S )Nr   r   r   r   r   r   bar   r   )r3   r  r   r   r4   )r9   r  r  r   r  r=   r>   r>   r?   %test_constructor_dict_order_insertion?  s    

z?TestDataFrameConstructors.test_constructor_dict_order_insertionc                 C   sR   t tjddgdddgitjdgd}t ddgddggtjdgd}t|| d S )Nr/   r.   r-   r   )r   rK   rM   r3   r4   r9   r`   r=   r>   r>   r?   )test_constructor_dict_nan_key_and_columnsJ  s    $zCTestDataFrameConstructors.test_constructor_dict_nan_key_and_columnsc                 C   sp   dddg}t |}t||d}t|j  s6tdddg}t |}t||d}t|j  sltd S )Nr.   r-   )r-   r-   r   )r   from_tuplesr   r   rP   r|   r  r8   )r9   tuplesmir<   r>   r>   r?   test_constructor_multi_indexP  s    



z6TestDataFrameConstructors.test_constructor_multi_indexc                 C   s   t dggdggddgd}t ddgtddgddtdggdggdd}t|| t dggdggddggd}t ddgtddggddggdtdggdggdd}t|| d S )	Nr/   r.   r   rC   r%   rG   r   )Zlevelscodesr   )r   r    r   r3   r4   r9   r<   r=   r>   r>   r?   test_constructor_2d_index]  s    z3TestDataFrameConstructors.test_constructor_2d_indexc              	   C   sV  d}t jt|d ttdtdd W 5 Q R X d}t jt|d  tddd	ddd
gd W 5 Q R X d}t jt|d. ttdddddgt	dddd W 5 Q R X t
dddgg}d}t jt|d tdgtdd|d W 5 Q R X t
dddg}d}t jt|d tdgtdd|d W 5 Q R X t jtdd" ttdddd gd!gd W 5 Q R X d"}t jt|d& ttjd#dddd gd!gd W 5 Q R X d$}t jt|d& ttjd#dddgd!d#gd W 5 Q R X d%}t jt|d  ttd&td&d'fd(d W 5 Q R X d)}t jt|d td*d+d	 W 5 Q R X d S ),Nz)Empty data passed with indices specified.re   r   r   r   r   r   r   r   r   r  z:Shape of passed values is \(4, 3\), indices imply \(3, 3\)   )r   r-   r   r   baz
2000-01-01r-   r+   r5  r   r   r(   z:Shape of passed values is \(1, 3\), indices imply \(1, 4\)rC   r   r   z:Shape of passed values is \(3, 1\), indices imply \(1, 4\)Must pass 2-d inputr-   r-   r-   r   rZ   r   r/   :Shape of passed values is \(2, 3\), indices imply \(1, 3\)r.   :Shape of passed values is \(2, 3\), indices imply \(2, 2\)z,2 columns passed, passed data had 10 columnsr      )r   Ztwosr  FT)rj   rk   rl   r   rK   r   r   rx   r2   r   rL   r   r   r   r   )r9   rn   rS   r>   r>   r?   test_constructor_error_msgsp  sF     $
  &**$z5TestDataFrameConstructors.test_constructor_error_msgsc                 C   s|   |dd t dD |dd t dD d}t|}tdd | D }t|| || }t|}t|| d S )Nc                 s   s   | ]}|d | fV  qdS )g      $@Nr>   rt   r>   r>   r?   r     s     zKTestDataFrameConstructors.test_constructor_subclass_dict.<locals>.<genexpr>r   c                 s   s   | ]}|d | fV  qdS )g      4@Nr>   rt   r>   r>   r?   r     s     r   c                 S   s   i | ]\}}|t | qS r>   )dictitems)ru   colvalr>   r>   r?   r     s      zLTestDataFrameConstructors.test_constructor_subclass_dict.<locals>.<dictcomp>)r   r   rE  r3   r4   )r9   Zdict_subclassr   r<   Zrefdfr>   r>   r?   test_constructor_subclass_dict  s    z8TestDataFrameConstructors.test_constructor_subclass_dictc           	      C   sx   ddl m} i }tj|d d d< | D ]&\}}|t}||  |||< q*t|}|j	|j
d}t|| d S )Nr   )defaultdictrZ   r   rB   )collectionsrI  rK   rM   rE  rD  r   to_dictr   r   rC   r3   r4   )	r9   r   rI  r   kvdctr  r=   r>   r>   r?   test_constructor_defaultdict  s    
z6TestDataFrameConstructors.test_constructor_defaultdictc                 C   sL   t ddddgg}tdgdgdgdgddddd	gd
}t|j| d S )N      @r   r   r   )r   r   r   r   r   r   r   r   r   )rK   rL   r   r3   assert_numpy_array_equalrP   r9   r=   r<   r>   r>   r?   test_constructor_dict_block  s    
z5TestDataFrameConstructors.test_constructor_dict_blockc                 C   s   dddddddd}t |td	}t|d
ks4t|d jtjksHt|d jtjks\tt |}t|d
kstt|d jtjkst|d jtjkstd S )Nr/   r.   r
  r  r   3)r  r   rT  r  rG   r-   rZ   r   )r   r   r5   r8   rH   rK   r!   r   )r9   	test_datar  r>   r>   r?   test_constructor_dict_cast  s    z4TestDataFrameConstructors.test_constructor_dict_castc              	   C   s   t ttdtdt ttdtjdd}d}tjt	|d t
|td}W 5 Q R X t|dksnt|d jtjkst|d jtjkstd S )	NrB     r  r   re   rG   r   rZ   )rD  r   r   r3   ZmakeStringIndexrK   r   randnrN   rO   r   r   r5   r8   rH   r   r!   )r9   rU  rn   r  r>   r>   r?   test_constructor_dict_cast2  s    z5TestDataFrameConstructors.test_constructor_dict_cast2c                 C   s2   ddt jdi}t|}t|d d ts.td S )NZCol1zA String)ZRow1Row2rZ  )rK   rM   r   r{   r   r8   )r9   r   r<   r>   r>   r?   !test_constructor_dict_dont_upcast  s    z;TestDataFrameConstructors.test_constructor_dict_dont_upcastc                 C   s<   t ddgddggddgddgd}t|d d ts8td S )Nr/   r.   r   r   r   )r   r{   r   r8   )r9   dmr>   r>   r?   "test_constructor_dict_dont_upcast2  s    "z<TestDataFrameConstructors.test_constructor_dict_dont_upcast2c                 C   s<   ddd}t |}t dd | D }tj||dd d S )N)r/   r.   r-   )r   r   r(   r   c                 S   s   i | ]\}}|t |qS r>   )r   ru   rL  rM  r>   r>   r?   r     s      zMTestDataFrameConstructors.test_constructor_dict_of_tuples.<locals>.<dictcomp>FZcheck_dtype)r   rE  r3   r4   r9   r   r`   r=   r>   r>   r?   test_constructor_dict_of_tuples  s    
z9TestDataFrameConstructors.test_constructor_dict_of_tuplesc                 C   sF   t dt ddd}t|}tdddgdddgd}t|| d S )	Nr-   r(   r   r   r/   r.   r   r   r   r   r3   r4   r`  r>   r>   r?   test_constructor_dict_of_ranges  s    z9TestDataFrameConstructors.test_constructor_dict_of_rangesc                 C   sL   t tdttdd}t|}tdddgdddgd}t|| d S )Nr-   r   r   r/   r.   )iterr   reversedr   r3   r4   r`  r>   r>   r?   "test_constructor_dict_of_iterators  s    z<TestDataFrameConstructors.test_constructor_dict_of_iteratorsc                 C   s\   dd t dD dd tt dD d}t|}tdddgdddgd}t|| d S )Nc                 s   s   | ]
}|V  qd S r   r>   r   r>   r>   r?   r     s     zPTestDataFrameConstructors.test_constructor_dict_of_generators.<locals>.<genexpr>r-   r   r   r/   r.   )r   re  r   r3   r4   r`  r>   r>   r?   #test_constructor_dict_of_generators  s    *z=TestDataFrameConstructors.test_constructor_dict_of_generatorsc                 C   s  dddddddddd	d
dd}t | }t|}tdd |D tdd |D dj}t|j|_t|| ddddd|d< |	dd|d f tdd |D t
dd |D dddj}t
|jdd|_t|}|j|j|jd}t|| d S )Nr   r/   r.   )r   r   r   jrj  r   r(   r   r         	   ))r   r   r*  )r   r   c                 S   s   g | ]}|d  qS r/   r>   rt   r>   r>   r?   rw     s     zNTestDataFrameConstructors.test_constructor_dict_multiindex.<locals>.<listcomp>c                 S   s   g | ]}|d  qS r   r>   rt   r>   r>   r?   rw     s     rB   g     ^@o   )r   rh  ri  rk  zc                 S   s   g | ]}|d  qS ro  r>   rt   r>   r>   r?   rw     s     c                 S   s   g | ]}|d  qS rp  r>   rt   r>   r>   r?   rw     s     F)Ztupleize_colsr5  )sortedrE  r   r   r1  TrC   r3   r4   insertr   r   r   )r9   r   Z_dr<   r=   r>   r>   r?    test_constructor_dict_multiindex  s2    


  z:TestDataFrameConstructors.test_constructor_dict_multiindexc           	   	      s   ddddg  fdd}|t j}|dd }|t}td	d d d d
d dd d d
d d dd d
d d d dd
gdd  D d}t|}t|}t|}t|| t|| t|| d S )Nz
1984-02-19z
1988-11-06z
1989-12-03z
1990-03-15c                    s    fddt D S )Nc                    s"   i | ]\}}| |d | iqS r.   r>   ru   r   sr   r>   r?   r   +  s      ziTestDataFrameConstructors.test_constructor_dict_datetime64_index.<locals>.create_data.<locals>.<dictcomp>	enumeraterz  Zdates_as_strrz  r?   create_data*  s    zUTestDataFrameConstructors.test_constructor_dict_datetime64_index.<locals>.create_datac                 S   s   t | dS )Nz%Y-%m-%d)r   strptimerv   r>   r>   r?   r   .  r   zRTestDataFrameConstructors.test_constructor_dict_datetime64_index.<locals>.<lambda>r   r   r/   r.   r-   r.   r   r(   c                 S   s   g | ]}t |qS r>   )r   )ru   rA   r>   r>   r?   rw   8  s     zTTestDataFrameConstructors.test_constructor_dict_datetime64_index.<locals>.<listcomp>rB   )rK   rh   r   r   r3   r4   )	r9   r~  Zdata_datetime64Zdata_datetimeZdata_Timestampr=   Zresult_datetime64Zresult_datetimeZresult_Timestampr>   r}  r?   &test_constructor_dict_datetime64_index&  s&    

z@TestDataFrameConstructors.test_constructor_dict_datetime64_indexc           	   	      s   ddddg  fdd}|dd }|d	d }|d
d }t dd d d dd dd d dd d dd dd d d ddgdd  D d}t |}t |}t |}t|| t|| t|| d S )Nr/   r.   r-   r   c                    s    fddt D S )Nc                    s"   i | ]\}}| |d | iqS rw  r>   rx  rz  r>   r?   r   G  s      zjTestDataFrameConstructors.test_constructor_dict_timedelta64_index.<locals>.create_data.<locals>.<dictcomp>r{  rz  Z	td_as_intrz  r?   r~  F  s    zVTestDataFrameConstructors.test_constructor_dict_timedelta64_index.<locals>.create_datac                 S   s   t | dS Nr  )rK   timedelta64r  r>   r>   r?   r   I  r   zSTestDataFrameConstructors.test_constructor_dict_timedelta64_index.<locals>.<lambda>c                 S   s
   t | dS )N)days)r   r  r>   r>   r?   r   J  r   c                 S   s
   t | dS r  r   r  r>   r>   r?   r   K  r   r   r  r(   c                 S   s   g | ]}t |d qS )r  r  )ru   tdr>   r>   r?   rw   T  s     zUTestDataFrameConstructors.test_constructor_dict_timedelta64_index.<locals>.<listcomp>rB   r   )	r9   r~  Zdata_timedelta64Zdata_timedeltaZdata_Timedeltar=   Zresult_timedelta64Zresult_timedeltaZresult_Timedeltar>   r  r?   'test_constructor_dict_timedelta64_indexB  s&    
zATestDataFrameConstructors.test_constructor_dict_timedelta64_indexc                 C   s   t jdddgdd}t jdddgdd}t||d	}|d
 j|jksJt|d j|jks^tt|t |t d	}|d
 j|jkst|d j|jkstd S )Nz2012-01rb   z2012-04rr   freqz
2012-02-01z
2012-03-01r  r   r   r   )pdZPeriodIndexr   rH   r8   r   r\   r^   r   r>   r>   r?   test_constructor_period_dict^  s    "z6TestDataFrameConstructors.test_constructor_period_dictc                 C   sN   |\}}t d|idgd}|d j|ks,tt dgdg|d}t|| d S )Nr   r   rB   r=  )r   rH   r8   r3   r4   )r9   Zea_scalar_and_dtypeZ	ea_scalarZea_dtyper<   r=   r>   r>   r?   &test_constructor_dict_extension_scalark  s
    z@TestDataFrameConstructors.test_constructor_dict_extension_scalarz
data,dtypez2020-01r   )leftrightr%   r  z
2011-01-01
US/Easternr,   c                 C   sl   t ddgddg|d}|d j|ks(t|d j|ks:ttj|gd |d}t ||d}t|| d S )	Nr   r/   r   r   r=  r.   rG   r   )r   rH   r8   r  rL   r3   r4   )r9   r   rH   r<   rS   r=   r>   r>   r?   &test_constructor_extension_scalar_datat  s    z@TestDataFrameConstructors.test_constructor_extension_scalar_datac                 C   s   t jddd}ttjdd|d}i }|jD ](}|jD ]}||||	|i |< q8q.t||d}t
|| i }|jD ](}|jD ]}||||	|i |< qqzt||dj}t
|| d S )Nz1/1/2000r   r<  r   r   rB   )r  period_ranger   rK   r   rX  r   rC   r  
setdefaultr3   r4   rt  )r9   r   r<   r   rF  rowr`   r>   r>   r?   "test_nested_dict_frame_constructor  s    



z<TestDataFrameConstructors.test_nested_dict_frame_constructorc              	   C   s  |dt d}t|dddgddgd}t|jdks6tt|jd	ksHtt|d
dgddd	gd}t|jd	ksttt|jdkst|tjkrd nt}t	
|" t|dddgddgtjd}W 5 Q R X |tjkr|jjtjkstn<|   st|jjtjkstt|j s$td}tjt|d t|dddgdgd W 5 Q R X d}tjt|d t|ddgddgd W 5 Q R X tjtdd  t|ddddgdgd W 5 Q R X t|}t	j|jttddd t	j|jttd	dd t|ddgd}t	j|jttd	dd t|dddgd}t	j|jttddd t|d}t|jdksrtt|d}t|jdkstd S )Nr0  rG   r   rZ   r   r/   r.   r5  r-   )r-   r   rC   rH   r@  re   rA  r>  r?  Tr  rB   r   )r   r-   r   )r-   r   )r   r   r5   rC   r8   r   rK   r   rO   r3   rN   r%   rP   rH   r   r  r!   rj   rk   rl   r   r   r   )r9   r   matr  warnrn   r>   r>   r?   _check_basic_constructor  sN       
  $z2TestDataFrameConstructors._check_basic_constructorc                 C   s8   |  tj tddgddgdgd}t|dks4td S )Nr   r   r   r/   r   r   r.   )r  rK   r   r   r5   r8   r  r>   r>   r?   test_constructor_ndarray  s    z2TestDataFrameConstructors.test_constructor_ndarrayc                 C   s   |  tj tjdtd}d|d< d|d< t|ddd	gd
dgd}d|d d
 ksVtd|d	 d ksjttjdtd}t|ddd	gd
dgd}tt||k std S )Nr0  rG   r   r   r   r/   r.   r   rZ   r   r/   r.   r5  )	r  ma
masked_allr   r   r8   rK   r  asarray)r9   r  r  r>   r>   r?   test_constructor_maskedarray  s    z6TestDataFrameConstructors.test_constructor_maskedarrayc              
   C   s"  t jdtd}t|dddgddgd}t|jdks8tt|jd	ksJtt	t
||k sdtt|dddgddgtjd
}|jjtjkstt |}d|d< d|d< t|dddgddgd}d|d d kstd|d d kstt jddd}t|dddgddgd}t|jdks&tt|jd	ks:tt|j	 sNtd}tjt|dF t 2 tjdtdd t|dddgddgtjd
}W 5 Q R X W 5 Q R X |jjtjkstt |}d|d< d|d< t|dddgddgd}d|d dd kstd|d dd ks*tt jdtd}t|dddgddgd}t|jdksdtt|jd	ksxtt	t
||k stt|dddgddgtd
}|jjtkstt |}d|d< d|d< t|dddgddgd}|d d dkst|d d dkstd S )Nr0  rG   r   rZ   r   r/   r.   r5  r-   r  r   r  rV   z'datetime64\[ns\] values and dtype=int64re   ignorezelementwise comparison failed)categorymessagerF   TF)r  r  r   r   r5   rC   r8   r   rK   r  r  r!   rP   rH   r   r   r3   rN   rO   warningscatch_warningsfilterwarningsDeprecationWarningr%   ry   r   r\   )r9   r  r  Zmat2rn   r>   r>   r?   %test_constructor_maskedarray_nonfloat  sh    

   

z?TestDataFrameConstructors.test_constructor_maskedarray_nonfloatc                 C   s   t jdtd }t|ddgddgd}ttjtjgtjtjgdddgddgtd	}t|| t j	dtd }t|ddgddgd}td
d
gd
d
gdddgddgtd	}t|| d S )N)r.   r.   rG   r   rZ   r/   r.   r5  r  r  r   )
r  r  r   Zharden_maskr   rK   rM   r3   r4   r   )r9   Zmat_hardr`   r=   r>   r>   r?   %test_constructor_maskedarray_hardened<  s$    z?TestDataFrameConstructors.test_constructor_maskedarray_hardenedc              	      s   t jjt jjdddgddgd d  tj tt	 t
 td}W 5 Q R X t
t jdtddd	gd
}t|| t
 fdd jjD td}t|| d S )Nr   )r   <f8)pricer  rG   Fmaskr   r.   r   r  r   c                    s   i | ]}| | qS r>   r>   )ru   rJ   r   r>   r?   r   b  s      zSTestDataFrameConstructors.test_constructor_maskedrecarray_dtype.<locals>.<dictcomp>)rK   r  rL   r   ry   mrecordsZ	mrecarrayr3   rN   rO   r   r   r4   rH   r   )r9   r`   r=   Zaltr>   r   r?   %test_constructor_maskedrecarray_dtypeR  s     z?TestDataFrameConstructors.test_constructor_maskedrecarray_dtypec              
   C   s  t jtjddd}dtddgfdtddgfd	td
dgfg}|d d  D ]*\}}|d| tjj|ddgdf qR|dtj	df |dtjjddgddf t
|dD ]}t| \}}tj||d}dd |D }tt t|}	W 5 Q R X t||d}
||	|
 tt t||d d d d}	W 5 Q R X t||d d d d}
||	|
 tt t|ddgd}	W 5 Q R X t||ddgd}
||	|
 qd S )NT)Zcheck_index_typeZcheck_column_typer         ?r   r   r/   r.   strr   defZmasked1_Fr  r  rw  Zmasked_noner   g      @r-   )r   c                 S   s(   i | ] \}}|t |d r | n|qS )filled)hasattrr  r^  r>   r>   r?   r     s      zHTestDataFrameConstructors.test_constructor_mrecarray.<locals>.<dictcomp>r   r   rB   r5  )	functoolspartialr3   r4   rK   rL   appendr  Zmasked_arrayr  	itertoolscombinationsr   r  Z
fromarraysrN   rO   r   )r9   Zassert_fr_equalrz   rJ   rS   combr   r   Zmrecsr`   r=   r>   r>   r?   test_constructor_mrecarraye  s>      
 
z4TestDataFrameConstructors.test_constructor_mrecarrayc                 C   s   t g d}|jjdkstd S )NrB   r   )r   rP   shaper8   rX   r>   r>   r?   test_constructor_corner_shape  s    
z7TestDataFrameConstructors.test_constructor_corner_shapez%data, index, columns, dtype, expectedr   r   r   r!   r   r   r   c                 C   s"   t ||||}|jj|kstd S r   )r   rP   rH   r8   )r9   r   rC   r   rH   r=   r<   r>   r>   r?   test_constructor_dtype  s    z0TestDataFrameConstructors.test_constructor_dtypezdata,input_dtype,expected_dtypeTFbooleanr   r   ZFloat64ZInt64r   stringc                 C   s(   t d|i|d}|d j| ks$td S )Nr   rG   )r   rH   r8   )r9   r   Zinput_dtypeZexpected_dtyper<   r>   r>   r?   0test_constructor_dtype_nullable_extension_arrays  s    zJTestDataFrameConstructors.test_constructor_dtype_nullable_extension_arraysc                 C   s   dddddd}t |tdd}|d	 jtjks6t|d
 jtjksJt|d jtjks^t|d jtjksrt|d jtj	kstd S )Nr/   Tr   y              @r   )r   r   r   complexr\   r   rB   r   r   r   r  r\   )
r   rK   rx   rH   r%   r8   r   r!   r   r   )r9   r   r<   r>   r>   r?   !test_constructor_scalar_inference  s    z;TestDataFrameConstructors.test_constructor_scalar_inferencec              	   C   sh   t tjddd}t |d jdgd d}t|| tjt	dd t ddd W 5 Q R X d S )Nr   Tr   r   zmust pass an indexre   F)
r   rK   r   rX  rP   r3   r4   rj   rk   rl   )r9   r<   r  r>   r>   r?   #test_constructor_arrays_and_scalars  s
    z=TestDataFrameConstructors.test_constructor_arrays_and_scalarsc                 C   s8   t |}t|| t |tjd}|jjtjks4td S NrG   )r   r3   r4   rK   r%   rP   rH   r8   )r9   r   r<   Z	df_castedr>   r>   r?   test_constructor_DataFrame  s    z4TestDataFrameConstructors.test_constructor_DataFramec                 C   s*   t t  dd}t g dd}t|| d S )Nr\   rG   r   )r9   actualr=   r>   r>   r?    test_constructor_empty_dataframe  s    z:TestDataFrameConstructors.test_constructor_empty_dataframec              	   C   s  t jd}t|dgt dd}|jjdks2tt jd}t|}|jjdksVt|jjdksfttddgt dd}|jjdksttddgd}|jjd	ksttt dd
}|jjdkstt j	ddgt
ddd}d}tjt|d t|ddgdgtd W 5 Q R X tt|j}t|| tt jdtdt jdt jddt dd
}t|jdkstt|jjt jkstd S )Nr   r   r5  r.   r   rZ   )r   r.   r   r   r.   rB   )r   r   r   r   rG   r/   z(could not convert string to float: 'foo're   rC   r   rH   r  )rK   r   rX  r   rx   rP   ndimr8   r  rL   r\   r2   rj   rk   rl   r   r  r3   r4   r   r   r!   r5   r   rH   )r9   r   rS   r\  r  rn   r>   r>   r?   test_constructor_more  s2     z/TestDataFrameConstructors.test_constructor_morec                 C   sx   t g g d}t g d}t|| t g ddgd}t i ddgd}t|| dd }t | ddgd}t|| d S )NrB   r   rZ   r   c                   s   s   d S r   r>   r>   r>   r>   r?   	empty_gen  s    zHTestDataFrameConstructors.test_constructor_empty_list.<locals>.empty_genr   )r9   r<   r=   r  r>   r>   r?   test_constructor_empty_list  s    
z5TestDataFrameConstructors.test_constructor_empty_listc                 C   s|   t ddgddggddgd}t|d s,t|d jtjks@tt dtd	i}d
d td	D }t |}t	|| d S )Nr/   r   r.   r   numr  r,  r   r   c                 S   s   g | ]}t |qS r>   rK   rL   rt   r>   r>   r?   rw     s     zLTestDataFrameConstructors.test_constructor_list_of_lists.<locals>.<listcomp>)
r   r   r8   rH   rK   r   rx   r   r3   r4   )r9   r<   r=   r   r`   r>   r>   r?   test_constructor_list_of_lists  s    z8TestDataFrameConstructors.test_constructor_list_of_listsc                 C   sx   t ddg}tjd d gtd}||d< |d |d< t|}tt|}t|| |jdksbt	t
|d j| d S )Nr/   r.   rG   r   )r.   r/   )r   rK   rL   r\   r   r  r3   r4   r  r8   rQ  rP   )r9   rT   rS   r<   r=   r>   r>   r?   .test_nested_pandasarray_matches_nested_ndarray  s    zHTestDataFrameConstructors.test_nested_pandasarray_matches_nested_ndarrayc                 C   sj   t dt dg}tddddgdddd	gg|d
}t|}tddddgdddd	gg|d
}t|| d S )NabcdZcdefr/   r.   r-   r   r   r(   rl  r   )r   r   r   Zfrom_arraysr3   r4   )r9   rz   r`   r3  r=   r>   r>   r?   2test_constructor_list_like_data_nested_list_column)  s
     
 zLTestDataFrameConstructors.test_constructor_list_like_data_nested_list_columnc              	   C   sR   t dt dg}d}tjt|d& tddddgdd	d
dgg|d W 5 Q R X d S )Nr   cdez#3 columns passed, passed data had 4re   r/   r.   r-   r   r   r(   rl  r   r   rj   rk   rl   r   r9   rz   rn   r>   r>   r?   0test_constructor_wrong_length_nested_list_column3  s    zJTestDataFrameConstructors.test_constructor_wrong_length_nested_list_columnc              	   C   sR   t dt dg}d}tjt|d& tddddgdd	d
dgg|d W 5 Q R X d S )Nr  r  zall arrays must be same lengthre   r/   r.   r-   r   r   r(   rl  r   r  r  r>   r>   r?   2test_constructor_unequal_length_nested_list_column;  s    zLTestDataFrameConstructors.test_constructor_unequal_length_nested_list_columnr   
2021-01-01rv   c                 C   s<   t |dddgdgd}t dtdgd i}t|| d S )Nr   r/   r.   rv   r   r  r-   )r   r   r3   r4   r`  r>   r>   r?   &test_constructor_one_element_data_listD  s    z@TestDataFrameConstructors.test_constructor_one_element_data_listc                 C   sj   G dd dt j}|ddg|ddgg}ddg}t||d	}tddgddgg|d	}tj||d
d d S )Nc                   @   s$   e Zd Zdd Zdd Zdd ZdS )zPTestDataFrameConstructors.test_constructor_sequence_like.<locals>.DummyContainerc                 S   s
   || _ d S r   )_lst)r9   lstr>   r>   r?   __init__X  s    zYTestDataFrameConstructors.test_constructor_sequence_like.<locals>.DummyContainer.__init__c                 S   s   | j |S r   )r  __getitem__r9   nr>   r>   r?   r  [  s    z\TestDataFrameConstructors.test_constructor_sequence_like.<locals>.DummyContainer.__getitem__c                 S   s
   | j  S r   )r  __len__r  r>   r>   r?   r  ^  s    zXTestDataFrameConstructors.test_constructor_sequence_like.<locals>.DummyContainer.__len__N)__name__
__module____qualname__r  r  r  r>   r>   r>   r?   DummyContainerW  s   r  r/   r   r.   r   r  r  r   Fr_  )r   Sequencer   r3   r4   )r9   r  Zlst_containersr   r`   r=   r>   r>   r?   test_constructor_sequence_likeS  s    
z8TestDataFrameConstructors.test_constructor_sequence_likec                 C   s   dd l }td| dtdi}tdttdi}tj||dd tttdttdg}t| dtd| dtdg}tj||dd d S )Nr   r   r   r   Fr_  )rL   r   r   r   r3   r4   )r9   rL   r`   r=   r>   r>   r?   test_constructor_stdlib_arrayg  s    $z7TestDataFrameConstructors.test_constructor_stdlib_arrayc                 C   s,   t td}t ttd}t|| d S Nr   r   r   r   r3   r4   r.  r>   r>   r?   test_constructor_ranget  s    z0TestDataFrameConstructors.test_constructor_rangec                 C   s@   t tdtdg}t ttdttdg}t|| d S r  r  r.  r>   r>   r?   test_constructor_list_of_rangesz  s    z9TestDataFrameConstructors.test_constructor_list_of_rangesc                 C   s<   G dd d}t dddggd }t | }t|| d S )Nc                   @   s   e Zd Zdd ZdS )zATestDataFrameConstructors.test_constructor_iterable.<locals>.Iterc                 s   s   t dD ]}dddgV  qd S )Nr   r/   r.   r-   )r   )r9   r   r>   r>   r?   __iter__  s    zJTestDataFrameConstructors.test_constructor_iterable.<locals>.Iter.__iter__N)r  r  r  r  r>   r>   r>   r?   Iter  s   r  r/   r.   r-   r   r   )r9   r  r=   r`   r>   r>   r?   test_constructor_iterable  s    
z3TestDataFrameConstructors.test_constructor_iterablec                 C   s0   t ttd}t ttd}t|| d S r  r   rd  r   r   r3   r4   r.  r>   r>   r?   test_constructor_iterator  s    z3TestDataFrameConstructors.test_constructor_iteratorc                 C   sH   t ttdttdg}t ttdttdg}t|| d S r  r  r.  r>   r>   r?   "test_constructor_list_of_iterators  s    z<TestDataFrameConstructors.test_constructor_list_of_iteratorsc                 C   s   dd t dD }dd t dD }ttt dtt dg}t||g}t|| dd t dD }t|}tt ddd}tj||dd	 d S )
Nc                 s   s   | ]
}|V  qd S r   r>   r   r>   r>   r?   r     s     zGTestDataFrameConstructors.test_constructor_generator.<locals>.<genexpr>r   c                 s   s   | ]
}|V  qd S r   r>   r   r>   r>   r?   r     s     c                 s   s   | ]}|d gV  qdS )r   Nr>   r   r>   r>   r?   r     s     r   r0   Fr_  )r   r   r   r3   r4   )r9   Zgen1Zgen2r=   r`   genr>   r>   r?   test_constructor_generator  s    z4TestDataFrameConstructors.test_constructor_generatorc                 C   s&   t i g}t dgd}t|| d S )Nr   rB   r   r.  r>   r>   r?   test_constructor_list_of_dicts  s    
z8TestDataFrameConstructors.test_constructor_list_of_dicts	dict_typec                 C   st   t ddggddgd}| }dg|d< dg|d< t |}t|| | }d|d< d|d< t |g}t|| d S )Nr.   r/   r   r   r   r   )r9   r  r=   r   r`   r>   r>   r?   ,test_constructor_ordered_dict_preserve_order  s    


zFTestDataFrameConstructors.test_constructor_ordered_dict_preserve_orderc                 C   s   | }d|d< d|d< | }d|d< d|d< ddd}t ddgddggddgd}t ||g}t|| t ddgddgddggddgd}t |||g}t|| d S )Nr.   r   r/   r   r*  r   r   )r9   r  Zrow_oneZrow_twoZ	row_threer=   r`   r>   r>   r?   0test_constructor_ordered_dict_conflicting_orders  s    
"zJTestDataFrameConstructors.test_constructor_ordered_dict_conflicting_ordersc                 C   s^   dd t dD }t|}tdddgdddgdddgddd	d
gdddgd}t|| d S )Nc                 S   s$   g | ]}t |d ddgt|dqS )r   r   r   r   )r   r  r   r>   r>   r?   rw     s     z[TestDataFrameConstructors.test_constructor_list_of_series_aligned_index.<locals>.<listcomp>r-   r   r/   r.   )r   r   r   r   r   r   0r  r   r5  rb  )r9   seriesr`   r=   r>   r>   r?   -test_constructor_list_of_series_aligned_index  s    zGTestDataFrameConstructors.test_constructor_list_of_series_aligned_indexc                 C   sJ   G dd dt }ddd}||g}|g}t|}t|}t|| d S )Nc                   @   s   e Zd ZdS )zTTestDataFrameConstructors.test_constructor_list_of_derived_dicts.<locals>.CustomDictNr  r  r  r>   r>   r>   r?   
CustomDict  s   r   r  r-   r   )rD  r   r3   r4   )r9   r   r   Zdata_customr   Zresult_customr`   r>   r>   r?   &test_constructor_list_of_derived_dicts  s    

z@TestDataFrameConstructors.test_constructor_list_of_derived_dictsc              	   C   s@   t jdt jdd}tjtdd t| W 5 Q R X d S )Nr   rm  r  z%All arrays must be of the same lengthre   )rK   r   rX  rj   rk   rl   r   )r9   r   r>   r>   r?   test_constructor_ragged  s    z1TestDataFrameConstructors.test_constructor_raggedc                 C   sF   t td}tddi|d}tddddgi|d}tj||dd d S )Nr-   r   r   rB   Fr_  )r   r   r   r3   r4   )r9   r_   r<   r=   r>   r>   r?   test_constructor_scalar  s    z1TestDataFrameConstructors.test_constructor_scalarc                 C   s"   t |d |jdgd}|  d S )Nr   r   )r   rC   r   )r9   r   r<   r>   r>   r?    test_constructor_Series_copy_bug  s    z:TestDataFrameConstructors.test_constructor_Series_copy_bugc              	   C   s   i }dddd|d< t ddddgddd	d
gd|d< t|}|jjsHttjtdd tddgdddd W 5 Q R X tddgt ddgddgdd}tddgddgdddgd}t	|| d S )Nr/   r.   r-   )r   r   r:  r   r   r   Zquxr:  r   rB   rZ   zambiguous orderingre   r   r   r   r  )
r   r   rC   Zis_monotonicr8   rj   rk   rl   r3   r4   r`  r>   r>   r?   &test_constructor_mixed_dict_and_Series  s     ""z@TestDataFrameConstructors.test_constructor_mixed_dict_and_Seriesc                 C   s8   ddgdg}t |}t ddgddgg}t|| d S )Nr/   r.   r-   r   r-   r   r   r`  r>   r>   r?    test_constructor_mixed_type_rows	  s    z:TestDataFrameConstructors.test_constructor_mixed_type_rowsztuples,lists)r>   r>   r-   r   r(   c                 C   s    t |}t |}t|| d S r   r   )r9   r2  listsr`   r=   r>   r>   r?   test_constructor_tuple  s    z0TestDataFrameConstructors.test_constructor_tuplec                 C   s4   t dddgi}t dtddgi}t|| d S )Nr   r  r  r   r   r3   r4   r.  r>   r>   r?   test_constructor_list_of_tuples"  s    z9TestDataFrameConstructors.test_constructor_list_of_tuplesc                 C   s   ddl m} |dtd}|dd|ddg}tddgddgd	}t|}t|| tddgddgd
}t|ddgd}t|| d S )Nr   )
namedtupleZPandasabr/   r-   r.   r   r   )r   rr  r   rr  r   )rJ  r  r   r   r3   r4   )r9   r  Znamed_tupler2  r=   r`   r>   r>   r?   $test_constructor_list_of_namedtuples'  s    z>TestDataFrameConstructors.test_constructor_list_of_namedtuplesc                 C   sd   ddl m} |ddtfdtfg}|dd|ddg}tddgddgd}t|}t|| d S )	Nr   make_dataclassPointrv   r   r-   r/   rv   r   )dataclassesr  r   r   r3   r4   )r9   r  r  r   r=   r`   r>   r>   r?   $test_constructor_list_of_dataclasses6  s    z>TestDataFrameConstructors.test_constructor_list_of_dataclassesc                 C   s   ddl m} |ddtfdtfg}|ddtfdtfdtfg}|dd	|d
d	d	g}tdtjgd	d	gtjd
gtjd	gd}t|}t|| d S )Nr   r  r  rv   r   HLinex0x1r-   r/   )rv   r   r  r  )r  r  r   r   rK   rM   r3   r4   )r9   r  r  r  r   r=   r`   r>   r>   r?   7test_constructor_list_of_dataclasses_with_varying_typesA  s    "zQTestDataFrameConstructors.test_constructor_list_of_dataclasses_with_varying_typesc              	   C   sb   ddl m} |ddtfdtfg}d}tjtt|d t|ddddd	g W 5 Q R X d S )
Nr   r  r  rv   r   z0asdict() should be called on dataclass instancesre   r/   r  )	r  r  r   rj   rk   	TypeErrorreescaper   )r9   r  r  rn   r>   r>   r?   1test_constructor_list_of_dataclasses_error_thrownQ  s
    zKTestDataFrameConstructors.test_constructor_list_of_dataclasses_error_thrownc              	   C   s~   ddddddddd	d
dddddddg}t dddgdddgdd	dgdddgd d dgd d dgd}t |}t|| d S )Nr/   r   rl  r   )FirstSecondThirdFourthr   r.   r%  rm  )r  r  r   r  r(   r-   r9  rn     r(  )r  r  r   r  YYYXXX)r  r  r  r   r"  r#  r   )r9   r   r=   r`   r>   r>   r?   #test_constructor_list_of_dict_order\  s    
z=TestDataFrameConstructors.test_constructor_list_of_dict_orderc              	   C   sv  t dddgdddgdd}t|}|jd	 dks4tt|j|j tj	d
}t |dd}t|}td|i}t
|| t |tddd}t|}td	|i}t
|| d}tjt|d t|ddgd W 5 Q R X t g dtd}t|}|jd	 dkstt |dd}t||gj}t||dddgd}t
|| t||gj}t||dd	dgd}t
|| d S )Nr/   r.   r-   r   r   r   rv   r   r   r   r[   r(  rB   z<Shape of passed values is \(10, 1\), indices imply \(10, 2\)re   r   rJ   rH   )rv   	Unnamed 0r&  r   )r   r   r   r8   r3   r   rC   rK   r   rX  r4   r   rj   rk   rl   r\   rt  )r9   r   r<   rS   ry  r=   rn   s1r>   r>   r?   test_constructor_Series_namedp  s4    z7TestDataFrameConstructors.test_constructor_Series_namedc                 C   s   t tddd}t tddd}tt|dgd|  tt|dgd|  t|dgdjshtt|dgdjs|td S )Nr   r   r[   r/   r   )r   r   r3   r4   r   to_framer   r8   )r9   s0r'  r>   r>   r?   )test_constructor_Series_named_and_columns  s    zCTestDataFrameConstructors.test_constructor_Series_named_and_columnsc                 C   s   t dddgdddgdd}t dddgdddgd	}tddg}t||d	}t||}|jd
 dksjtt|| t||d	}t||}|jd
 d
kstt|j	| t|| d S )Nr/   r.   r-   r   r   r   rv   r   rB   r   )
r   r   r   r   r   r8   r3   r4   r   rC   )r9   r'  s2Zother_indexdf1Zexp1r   Zexp2r>   r>   r?   +test_constructor_Series_differently_indexed  s    zETestDataFrameConstructors.test_constructor_Series_differently_indexedz#name_in1,name_in2,name_in3,name_out)r_   r_   r_   r_   )r_   r_   NN)r_   NNN)idx1idx2NN)r/  r/  r0  N)r/  r0  Zidx3N)NNNNc           
      C   s   t dddg|dt dddg|dt dddg|dg}dd t|d	d
dgD }t|}t dddddg|d}tdddtjtjgtjdddtjgtjtjdddgd|d}	t||	 d S )Nr   r   r   r[   r   ec                 S   s$   i | ]\}}|t d ddg|dqS )r   r/   r.   rB   rs   )ru   r   r   r>   r>   r?   r     s     zJTestDataFrameConstructors.test_constructor_index_names.<locals>.<dictcomp>rv   r   rr  r   r/   r.   )rv   r   rr  rB   )r   r   r   rK   rM   r3   r4   )
r9   Zname_in1Zname_in2Zname_in3Zname_outindicesr  r`   Zexp_indr=   r>   r>   r?   test_constructor_index_names  s"    	z6TestDataFrameConstructors.test_constructor_index_namesc                 C   s\   t |jd d }t |jd d }t|j||d}t|jt| t|jt| d S )Nr   r-   r   )r   rC   r   r   r6   r3   r   r   )r9   r   rC   r   r`   r>   r>   r?   test_constructor_manager_resize  s
    z9TestDataFrameConstructors.test_constructor_manager_resizec              	   C   s   t |d t|d dddgd}t||jd d ddgf  d}tjt|d( t |d t|d d d d W 5 Q R X d S )Nr   rZ   r  r   zdoes not match index lengthre   )r   r   r3   r4   r   rj   rk   rl   )r9   r   r<   rn   r>   r>   r?   %test_constructor_mix_series_nonseries  s     z?TestDataFrameConstructors.test_constructor_mix_series_nonseriesc              	   C   s`   d}t jt|d" ttjdgddggtjd}W 5 Q R X ttjdgddgg}t || d S )Nz6float-dtype values containing NaN and an integer dtypere   r/   r   rG   )r3   rN   rO   r   rK   rM   r%   r4   r9   rn   r<   r=   r>   r>   r?   %test_constructor_miscast_na_int_dtype  s
    &z?TestDataFrameConstructors.test_constructor_miscast_na_int_dtypec                 C   s^   t ddggddgd}t ddgg}ddg|_t|| t jdgddgd}t|| d S )Nrm  r   r   r   )rm  r   )r   r   r3   r4   Zfrom_records)r9   r<   ZedfZidfr>   r>   r?   "test_constructor_column_duplicates  s    
z<TestDataFrameConstructors.test_constructor_column_duplicatesc                 C   s   t ddgddgtd}t ddgddgtd}t|| t ddgddgtjd}t|| t ddgddgtjd}t|| t ddgddgdd}t|| d S )Nr   r/   r  ZU5)r   r\   r  r3   r4   rK   Zstr_Zunicode_rR  r>   r>   r?   (test_constructor_empty_with_string_dtype  s    zBTestDataFrameConstructors.test_constructor_empty_with_string_dtypec                 C   s.   t g dg|d}t dg|d}t|| d S )Nc1r  r   rH   r   )r9   Znullable_string_dtyper=   r<   r>   r>   r?   ,test_constructor_empty_with_string_extension  s    zFTestDataFrameConstructors.test_constructor_empty_with_string_extensionc              	   C   sX  t ddddgdddgd}t|t t|jd	|j|j t d
dddgdddgd}t|t t|jd|j|j t dddgddgd}t|t tj	ddgddggt
dddgddgd d}tjt|d t dddg W 5 Q R X tjt|d t dddgd W 5 Q R X d}tjt|d t dddgddgt W 5 Q R X d S )N        r/   r.   r-   r   r   r   r   r!   r   r%   rG   z*DataFrame constructor not properly called!re   r   zincompatible data and dtype)r   r3   r4   rK   r   r  r   rC   r   rL   r\   rj   rk   rl   r  r   )r9   r<   rn   r>   r>   r?   test_constructor_single_value  s6      	z7TestDataFrameConstructors.test_constructor_single_valuec                 C   s  t t jj}t t jj}t dj}t t jj}tdddtdtdddddd	t 	d
d}|j
}tt dgt |gd  t |gd  tdd}t|| tdddddd|t jd|d|t jd|dit 	d
d}|j
}tt dgt dg t dg t dg t |g ddd||gd}t|| tdddddd|t jdgd
 |d|t jdgd
 |dit 	d
d}|j
}tt dgt dg t dg t dg t |g ddd||gd}t|| d S )NrV   r/   r   r   Z20010101  r.   r   )r   rZ   r   r  Er   rB   r%   ZABCDEr   r   r   r   rG   r!   r\   )rK   rH   int_rJ   Zfloat_r   r   r   r   rx   r   r   r   r3   rR   rL   )r9   ZintnameZ	floatnameZdatetime64nameZ
objectnamer<   r`   r=   r>   r>   r?   test_constructor_with_datetimes4  s    

     





     





z9TestDataFrameConstructors.test_constructor_with_datetimesc                 C   s6   t dddd}dd |D }t|}|jdks2td S )Nr;  r  r   startr  r+   c                 S   s   g | ]}|  qS r>   to_pydatetimeru   tsr>   r>   r?   rw     s     zNTestDataFrameConstructors.test_constructor_with_datetimes1.<locals>.<listcomp>rV   )r   r   rH   r8   )r9   ind	datetimesZ
datetime_sr>   r>   r?    test_constructor_with_datetimes1{  s    z:TestDataFrameConstructors.test_constructor_with_datetimes1c                 C   sv   t dddd}dd |D }dd |D }t|dgd	}||d
< |j}ttdtdgdd
gd}t|| d S )Nr;  r  r   rD  c                 S   s   g | ]}|  qS r>   rF  rH  r>   r>   r?   rw     s     zNTestDataFrameConstructors.test_constructor_with_datetimes2.<locals>.<listcomp>c                 S   s   g | ]}|  qS r>   )r   rH  r>   r>   r?   rw     s     rK  r   datesdatetime64[ns]r\   rB   )r   r   r   r   rK   rH   r3   rR   )r9   rJ  rK  rM  r<   r`   r=   r>   r>   r?    test_constructor_with_datetimes2  s    z:TestDataFrameConstructors.test_constructor_with_datetimes2c                 C   s   t d}|tddd}td|idgd}|jd |ks@tt|j	t
ddi td|ig}|jd |ksvtt|j	t
ddi d S )	Nr    r/   zEnd Dater   rB   r   datetime64[ns, US/Eastern])pytztimezoneZlocalizer   r   iatr8   r3   rR   r   r   )r9   r,   rA   r<   r>   r>   r?    test_constructor_with_datetimes3  s    
 
 
z:TestDataFrameConstructors.test_constructor_with_datetimes3c                 C   s   t ddd}td|i}|jd jd ks,tt dddd}td|i}t|jd jdks^tt dddd}td|i}t|jd jdkstd S )	N20130101r-   r<  rD   r   UTCr*   r  )r   r   rT  r,   r8   r  )r9   drr<   r>   r>   r?    test_constructor_with_datetimes4  s    z:TestDataFrameConstructors.test_constructor_with_datetimes4c                 C   s^   t ddddd}td| jddi}t }||d< t|| td|i}t|| d S )	N1/1/2011r   10sr  r+   r  r,   r   TZdropr   r   Z	to_seriesZreset_indexr3   r4   )r9   r   r=   r<   r>   r>   r?    test_constructor_with_datetimes5  s    z:TestDataFrameConstructors.test_constructor_with_datetimes5c                 C   sV   t ddddd}t dddd}t||d}t| jdd	|d}t|| d S )
NrZ  r   r[  r  r\  r+   r  r   Tr]  r^  )r9   r   Zi_no_tzr<   r=   r>   r>   r?    test_constructor_with_datetimes6  s
    z:TestDataFrameConstructors.test_constructor_with_datetimes6rS   rb   c                 C   s*   t |j}ttdg}t|| d S )NrN  )r   r   r   rK   rH   r3   rR   )r9   rS   r`   r=   r>   r>   r?   %test_constructor_datetimes_with_nulls  s    
z?TestDataFrameConstructors.test_constructor_datetimes_with_nullsorderKr   r   FrH   zdatetime64[M]zdatetime64[D]zdatetime64[h]zdatetime64[m]zdatetime64[s]zdatetime64[ms]zdatetime64[us]rN  c                 C   s\   t jdddgdddgg||d}t|}tdddgdddgg}|j|d}t|| d S )	Nz
2015-01-01z
2015-01-02z
2015-01-03z
2017-01-01z
2017-01-02z
2017-02-03rH   rc  rG   )rK   rL   r   r   r3   r4   r9   rc  rH   nar<   r=   r>   r>   r?   !test_constructor_datetimes_non_ns  s    z;TestDataFrameConstructors.test_constructor_datetimes_non_nsztimedelta64[D]ztimedelta64[h]ztimedelta64[m]ztimedelta64[s]ztimedelta64[ms]ztimedelta64[us]timedelta64[ns]c                 C   s   t jt ddt ddgt ddt ddgg||d}t|d}ttddtddgtddtddgg}t|| d S )Nr/   r  r.   r   r   rf  rj  )rK   rL   r  r   r   r   r3   r4   rg  r>   r>   r?   !test_constructor_timedelta_non_ns  s    z;TestDataFrameConstructors.test_constructor_timedelta_non_nsc              	   C   s2  t dd tdD }|j}ttdgd }t|| t dd tdD }|j}ttdgd }t|| t ddd	gi}|jjd
 tdkst	t ddg}|jjd
 tdkst	t ddg}|jjd
 tdkst	t dddgi}|jjd
 tdkst	t dddgi}|jjd
 tdksFt	t dditdd}|jjd
 tdksvt	t dditdd}|jjd
 tdkst	t ddddgddddgt
ddd tdD ddddgd}|j}ttdtdtdtdtdgt
dd}t|| d S ) Nc                 S   s   g | ]}t d qS )r   )rK   rx   rt   r>   r>   r?   rw     s     zSTestDataFrameConstructors.test_constructor_for_list_with_dtypes.<locals>.<listcomp>r   r   c                 S   s    g | ]}t jt d ddqS )r   r$   rG   )rK   rL   rx   rt   r>   r>   r?   rw     s     r$   r   l        l       r   r%   r/   r.   r   r   r!   r-   rB   r   rl  g333333?gffffff@gffffff@g333333@r  c                 S   s   g | ]}t d ddqS )  r/   )r   r   r>   r>   r?   rw   B  s     rP  )r   r   r   r   r1  r\   rN  Zabcde)r   r   r   r   rK   rH   r3   rR   rQ   r8   r   )r9   r<   r`   r=   r>   r>   r?   %test_constructor_for_list_with_dtypes  sR    


	
z?TestDataFrameConstructors.test_constructor_for_list_with_dtypesc                 C   s@   t |dd}d|d< |d dk s(t|d dk r<td S )NTr   r   r   )r   r  r8   )r9   r   copr>   r>   r?   test_constructor_frame_copyS  s    z5TestDataFrameConstructors.test_constructor_frame_copyc                 C   s   |s^t |j}d|jd< |jd dk s.tt |jdd}d|jd< |jd dk rtnz|j }t |}|jjd jjstd|d< |j	d dkstt |dd}|jjd jjrtd	|d< |j	d d	kstd S )
Nr   Trn  r(   r   r   r   Fi  )
r   rP   r  r8   r   r6   rz   r   r   rQ   )r9   r   r:   r<   rS   r>   r>   r?   test_constructor_ndarray_copyY  s     



z7TestDataFrameConstructors.test_constructor_ndarray_copyc                 C   sB   |j }td|d idd}d|d d d < |d dk r>td S )Nr   Trn  r   )r  r   r  r8   )r9   r   r  r<   r>   r>   r?   test_constructor_series_copyr  s    z6TestDataFrameConstructors.test_constructor_series_copyr<   rB   g?g@r   rl  r>  r   c              	   C   s   t t|jD ]}|jd d |f  qtt|jt|j }t|dkr|tjt	dd |j
d d tjf  W 5 Q R X nft|dkrt|jd d |d f |j
d d tjf  n*t|jd d |f |j
d d tjf  d S )Nr   z^nan$re   r/   )r   r5   r   rQ   rK   rx   r   rj   rk   KeyErrorr   rM   r3   rR   r4   )r9   r<   r   Zindexerr>   r>   r?   test_constructor_with_nas{  s     0z3TestDataFrameConstructors.test_constructor_with_nasc                 C   s:   t dtjdgi}|d jtjks&t|d d r6td S )Nr   Fr/   )r   rK   rM   rH   r   r8   )r9   r   r>   r>   r?   &test_constructor_lists_to_object_dtype  s    z@TestDataFrameConstructors.test_constructor_lists_to_object_dtypec                 C   s\   t dddg}t|dd}t|d}t||jd}t||||d}t|| d S )	Nr   rZ   r   r   r/   r  rG   r  )	r   rK   rL   r2   Zbroadcast_tor   rH   r3   r4   )r9   catrS   r`   r=   r>   r>   r?   *test_constructor_ndarray_categorical_dtype  s    zDTestDataFrameConstructors.test_constructor_ndarray_categorical_dtypec                 C   s   t dtdidd}ttdddd}t|d | ttddd}| }ttdddd}t|d | |jdd}ttdddd}t|d | t tddd}ttdddd}t|d | d S )	Nr   r   r  rG   rI   r   r   r[   )r   r   r   r3   rR   r)  )r9   r<   r=   ry  r`   r>   r>   r?   test_constructor_categorical  s    z6TestDataFrameConstructors.test_constructor_categoricalc              	   C   sX   d}t jt|d tttdg}W 5 Q R X tdttdddi}t || d S )N%will be changed to match the behaviorre   r   r   r  rG   r3   rN   rO   r   r   r   r   r4   r7  r>   r>   r?   -test_construct_from_1item_list_of_categorical  s
    zGTestDataFrameConstructors.test_construct_from_1item_list_of_categoricalc              	   C   sx   d}t jt|d" tttdttdg}W 5 Q R X tttdddttdddddd	gd
}t || d S )Nry  re   r   Zabdr  rG   r0   r   r/   r   rz  r7  r>   r>   r?   (test_construct_from_list_of_categoricals  s    &zBTestDataFrameConstructors.test_construct_from_list_of_categoricalsc              	   C   sl   d}t jt|d tttdtdg}W 5 Q R X tttdddtdddd	gd
}t || d S )Nry  re   r   r  r  rG   r0   r   r/   r   rz  r7  r>   r>   r?   %test_from_nested_listlike_mixed_types  s    " z?TestDataFrameConstructors.test_from_nested_listlike_mixed_typesc              
   C   sd   d dg}d}tjt|d< tjt|d" ttt	dtt	dg W 5 Q R X W 5 Q R X d S )Nrd   z;Length of values \(6\) does not match length of index \(3\)ry  re   r   Zabdefg)
ri   rj   rk   rl   r3   rN   rO   r   r   r   )r9   rn   Zmsg2r>   r>   r?   0test_construct_from_listlikes_mismatched_lengths  s    zJTestDataFrameConstructors.test_construct_from_listlikes_mismatched_lengthsc                 C   s   ddddg}t |d}t |dd}t|| ddddg}t |d}t |dd}t|| td	dd
}t ttjtjtjgdddgd}||_t	d|i}t	dt dddgddi|d}t
|| d S )Nr/   r.   r-   r  rG   r   r   r   Z20000101r<  )rP   
categoriesrv   rB   )r   r   r3   rR   r   r   rK   rM   rC   r   r4   )r9   rE  r  resrC   r=   r<   r>   r>   r?   #test_constructor_categorical_series  s     z=TestDataFrameConstructors.test_constructor_categorical_seriesc                 C   s,   t dddgidg|d}|jj|ks(td S )Nr   r/   r.   r   r<  )r   r   rH   r8   )r9   rH   r   r>   r>   r?   %test_check_dtype_empty_numeric_column	  s    z?TestDataFrameConstructors.test_check_dtype_empty_numeric_columnc                 C   sF   t dddgidg|d}|r0|tjkr0t| |jjjdksBtd S )Nr   r/   r.   r   r<  r\   )	r   r3   BYTES_DTYPESr  &mark_array_manager_not_yet_implementedr   rH   rJ   r8   )r9   requestrH   r:   r   r>   r>   r?   $test_check_dtype_empty_string_column	  s    
z>TestDataFrameConstructors.test_check_dtype_empty_string_columnc                 C   sL   t dtd j}t dti}t|| tt dtdj}t|| d S )Nr   r%  )r   r\   r)  r   r3   rR   r   r.  r>   r>   r?   test_to_frame_with_falsey_names%	  s
    z9TestDataFrameConstructors.test_to_frame_with_falsey_namesr"   r  c                 C   sb   t ddddddgi|pdd}t td	dg|d
}t|| t dtd	i|d}t|| d S )Nr   r   r/   r.   r-   r   r%   rG   r   r<  )r   r   r3   r4   )r9   rH   r=   r`   r>   r>   r?   test_constructor_range_dtype.	  s
    z6TestDataFrameConstructors.test_constructor_range_dtypec                 C   s\   G dd dt }tdddgdddgg}t||dddg|dddgg}t|| d S )	Nc                   @   s   e Zd ZdS )zETestDataFrameConstructors.test_frame_from_list_subclass.<locals>.ListNr  r>   r>   r>   r?   List=	  s   r  r/   r.   r-   r   r   r(   )r   r   r3   r4   )r9   r  r=   r`   r>   r>   r?   test_frame_from_list_subclass;	  s    $z7TestDataFrameConstructors.test_frame_from_list_subclassextension_arrZaabbcz1/1/2017z1/1/2018)rE  endr  c                 C   s$   t t|}t |}t|| d S r   r
  )r9   r  r=   r`   r>   r>   r?   %test_constructor_with_extension_arrayD	  s    z?TestDataFrameConstructors.test_constructor_with_extension_arrayc                 C   s^   t  }||f}t|ttdtddi|gd}tdddgtt|gd}t|| d S )Nr-   rB   r   r   r/   r.   )r   todayr   r   r   r   r3   r4   )r9   rM  tupr`   r=   r>   r>   r?   *test_datetime_date_tuple_columns_from_dictS	  s
    "zDTestDataFrameConstructors.test_datetime_date_tuple_columns_from_dictc              	   C   s   t dddgtdddgd}t dd	dd
dgtdddddgd}t||g}ttdddtjtjgdddddggdddddgd}t|| d S )N'   r(   r   ZfemaleZmaleunknownrB   r.            frq   g      c@g     @n@g     b@r   r   )r   r   r   rK   rL   rM   r3   r4   )r9   r'  r,  r`   r=   r>   r>   r?   /test_construct_with_two_categoricalindex_series[	  s    $zITestDataFrameConstructors.test_construct_with_two_categoricalindex_seriesc                 C   s  t tdd}t|d d}t|d d}t||d}ttdddd	tdd
dd	td
ddd	tdddd	tdddd	tdddd	tdddd	tdddd	tdddd	tdddd	g
dd}tdgd t	j
gd  t	j
gd dgd  d|d}t|| d S )Nr   r   r   r   r
  gMbXg#@r  )closedg3@g33333=@gC@g     H@g33333M@g33333SQ@gS@gfffffFV@r   T)ZorderedrB   )r   r   r   Zvalue_countsheadtailr   r   r   rK   rM   r3   r4   )r9   rT   ser1ser2r`   rC   r=   r>   r>   r?   1test_constructor_series_nonexact_categoricalindexi	  s.    , zKTestDataFrameConstructors.test_constructor_series_nonexact_categoricalindexc                 C   s   t dddddt dddddfg}tj|ddgd	}t|}|d
 d |d d ksXt|d d |d d ksttt|d
 }t|d tst|d |d d kstd S )NrP  rn  r   rm  rW  r   )DateM8[us])Forecastingr  rG   r  r  r/   )r   rK   rL   r   r8   r   r{   r   )r9   rM  rS   r<   ry  r>   r>   r?   test_from_M8_structured	  s    "z1TestDataFrameConstructors.test_from_M8_structuredc                 C   s>   G dd dt }td|dddddgi}|j jdks:td S )Nc                   @   s   e Zd ZdS )zOTestDataFrameConstructors.test_from_datetime_subclass.<locals>.DatetimeSubclassNr  r>   r>   r>   r?   DatetimeSubclass	  s   r  r   i  r/   rN  )r   r   rH   r8   )r9   r  r   r>   r>   r?   test_from_datetime_subclass	  s    z5TestDataFrameConstructors.test_from_datetime_subclassc              	   C   s@   t dddd}d}tjt|d t|tdd W 5 Q R X d S )	Nr'   r-   r)   r*   z@Shape of passed values|Passed arrays should have the same lengthre   r   rB   )r   rj   rk   rl   r   r   )r9   r;   rn   r>   r>   r?   (test_with_mismatched_index_length_raises	  s    zBTestDataFrameConstructors.test_with_mismatched_index_length_raisesc                 C   sR   t dddd}t|}ttjt||d}t|d jtdsNt	d S )Nz1/1/2000 00:00:00z1/1/2000 1:59:50r[  r  r  rZ   rV   )
r   rK   r  r   r   rX  r5   Z
issubdtyperH   r8   )r9   r   rM  r<   r>   r>   r?   !test_frame_ctor_datetime64_column	  s    
z;TestDataFrameConstructors.test_frame_ctor_datetime64_columnc                 C   s   ddddgddddgg}t tjdddd |D d}t|jtsHtt|jtrXtt tjdd|d	}t|jts~td S )
Nr   r   rv   r   r   c                 S   s   g | ]}t |qS r>   r  rt   r>   r>   r?   rw   	  s     zYTestDataFrameConstructors.test_dataframe_constructor_infer_multiindex.<locals>.<listcomp>rB   r   )	r   rK   r   rX  r{   rC   r   r8   r   )r9   Zindex_listsmultir>   r>   r?   +test_dataframe_constructor_infer_multiindex	  s    zETestDataFrameConstructors.test_dataframe_constructor_infer_multiindex
input_valsr  r   rZ  Hr`  r\  c                 C   s6   t d|i|d}t d|id|i}t|| d S )Nr   rG   )r   r   r3   r4   )r9   r  r   r`   r=   r>   r>   r?   test_constructor_list_str	  s    z3TestDataFrameConstructors.test_constructor_list_strc                 C   s<   t dddd gi|d}t dddd gitd}t|| d S )Nr   r   r   rG   z1.0z2.0)r   r\   r3   r4   )r9   r   r`   r=   r>   r>   r?   test_constructor_list_str_na	  s    z6TestDataFrameConstructors.test_constructor_list_str_nar   c                    s  |r8|t jt j ks8|t jt j t j kr.|s8t| tj	ddg|d tj	ddg|dj
jdkrld S tj	ddg|dt d|dd	d
  fdd}|s|  djd< djd< |stfddjjD dkstdd< |r j
jdkrH d  j
ddks*td j
ddks|tn4 d  j
dksbtd j
dks|tjd dkstnjd dkstd S )Nr/   r.   rG   r-   r   )SUr   r   r   rn  c                 S   s0   t | tjr| jS t | jtjr(| jjS td S r   )r{   rK   ZndarraybaserH   Z_ndarrayr  r}   r>   r>   r?   get_base	  s
    z<TestDataFrameConstructors.test_dict_nocopy.<locals>.get_basec                      sn   t fddjjD dks"tt  fddjjD dksFtt fddjjD dksjtd S )Nc                 3   s   | ]}| kV  qd S r   r>   rt   r   r>   r?   r   	  s     zRTestDataFrameConstructors.test_dict_nocopy.<locals>.check_views.<locals>.<genexpr>r/   c                 3   s(   | ] }t |jtjr| kV  qd S r   r{   rH   rK   rt   )r   r  r>   r?   r   	  s   c                 3   s(   | ] }t |jtjr| kV  qd S r   r  rt   )r   r  r>   r?   r   	  s   )sumr6   rz   r8   r>   r   r   r   r<   r  r>   r?   check_views	  s    "	z?TestDataFrameConstructors.test_dict_nocopy.<locals>.check_viewsr   r   r0   c                 3   s   | ]}| kV  qd S r   r>   rt   r  r>   r?   r   
  s     z=TestDataFrameConstructors.test_dict_nocopy.<locals>.<genexpr>rr   rc   r  )r3   STRING_DTYPESr  DATETIME64_DTYPESTIMEDELTA64_DTYPESBOOL_DTYPESr  r  rK   rL   rH   rp   r  r   rQ   r  r6   rz   r8   type)r9   r  r   Zany_numeric_ea_dtypeZany_numpy_dtyper:   r  r>   r  r?   test_dict_nocopy	  sB    	
	

"	z*TestDataFrameConstructors.test_dict_nocopyc                 C   s2   t tddddgd}t dgd}t|| d S )Nr/   r   r[   r   r   r
  r.  r>   r>   r?   'test_from_series_with_name_with_columns'
  s    zATestDataFrameConstructors.test_from_series_with_name_with_columnsc                 C   sd   t dddgdddggdddgdd	d
ggd}t dddgdddggtdddgd}t|| d S )Nr/   r.   r-   r   r   r(   r   r   r   r   r   )r   r   )r   r   )r   r   )r   r   r1  r3   r4   r.  r>   r>   r?   test_nested_list_columns-
  s     z2TestDataFrameConstructors.test_nested_list_columnsc           
      C   s  t jddd}|jtdd}t|}|jdks8t|j	|j
k sLt||k  s`tt jdddd	g}|jtdd}t|}|jdkst|j	|j
k st||k  sttj||||f j}t|}t||||d
}	t||	 d S )Nz
2016-04-05r-   r<  r/   r   r/   r-   r   r   r(   r  )r  r  r1   r   r\   r2   r   r  r8   r   rH   r  ZIntervalIndexZfrom_breaksrK   Zr_rt  r3   r4   )
r9   pir   r<   iiZdata2r   Zdata3r   r=   r>   r>   r?   1test_from_2d_object_array_of_periods_or_intervals8
  s     zKTestDataFrameConstructors.test_from_2d_object_array_of_periods_or_intervalszcol_a, col_bc              	   C   s0   d}t jt|d t||d W 5 Q R X d S )Nz,Per-column arrays must each be 1-dimensionalre   r   r  )r9   Zcol_aZcol_brn   r>   r>   r?   test_error_from_2darrayN
  s    	z1TestDataFrameConstructors.test_error_from_2darrayc                 C   s   t dddgdddgdd}t|d j|d jr6td	|jd
< t d	tjtjgtjtjtjgtjtjtjgddddgtd}t	|| d S )Nr/   r.   r-   r   r   r   F)rC   r   r   r   r   r  rC   rH   )
r   rK   r   Z_valuesr8   rQ   rM   r\   r3   r4   r7  r>   r>   r?   &test_from_dict_with_missing_copy_false[
  s    
	z@TestDataFrameConstructors.test_from_dict_with_missing_copy_false)r  r  r  r@   rE   rU   rY   ra   ro   rj   markparametrizer~   r   r   r   r   r   r   r   r   r   r  Zskip_array_manager_invalid_testr   r   r   Zskipifr   r   r   r   r   r   r   r   rK   rL   r\   r   r   r	  r  r  r  r  r  r  rM   r  r   r$  r)  r-  r/  r4  r8  rC  rH  rO  rS  rV  rY  r[  r]  ra  rc  rf  rg  rv  r  r  r  r  r   r   r   r
   r   r	   r  r  r  r  r  r  r  r  Zslowr  r  r   r   r   rH   r   r  r  ZBooleanDtypeZFloat64DtypeZ
Int64DtypeZStringDtyper  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  rD  r   r  r  r  r  r  r  r  r  r  r	  r  r  r  r  r  r$  r(  r+  r.  r3  r4  r6  r8  r9  r:  r=  r?  rC  rL  rO  rU  rY  r_  ra  r   rW   rh   rb   rb  ri  rk  rm  rp  rq  Z&skip_array_manager_not_yet_implementedrr  r   rt  ru  rw  rx  r{  r|  r}  r~  r  r3   ZALL_INT_NUMPY_DTYPESZALL_INT_EA_DTYPESZFLOAT_NUMPY_DTYPESZCOMPLEX_DTYPESr  r  r  r  r  r  ZOBJECT_DTYPESr  r  Zarm_slowr  r  r   r   r   r   r  r  r  r  r  r  r  r  r  r  r   r  r  r  r  r  r  r  r  r>   r>   r>   r?   r&   H   s  
!

	


&	


&	



4	

=J
/"

	

'
	

	


2&
 G;(.8  4


 			

	["r&   c                   @   s   e Zd Zdd Zdd ZdS )&TestDataFrameConstructorIndexInferencec                 C   s|   t jdddd}ttjt||}t jdddd}ttjt||}t||d}t jdddd}t	|j
| d S )Nz1/1/19991/1/2012rr   r  z1/1/1980z	12/1/2001)r'  r,  )r  r  r   rK   r   rX  r5   r   r3   r   rC   )r9   Zrng1r'  Zrng2r,  r<   r  r>   r>   r?   Atest_frame_from_dict_of_series_overlapping_monthly_period_indexesn
  s    zhTestDataFrameConstructorIndexInference.test_frame_from_dict_of_series_overlapping_monthly_period_indexesc              	   C   s  t ddd}ttd|d}ttd|dd}ttd|dd}ttd}t|||d}tt|jt|j t|j td}t	
|j| t|||d	}tt|jt|j t|j td}	t	
|j|	 t|||d
}
tt|jt|j t|j td}	t	
|
j|	 t|||d}tt|jt|j t|j td}t	
|j| d}tjt|d t||||d W 5 Q R X tjt|d t|||d W 5 Q R X tjt|d t|||d W 5 Q R X d S )Nr'   r-   r<  rB   rW  z
US/Centralr   rG   )r   r   rZ   )rZ   r   r   )r   rZ   r   z0Cannot join tz-naive with tz-aware DatetimeIndexre   )r   rZ   r   r  )r   rZ   r  )r  r   rZ   )r   r   r   r]   r   r   r   rC   r\   r3   r   rj   rk   r  )r9   r;   r  r  Zser3Zser4r-  Z	exp_indexr   Z
exp_index3r   Zdf4Z
exp_index4rn   r>   r>   r?   /test_frame_from_dict_with_mixed_tzaware_indexesy
  sH        zVTestDataFrameConstructorIndexInference.test_frame_from_dict_with_mixed_tzaware_indexesN)r  r  r  r  r  r>   r>   r>   r?   r  m
  s   r  c                   @   s   e Zd Zdd ZdS ))TestDataFrameConstructorWithDtypeCoercionc              	   C   s  t jdd}d}tjt|d t|dd W 5 Q R X td  t| dd W 5 Q R X t j|d< d}tjt|d t|dd W 5 Q R X tjt|d t	|d	 dd W 5 Q R X d
}t
jt|d t|d W 5 Q R X t
jt|d t	|d	 d W 5 Q R X d S )Nr   r   z!if they cannot be cast losslesslyre   rF   rG   r   z)passing float-dtype values containing NaNr   z9Cannot convert non-finite values \(NA or inf\) to integer)rK   r   rX  r3   rN   rO   r   roundrM   r   rj   rk   rl   r   )r9   rS   rn   r>   r>   r?   "test_floating_values_integer_dtype
  s"    
zLTestDataFrameConstructorWithDtypeCoercion.test_floating_values_integer_dtypeN)r  r  r  r  r>   r>   r>   r?   r  
  s   r  c                   @   s   e Zd Zejdddgdd Zejdddgd	d
 Zdd Zdd Z	dd Z
dd Zdd Zdd Zdd Zdd Zdd Zejdeegdd Zd d! Zd"d# Zd$S )%&TestDataFrameConstructorWithDatetimeTZr,   r  zdateutil/US/Easternc           
      C   s   t dddd}||}td|d|d}td|j}|d	 j|ksHtd
d |D }dd |D }td|i}||d< ||d< ||d< |j}tt	dt|dt	dt|dgddddgd}	t
||	 d S )N2011/1/12012/1/1W-FRIr  r   r  rB   rc   rZ   c                 S   s   g | ]}|  qS r>   rF  rH  r>   r>   r?   rw   
  s     zeTestDataFrameConstructorWithDatetimeTZ.test_construction_preserves_tzaware_dtypes.<locals>.<listcomp>c                 S   s   g | ]}|  qS r>   rF  rH  r>   r>   r?   rw   
  s     rX  dr_tzdatetimes_naivedatetimes_with_tzrN  r  )r   r]   r   r	   tzinforH   r8   r   r   rK   r3   rR   )
r9   r,   rX  r  r<   Ztz_expectedr  r  r`   r=   r>   r>   r?   *test_construction_preserves_tzaware_dtypes
  s*    

	zQTestDataFrameConstructorWithDatetimeTZ.test_construction_preserves_tzaware_dtypespydtTFc           	   	   C   s  |}t d|d}|r| }t d}tt td|gidd}W 5 Q R X td|gi}t|| tt td|idgdd}W 5 Q R X t|| tt t|gdd}W 5 Q R X t|| tt ttj|gt	ddd}W 5 Q R X t|| tt t|dgdgdd}W 5 Q R X t|| tt tt
|ggdd}W 5 Q R X t|| tt t|ggdgdd}W 5 Q R X t|| d S )	N2019r  r   rN  rG   r  r  r<  )r   rG  r3   rN   rO   r   r4   rK   rL   r\   r   Zassert_equal)	r9   Ztz_aware_fixturer  r,   rI  ts_naiver`   r=   r<   r>   r>   r?   'test_constructor_data_aware_dtype_naive
  s6    "zNTestDataFrameConstructorWithDatetimeTZ.test_constructor_data_aware_dtype_naivec                 C   s   t tdddddd}tddd}t||d	}|d
 jsBtd|d
 jd
ksTtt|d
 t|d
d t|d t|dd d S )NrV  r-   r  r*   r   r[   Z20130110r<  r  r   zM8[ns, US/EasternrZ   )	r   r   r   rH   r8   rJ   r3   rR   r   )r9   r_   rX  r<   r>   r>   r?   test_from_dict	  s    z5TestDataFrameConstructorWithDatetimeTZ.test_from_dictc                 C   s   t ddddd}t|}t|d t|dd tt|}t|d t|dd t dddd}t|}t|d t|dd tt|}t|d t|dd d S )	NrV  r-   r  r   )r+   r,   rJ   r[   r*   r   )r   r   r3   rR   r   )r9   r0  r   r>   r>   r?   test_from_index  s    z6TestDataFrameConstructorWithDatetimeTZ.test_from_indexc                 C   sB   t ddd}t||d}td|d|d td|jd|d d S )Nr  r   r<  rB   r   r   )r   r   r   rP   )r9   rX  ry  r>   r>   r?   +test_frame_dict_constructor_datetime64_1680'  s    zRTestDataFrameConstructorWithDatetimeTZ.test_frame_dict_constructor_datetime64_1680c                 C   s@   t dddd}t|}td|d|d}|d   s<td S )	Nr  r  r  r  r   r  rB   rZ   )r   r   r   r   r  r8   )r9   rX  rI  r   r>   r>   r?   +test_frame_datetime64_mixed_index_ctor_1681/  s    zRTestDataFrameConstructorWithDatetimeTZ.test_frame_datetime64_mixed_index_ctor_1681c                 C   sX   t ddddd}t|dgd}tdtdddtd	ddtd
ddgi}t|| d S )Nz20130101T10:00:00r-   rt  r  )rE  r+   r  r,   Z
timestampsr   r  z20130101T10:01:00z20130101T10:02:00)r   r   r   r3   r4   )r9   rX  r`   r=   r>   r>   r?   test_frame_timeseries_column7  s    


	zCTestDataFrameConstructorWithDatetimeTZ.test_frame_timeseries_columnc                 C   sj   ddg}ddddddd	d
}t |dddg|d}t ddtjtjfg|tdddgd}t|| d S )NNevadaOhio333333@333333@)r@    r  333333?@)rl  r@  r  )r  r  r@  r  i  r   )r  r  )r  r  r5  )r   rK   rM   r   r3   r4   )r9   r   popr`   r=   r>   r>   r?   test_nested_dict_constructionF  s    
zDTestDataFrameConstructorWithDatetimeTZ.test_nested_dict_constructionc                 C   sd   t dddd}|jtdd}t|}|jdks8t|j|j	k
 sLt||k
 
 s`td S )Nz2016-04-05 04:30r-   rW  r*   r/   r   r  )r   r1   r   r\   r2   r   r  r8   r   rH   r  )r9   r;   r   r<   r>   r>   r?   test_from_tzaware_object_arrayU  s    zETestDataFrameConstructorWithDatetimeTZ.test_from_tzaware_object_arrayc              	   C   s   t jtdtdtdgtdddtjtdddgtdd	dtjtd
d	dggtdj}t|dddgd}dddg}|j|k	 st
d S )Nz2013-01-01 00:00:00z2013-01-02 00:00:00z2013-01-03 00:00:00z2013-01-01 00:00:00-0500r  r  z2013-01-03 00:00:00-0500z2013-01-01 00:00:00+0100ZCETz2013-01-03 00:00:00+0100rG   r   rZ   r   r   rN  rQ  zdatetime64[ns, CET])rK   rL   r   r  rb   r\   rt  r   r   r  r8   )r9   rS   r  Zexpected_dtypesr>   r>   r?   $test_from_tzaware_mixed_object_array^  s,    



zKTestDataFrameConstructorWithDatetimeTZ.test_from_tzaware_mixed_object_arrayc                 C   s:   t dd}t|dd}t|d}t|| d S )Nr   r  zdatetime64[ns, UTC]rG   )rK   rx   r2   r   r   r3   r4   )r9   Z
array_dim2r<   r=   r>   r>   r?   test_from_2d_ndarray_with_dtype}  s    zFTestDataFrameConstructorWithDatetimeTZ.test_from_2d_ndarray_with_dtyper   c              	   C   sh   |dddh}d|j  d}tjt|d td|i W 5 Q R X tjt|d t| W 5 Q R X d S )Nr/   r.   r-   'z' type is unorderedre   r   )r  rj   rk   r  r   r   )r9   r   rP   rn   r>   r>   r?   !test_construction_from_set_raises  s    zHTestDataFrameConstructorWithDatetimeTZ.test_construction_from_set_raisesc                 C   s>   t jdddddd}t|}tdd |jjD s:td S )	Nr   r9  rN  rG   r   r-   c                 s   s   | ]}t |tV  qd S r   )r{   r   )ru   rS   r>   r>   r?   r     s     zeTestDataFrameConstructorWithDatetimeTZ.test_construction_from_ndarray_datetimelike.<locals>.<genexpr>)rK   rx   r2   r   r  r6   rz   r8   r   r>   r>   r?   +test_construction_from_ndarray_datetimelike  s    zRTestDataFrameConstructorWithDatetimeTZ.test_construction_from_ndarray_datetimelikec              	   C   s   t jdd}tdgj}d}tjt|d t	|dg|d W 5 Q R X tddd	g}tjt|d t	|dd
gd W 5 Q R X d S )Nr   r.   r   z'len\(arrays\) must match len\(columns\)re   r   r<  r   rP  r   r   )
rK   r   rX  r  rL   rH   rj   rk   rl   r   )r9   rS   rH   rn   Zarr2r>   r>   r?   >test_construction_from_ndarray_with_eadtype_mismatched_columns  s    zeTestDataFrameConstructorWithDatetimeTZ.test_construction_from_ndarray_with_eadtype_mismatched_columnsN)r  r  r  rj   r  r  r  r  r  r  r  r  r  r  r  r  r  set	frozensetr  r  r  r>   r>   r>   r?   r  
  s"   

&	

r  c                 C   s"   t | tr| jd S | jd S d S )Nr   r   )r{   r   rQ   r  r>   r>   r?   get1  s    

r  c                   @   s   e Zd Zejeedgddd Zej	dddgdd	 Z
d
d Zdd Zdd Zej	dejejgdd Zej	deejgdd Zej	deejgdd Zdd ZdS )TestFromScalarN)paramsc                    s   |j }dtdi tkr$dg d< |d kr:tjf S |tkrhtkrX fddS  fddS n$tkr~ fddS  fd	dS d S )
NrC   r.   r   r   c                    s   | | df |S )Nr0   r>   rv   kwargsextrarm   r>   r?   r     s
    z,TestFromScalar.constructor.<locals>.<lambda>c                    s   d| if |S Nr   r>   r  r  r>   r?   r     r   c                    s   | | gf |S r   r>   r  r  r>   r?   r     r   c                    s   d| | gif |S r  r>   r  r  r>   r?   r     s
   
 )paramr   r   r  r  rD  r   )r9   r  rm   boxr>   r  r?   r     s    
zTestFromScalar.constructorrH   rV   rg   c                 C   s8   |t j|d}t|j|ks"tt| s4td S r  )r  rb   rK   r  r   r8   r   )r9   rH   r   r}   r>   r>   r?   test_from_nat_scalar  s    z#TestFromScalar.test_from_nat_scalarc                 C   s(   t d}||dd}t||ks$td S )Nr/   rg   rG   r   r  r8   )r9   r   r  r}   r>   r>   r?   *test_from_timedelta_scalar_preserves_nanos  s    z9TestFromScalar.test_from_timedelta_scalar_preserves_nanosc                 C   s,   |t d }||dd}t||ks(td S )Nr/   rV   rG   r  )r9   r   Zfixed_now_tsrI  r}   r>   r>   r?   *test_from_timestamp_scalar_preserves_nanos  s    z9TestFromScalar.test_from_timestamp_scalar_preserves_nanosc                 C   s4   t d}| }||td}tt|tjs0td S )Nr/   rG   )r   Zto_timedelta64r\   r{   r  rK   r  r8   )r9   r   r  Ztd64r}   r>   r>   r?   #test_from_timedelta64_scalar_object  s    z2TestFromScalar.test_from_timedelta64_scalar_objectclsc              	   C   s   |dd}t jdt jdi| }d}|t jkr:dddg}tjt|d |||d	 W 5 Q R X |d
d}tjt|d |||d	 W 5 Q R X d S )Nrb   rc   rg   rV   zCannot castrd   z?dtype datetime64\[ns\] cannot be converted to timedelta64\[ns\]re   rG   r   )rK   rh   r  ri   rj   rk   r  )r9   r   r  r  rH   rn   r>   r>   r?   (test_from_scalar_datetimelike_mismatched  s    


z7TestFromScalar.test_from_scalar_datetimelike_mismatchedc                 C   sB   t ddd}|tjkr"t|d}||}tt||ks>td S )N'  r/   r  )r   rK   rh   r  r  r8   r9   r   r  r  r`   r>   r>   r?    test_from_out_of_bounds_datetime  s
    
z/TestFromScalar.test_from_out_of_bounds_datetimec                 C   sN   t dddt ddd }|tjkr.t|d}||}tt||ksJtd S )Nr  r/   i  r  )r   rK   r  r  r  r8   r  r>   r>   r?   !test_from_out_of_bounds_timedelta  s
    
z0TestFromScalar.test_from_out_of_bounds_timedeltac              	   C   sh   d}t d|d}t d}tjtdd ||dd}W 5 Q R X t|jdksRtt||ksdtd S )Nr  r  r  zData is timezone-awarere   rV   rG   )r   r3   rN   rO   rK   r  r   r8   )r9   r   r,   rI  r  r`   r>   r>   r?   test_tzaware_data_tznaive_dtype  s    z.TestFromScalar.test_tzaware_data_tznaive_dtype)r  r  r  rj   Zfixturer   rD  r   r  r  r  r  r   r  rK   rh   r  r  r   r  r   r  r  r>   r>   r>   r?   r    s   




r  )BrJ  r   r   r   r   r   r  r  r  r  numpyrK   Znumpy.mar  Znumpy.ma.mrecordsr  rj   rR  Zpandas.compatr   Zpandas.util._test_decoratorsutilZ_test_decoratorsr  Zpandas.core.dtypes.commonr   Zpandas.core.dtypes.dtypesr	   r
   r   r   Zpandasr  r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   Zpandas._testingZ_testingr3   Zpandas.arraysr   r   r   r   Zpandas.core.apir    r   r   r&   r  r  r  r  r  r>   r>   r>   r?   <module>   sp   D
                    99 `