Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/frame/__pycache__/test_constructors.cpython-37.pyc
Назад
B �5�g � � @ s� d dl mZmZ d dlmZmZmZ d dlZd dlZd dlZd dl Z d dl Z d dl m Z d dlm mZ d dlZd dlZd dlmZ d dlm mZ d dlmZ d dlmZm Z m Z m Z d dl Z d dl 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 dl/m0Z1 d dl2m3Z3m4Z4m5Z5m6Z6 d d d gZ7d d ddddddgZ8G dd� d�Z9G dd� d�Z:G dd� d�Z;dd� Z<G d d � d �Z=dS ) � )� 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� date_range�isna)� DatetimeArray� IntervalArray� PeriodArray� SparseArrayZfloat16Zfloat32�float64�uint8Zuint16Zuint32�uint64Zint8Zint16�int32�int64c @ s( e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zej � d d d g�dd� �Z dd� Z ej � ddd� dd� dd� dd� dd� dd� dd� dd� d d� d d� d d� d d� d d� g �d!d"� �Z ej � d#g ged$�ed%�fg g ged&�ed%�fd'd(� d)D �ged$�ed%�fg�d*d+� �Zd,d-� Zd.d/� Zd0d1� Zd2d3� Zejd4d5� �Zd6d7� Zej jed8d9�d:d;� �Zd<d=� Zd>d?� Z d@dA� Z dBdC� Z dDdE� Z dFdG� Z ej � dHe!j"dIge#dJ�e!�"d&dK g�dLge!j"dMge#dJ�e!�"dNg�d&dK d& gg�dOdP� �Z$dQdR� Z%dSdT� Z&dUdV� Z'dWdX� Z(dYdZ� Z)d[d\� Z*d]d^� Z+d_d`� Z,ej � dad&e!j-dbdcg�ddde� �Z.ej � dfd&e!j-dbe/dg�g�dhdi� �Z0ej � dfe!j-dbe/dg�g�djdk� �Z1dldm� Z2dndo� Z3dpdq� Z4drds� Z5dtdu� Z6dvdw� Z7dxdy� Z8dzd{� Z9d|d}� Z:d~d� Z;d�d�� Z<d�d�� Z=d�d�� Z>d�d�� Z?d�d�� Z@d�d�� ZAd�d�� ZBd�d�� ZCd�d�� ZDd�d�� ZEd�d�� ZFej � d�eGd��eHd �feId%d�d��eJd�d��feKd�d�d��eLd�d��fg�d�d�� �ZMd�d�� ZNd�d�� ZOd�d�� ZPd�d�� ZQd�d�� ZRd�d�� ZSd�d�� ZTej jUd�d�� �ZVd�d�� ZWej � d�dbeXeYd���d�d�ge#e!jZfdbdbd�d�gd�e!�[d��fdbeXeYd���d�d�ge\e!�[d��fi dbd�d�gdbe!jZfd�d$ieXeYd���eXd��e\e!�[d��fg�d�d�� �Z]ej � d�d�d�dbgd�e^j_fd�d�dbgd�e^j`fd$d&dbgd�e^jafd�d�d�gd�e^jbff�d�dȄ �Zcd�dʄ Zdd�d̄ Zed�d΄ Zfd�dЄ Zgd�d҄ Zhd�dԄ Zid�dք Zjd�d Zkd�dڄ Zlej � d�eKd܃ggd�eKd܃igd�eKd܃gid�eKd܃ig�d�d߄ �Zmd�d� Znd�d� Zod�d� Zpd�d� Zqd�d� Zrd�d� Zsd�d� Ztd�d� Zud�d� Zvej � d�ewexg�d�d� �Zyej � d�ewexg�d�d�� �Zzd�d�� Z{d�d�� Z|d�d�� Z}d�d�� Z~d��d � Z�d�d� Z��d�d� Z�ej � �dd)g fd)g f�dd)d)gf�dg g gfg g fg g gfd$gd&gfd$gd&ggfd$d&�dg�dd��d gfd$d&�dg�dd��d ggfg��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�ej � �d �d �d �d!�d"�d#�d$�d%g��d&�d'� �Z��d(�d)� Z��d*�d+� Z��d,�d-� Z��d.�d/� Z��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�ej � �dDe!�"dbdbdbdbe���� dbg�e!�"dbdbe���� dbg�e!���dE�gdbgge!���dE�ge^j�ggdbge!���dE�ggdbge^j�gge^j�ge!���dE�gge^j�gdbggg��dF�dG� �Z�ej � �dH�dI�dJ�dK�dLg�ej � �dM�dN�dO�dP�dQ�dR�dS�dT�dUg��dV�dW� ��Z�ej � �dH�dI�dJ�dK�dLg�ej � �dM�dX�dY�dZ�d[�d\�d]�d^g��d_�d`� ��Z��da�db� Z��dc�dd� Z�ej��de�df� �Z�ej��dg�dh� �Z��di�dj� Z��dk�dl� Z��dm�dn� Z��do�dp� Z��dq�dr� Z��ds�dt� Z��du�dv� Z��dw�dx� Z��dy�dz� Z�ej � �dMe�j�e�j� e�j� e�j� e�j� e�j� e�j� ��d{�d|� �Z�ej�ej � �dMe�j�e�j� e�j� ��d}�d~� ��Z��d�d�� Z�ej j�ej � �dMdb�d��d�g��d��d�� ��Z��d��d�� Z�ej � �d�e�eX�d���e�d$e!j-e!j-e!j-g�e�eId%d$�eId$d��g�e�e^jŐd��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�ej � �d�d$d&g�d��d�geXeΐd�d&�d��d���eXeΐd�d&�d�d��d���eId%d�d��gg��d��d�� �Zϐd��d�� Z�ej � �d�d�d�g�ej��d��d�� ��Zѐd��d�� ZҐd��d�� ZӐd��d�� Z�dbS (� �TestDataFrameConstructorsc C sl t d��d�}td|idgd�}td|gi�}t�||� t|dgd��}t|gdgd��}t�||� d S )Nz2019-11-03 01:00:00-0700zAmerica/Los_Angeles�dtr )�index� )r% �value)r Z tz_convertr �tm�assert_frame_equal)�selfr% �df�expected� r. �Z/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/frame/test_constructors.py�)test_constructor_dict_with_tzaware_scalarG s zCTestDataFrameConstructors.test_constructor_dict_with_tzaware_scalarc C st t �dt jgddgg�}t|dd�}|jj|jks6t�t|jd �sHt�t |d ddd�}|jd }t � ||� d S ) Nr' � � �i8)�dtype)r r' r )r4 �name) �np�array�nanr �valuesr4 �AssertionErrorr �ilocr r) �assert_series_equal)r+ �arrr, Zserr- r. r. r/ �-test_construct_ndarray_with_nas_and_int_dtypeT s zGTestDataFrameConstructors.test_construct_ndarray_with_nas_and_int_dtypec C s0 t t�� t�� g�}|d jt�d�ks,t�d S )Nr zM8[ns])r r �nowr4 r6 r: )r+ r, r. r. r/ �%test_construct_from_list_of_datetimes` s z?TestDataFrameConstructors.test_construct_from_list_of_datetimesc C s\ t ddgdd�}|�d�}ttj|�� dd�dd�}|j|jksDt�t|�}t� ||� d S )Nz2013-1-1 13:00z2013-1-2 14:00�B)r5 z US/Pacific�object)r4 ) r � tz_localizer r6 r7 �tolistr4 r: r) r< )r+ Znaive�idxr- �resultr. r. r/ �+test_constructor_from_tzaware_datetimeindexd s zETestDataFrameConstructors.test_constructor_from_tzaware_datetimeindexc C sh t �dd�}t j|gtd�}|tkr0|�dd�}d�ddg�}tjt |d�� ||d d� W d Q R X d S ) N�NaT�ns)r4 r' �|z+Could not convert object to NumPy timedeltaz=Invalid type for timedelta scalar: <class 'numpy.datetime64'>)�matchzm8[ns]) r6 � datetime64r7 rB r �reshape�join�pytest�raises� ValueError)r+ �frame_or_seriesZnatr= �msgr. r. r/ �,test_array_of_dt64_nat_with_td64dtype_raisesq s zFTestDataFrameConstructors.test_array_of_dt64_nat_with_td64dtype_raises�kind�m�Mc C s| |dkrd}t }nd}t}tjddd��|��dd�}|tkrN|d d �d f }||td�}|jj d j tkspt �t |jj d � � d |�s�t �|||�td�}|jj d j tks�t �t |jj d � � d |�s�t �|||�tt�d�}|jj d j tks�t �t |jj d � � d |��s t �|tk�rxd d � |D �}||td�}|jj d j tk�sXt �t |jj d � � d |��sxt �d S ) NrW zM8[ns]zm8[ns]� r3 )r4 r2 r1 r c S s g | ] }t |��qS r. )r )�.0�xr. r. r/ � <listcomp>� s zXTestDataFrameConstructors.test_datetimelike_values_with_object_dtype.<locals>.<listcomp>)r r r6 �arange�viewrM r rB �_mgr�arraysr4 r: � isinstance�ravelr r )r+ rU rR r4 Z scalar_typer= �objZsersr. r. r/ �*test_datetimelike_values_with_object_dtype� s, zDTestDataFrameConstructors.test_datetimelike_values_with_object_dtypec C st t td�dd�}t td�dd�}t|dgd�}tg dgd�}t�||� t|dgd�}tg dgd�}t�||� d S )N� r' )r5 r )�columns)r �ranger r) r* )r+ rZ �yrF 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 )N)r r. r. r. r/ rj � rk c C s t i �S )N)r r. r. r. r/ rj � rk c C s t d�S )Nr. )r r. r. r. r/ rj � rk c C s t g �S )N)r r. r. r. r/ rj � rk c C s t dd� dD ��S )Nc s s | ] }|V qd S )Nr. )rY �_r. r. r/ � <genexpr>� s z5TestDataFrameConstructors.<lambda>.<locals>.<genexpr>r. )r r. r. r. r/ rj � rk c C s t td��S )Nr )r rf r. r. r. r/ rj � rk c C s t d d�S )N)�data)r r. r. r. r/ rj � rk c C s t i d�S )N)rn )r r. r. r. r/ rj � rk c C s t dd�S )Nr. )rn )r r. r. r. r/ rj � rk c C s t g d�S )N)rn )r r. r. r. r/ rj � rk c C s t dd� dD �d�S )Nc s s | ] }|V qd S )Nr. )rY rl r. r. r/ rm � s z5TestDataFrameConstructors.<lambda>.<locals>.<genexpr>r. )rn )r r. r. r. r/ rj � rk c C s t td�d�S )Nr )rn )r rf r. r. r. r/ rj � rk c C s@ t � }|� }t|j�dks t�t|j�dks0t�t�||� d S )Nr )r �lenr& r: re r) r* )r+ ri r- rF r. r. r/ �test_empty_constructor� s z0TestDataFrameConstructors.test_empty_constructorz)emptylike,expected_index,expected_columnsr' r r1 c c s | ] }|V qd S )Nr. )rY rl r. r. r/ rm � s z#TestDataFrameConstructors.<genexpr>r. c C s$ t ||d�}t |�}t�||� d S )N)r& re )r r) r* )r+ Z emptylikeZexpected_indexZexpected_columnsr- rF r. r. r/ �test_emptylike_constructor� s z4TestDataFrameConstructors.test_emptylike_constructorc C s8 t �� \}}t||d�}t|�}|d jtjks4t�d S )N)r& �foo)r) ZgetMixedTypeDictr r4 r6 �object_r: )r+ Zfloat_string_framer& rn Z indexed_frameZunindexed_framer. r. r/ �test_constructor_mixed� s z0TestDataFrameConstructors.test_constructor_mixedc C s� d}t jt|d�� tddddgitjd�}W d Q R X |d jtksHt�tt� d��}t� d�� � |d< d � d d g�}t j t|d�� t� d�|d < W d Q R X t� d�� � |d < d S )NzBeither all columns will be cast to that dtype, or a TypeError will)rK �a�b�c)r4 )� r1 rr rJ z3Wrong number of items passed 2, placement implies 1z5Expected a 1D array, got an array with shape \(4, 2\)�testZfoo2)r) �assert_produces_warning� FutureWarningr r6 r r4 rB r: �onesrD rN rO rP rQ )r+ rS rr r, r. r. r/ � test_constructor_cast_failure� s "z7TestDataFrameConstructors.test_constructor_cast_failurec C sD t dgdgdgd��}t |tdd�}d|d< |d d dks@t�d S ) Ng �?g @g @)�col1�col2�col3T)r4 �copyg i@r~ r )r �floatr: )r+ Zorig_dfZnew_dfr. r. r/ �test_constructor_dtype_copy� s z5TestDataFrameConstructors.test_constructor_dtype_copyc C sB t ddgg�}t ||d jd�}d|d d< |jd dks>t�d S )Nr' r1 r )r4 �c )r r )r r4 r9 r: )r+ r, �should_be_viewr. r. r/ �,test_constructor_dtype_nocast_view_dataframe s zFTestDataFrameConstructors.test_constructor_dtype_nocast_view_dataframec C sD t ddgg�}t |j|d jd�}d|d d< |jd dks@t�d S )Nr' r1 r )r4 �a )r r )r r9 r4 r: )r+ r, r� r. r. r/ �+test_constructor_dtype_nocast_view_2d_array s zETestDataFrameConstructors.test_constructor_dtype_nocast_view_2d_arrayc C s@ t ddgd dggtd�}|jd d ks*t�|jd dks<t�d S )Nr' �2ru )r4 )r' r )r r' )r rB �locr: )r+ r, r. r. r/ � test_constructor_dtype_list_data s z:TestDataFrameConstructors.test_constructor_dtype_list_dataz NumPy change.)�reasonc C s� t � }t�d�}tjtdd�� t |g� W d Q R X tjtdd�� t |g� W d Q R X t dddgi�}tjtdd�� t ||g� W d Q R X d S )N)r r zshape=\(1, 0, 0\))rK �Ar' r1 zshape=\(2, 2, 1\))r r6 �emptyrO rP rQ )r+ ru rv r. r. r/ �"test_constructor_list_of_2d_raises s z<TestDataFrameConstructors.test_constructor_list_of_2d_raisesc C sR d dd�}d dd�}|d�}||� |ddddd ��}||� |d �}||� d S ) Nc S s� | dkr t }dd� |D �}n| dkr6t}dd� |D �}x$t||�D ]\}}|j|ksBt�qBW |d krhi }|�dd� t||�D �� t|�S )N�intc S s" g | ]}t jt j�d �|d��qS )� )r4 )r6 r7 �random�rand)rY �dr. r. r/ r[ & s zjTestDataFrameConstructors.test_constructor_mixed_dtypes.<locals>._make_mixed_dtypes_df.<locals>.<listcomp>r� c S s& g | ] }t jt jjd d d�|d��qS )r� )�size)r4 )r6 r7 r� �randint)rY r� r. r. r/ r[ * s c S s i | ]\}}||�qS r. r. )rY r� ru r. r. r/ � <dictcomp>1 s zjTestDataFrameConstructors.test_constructor_mixed_dtypes.<locals>._make_mixed_dtypes_df.<locals>.<dictcomp>)�MIXED_INT_DTYPES�MIXED_FLOAT_DTYPES�zipr4 r: �updater )�typ�ad�dtypesr_ r� ru r. r. r/ �_make_mixed_dtypes_df" s zVTestDataFrameConstructors.test_constructor_mixed_dtypes.<locals>._make_mixed_dtypes_dfc S s< |d krt t }x&|D ] }|| kr| j| |kst�qW d S )N)r� r� r� r: )r, r� r� r. r. r/ �_check_mixed_dtypes4 s zTTestDataFrameConstructors.test_constructor_mixed_dtypes.<locals>._check_mixed_dtypesr� r' rr �bar)r� rA �Cr� )N)Nr. )r+ 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|jjksHt �|j|j jksZt �d S )Nr� )ru rv ) r6 r� r� �astypeZ complex64� complex128r r4 ru r: rv )r+ ru rv r, r. r. r/ � test_constructor_complex_dtypesG s z9TestDataFrameConstructors.test_constructor_complex_dtypesc C sx t ddd gi|d�}|�� }t dddgi�}t�||� |jd d ksJt�t ddtjgi|d�}t�|jd �stt�d S )Nr� rZ )r4 FT)r' r ) r r r) r* r; r: r6 r8 �isnan)r+ � string_dtyper, rF r- r. r. r/ �$test_constructor_dtype_str_na_valuesP s z>TestDataFrameConstructors.test_constructor_dtype_str_na_valuesc C s� |j dd�}t|jj�d d d� |j_|j}t|�}t�|jt |jj�� t||d�}t�|jt |jj�� t�|j|� t � t |��d d d� }t||ddgd�}t||d�j ddgd�}t�||� d S )NF)r& ���r� rA )r& re )re )Z to_records�listr4 �namesr& r r) �assert_index_equalre r r6 r\ ro �reindexr* ) r+ � float_frameZrecr& r, �df2�rng�df3r- r. r. r/ �test_constructor_rec[ s z.TestDataFrameConstructors.test_constructor_recc C s8 t tjdtd�tjdtd�d��}|jjtjks4t�d S )Nr� )r4 )r r' ) r r6 r| �bool�zerosr9 r4 �bool_r: )r+ r, r. r. r/ �test_constructor_booln s "z/TestDataFrameConstructors.test_constructor_boolc C s� t jdd� tdd�D �t jd�}td|i�}|d jt jks@t�ddd d d g}d d g}t jt|�f|d�}||d d �< t|�}|d jt jks�t�d S )Nc S s g | ] }d | �qS )l r. )rY �ir. r. r/ r[ t s zMTestDataFrameConstructors.test_constructor_overflow_int64.<locals>.<listcomp>r' r� )r4 ru )l �Y�2�< i )l FJLW]L* � )l �M�:M�^ �- )l 8��M�a i )l r*R3_PvN i )�uid�u8)Zscorer� r� ) r6 r7 rf r! r r4 r: r� ro )r+ r9 rF Z data_scoresr4 rn Z df_crawlsr. r. r/ � test_constructor_overflow_int64r s z9TestDataFrameConstructors.test_constructor_overflow_int64r9 l )r4 �A l l���� l���� c C s: |d }t |�}|d jtks"t�|d d |ks6t�d S )Nr )r r4 rB r: )r+ r9 r( rF r. r. r/ � test_constructor_int_overflow� s z7TestDataFrameConstructors.test_constructor_int_overflowc C s` dd l }d}tt|��}|�|� dd� |D �}ttt|dgg| ���}|t|j�ks\t�d S )Nr �d c S s g | ]}d |d���qS )r� r� r. )rY r� r. r. r/ r[ � s zJTestDataFrameConstructors.test_constructor_ordereddict.<locals>.<listcomp>) r� r� rf �shuffler r r� re r: )r+ r� Znitems�numsr- r, r. r. r/ � test_constructor_ordereddict� s z6TestDataFrameConstructors.test_constructor_ordereddictc C s2 t jdd�}t jdd�dd � }t||d��}t|�dks>t�t|�dksNt�t �|d |�d�� tt� tj gd |j g�|j dd�}t �||d � t||d�dd d gd �}t|�t|�ks�t�d|ks�t�t |d ��� s�t�tt� �d ks�t�d }tjt|d�� tddd�dddgd�� W d Q R X d S )N� )�nperrd )r~ r � r~ r )r& r5 r� Zcol4)re r z<Mixing dicts with non-Series may lead to ambiguous ordering.)rK ru rv )ru rv rw )r� rA )r) �makeTimeSeriesr ro r: r< �renamer r6 Z concatenater8 r9 r& r �allrO rP rQ )r+ �datetime_series�datetime_series_short�frame�exprS r. r. r/ �test_constructor_dict� s* z/TestDataFrameConstructors.test_constructor_dictc C s, t dddd�i�}t�|jtddg�� d S )Nr� r' r1 )�1r� r� r� )r r) r� r& r )r+ r� r. r. r/ � test_constructor_dict_length1� s z7TestDataFrameConstructors.test_constructor_dict_length1c C s, t dddg�}ti |d�}|j|ks(t�d S )Nr r' r1 )r& )r r r& r: )r+ rE r� r. r. r/ � test_constructor_dict_with_index� s z:TestDataFrameConstructors.test_constructor_dict_with_indexc C sN t dddg�}ti ||d�}|j|ks*t�|j|ks8t�t|j�dksJt�d S )Nr r' r1 )r& re r2 )r r r& r: re ro �_series)r+ rE r� 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�ddgd�}tj|jtd�dd� d S )N)r� rA r� rA )re r T)�exact)r r) r� r& r )r+ r� r. r. r/ �$test_constructor_dict_of_empty_lists� s z>TestDataFrameConstructors.test_constructor_dict_of_empty_listsc C s^ t dd idgd�}t dd gidgd�}|�dd�d ks:t�|�dd�d ksNt�t�||� d S )Nru r )r& )r � _get_valuer: r) r* )r+ Z frame_noneZframe_none_listr. r. r/ � test_constructor_dict_with_none� s z9TestDataFrameConstructors.test_constructor_dict_with_nonec C sZ d}t jt|d�� tddi� W d Q R X t jt|d�� tddidgd� W d Q R X d S )Nz2If using all scalar values, you must pass an index)rK ru gffffff�?)re )rO rP rQ r )r+ rS r. r. r/ � test_constructor_dict_errors� s z6TestDataFrameConstructors.test_constructor_dict_errors�scalarN�Dc C s. t d|idgd�}t dgd�}t�||� d S )Nru rv )re )r r) r* )r+ r� rF r- r. r. r/ �%test_constructor_invalid_items_unused� s z?TestDataFrameConstructors.test_constructor_invalid_items_unusedr( r8 c s� d|dg� d|g�ddgddgddgg�� ��fdd �t d�D �}t|��d�jddd �}ttjd d d ��dd��� d�}t�||� t|�d�jddd �}t�||� t|�� d�}t�||� d S )Nr' r2 ru r rx r1 rd c s" i | ]}t �| �d �� | �qS ))r& )r )rY rw )�colsrE r9 r. r/ r� s zKTestDataFrameConstructors.test_constructor_dict_nan_key.<locals>.<dictcomp>)�axisrX r# )r4 )r& re )r& )rf r � sort_valuesr6 r\ rM r) r* )r+ r( rn rF r- r. )r� rE r9 r/ � test_constructor_dict_nan_key� s z7TestDataFrameConstructors.test_constructor_dict_nan_keyc s� t d|dfd|fg�� t d|f|dfg��ddgdd gdd gg�� ��fd d �td�D �}t|��d�jd|fdd �}ttjddd��dd��� d�}t�||� t|�d�jd|fdd �}t�||� t|�� d�}t�||� d S )N)� � � � ru r1 r r2 r' rx rd c s" i | ]}t �| �d �� | �qS ))r& )r )rY rw )r� rE r9 r. r/ r� s zQTestDataFrameConstructors.test_constructor_dict_nan_tuple_key.<locals>.<dictcomp>)r� rX r# )r4 )r& re )r& ) r rf r r� r6 r\ rM r) r* )r+ r( rn rF r- r. )r� rE r9 r/ �#test_constructor_dict_nan_tuple_key s z=TestDataFrameConstructors.test_constructor_dict_nan_tuple_keyc C sL t jdd�}t jdd�}||d�}t|d�}t|td�d�}t �||� d S )Nr� )r� r� )rv ru )rn �ba)rn re )r) r� r r� r* )r+ r� r� r� r� r- r. r. r/ �%test_constructor_dict_order_insertion" s z?TestDataFrameConstructors.test_constructor_dict_order_insertionc C sR t tjddgdddgitjdgd�}t ddgddggtjdgd�}t�||� d S )Nr' r1 r2 )re )r r6 r8 r) r* )r+ rF r- r. r. r/ �)test_constructor_dict_nan_key_and_columns- s $ zCTestDataFrameConstructors.test_constructor_dict_nan_key_and_columnsc C sp dddg}t �|�}t||d�}t|�j�� �� s6t�dddg}t �|�}t||d�}t|�j�� �� slt�d S )N)r1 r2 )r2 r2 )r& re )r � from_tuplesr r r9 ra r� r: )r+ �tuples�mir, r. r. r/ � test_constructor_multi_index3 s z6TestDataFrameConstructors.test_constructor_multi_indexc C s� t dggdggddgd�}t ddgtjddgdd�tdggdggd�d�}t�||� t dggdggddggd�}t ddgtddggddggd�tdggdggd�d�}t�||� d S ) Nr' r1 )re r& r# )r4 r )Zlevels�codes)r& re )r �pdZ Int64Indexr r) r* )r+ r, r- r. r. r/ �test_constructor_2d_index@ s z3TestDataFrameConstructors.test_constructor_2d_indexc C sV d}t jt|d�� tt�d�td�d� W d Q R X d}t jt|d�� tddd �ddd gd �� W d Q R X d }t jt|d��. tt�d ��d�dddgt ddd�d� W d Q R X t� dddgg�}d}t jt|d�� tdgt dd�|d� W d Q R X t� dddg�}d}t jt|d�� tdgt dd�|d� W d Q R X t jtd d��" tt� d �d d d gd!gd� W d Q R X d"}t jt|d��& ttj �d#d�d d d gd!gd� W d Q R X d$}t jt|d��& ttj �d#d�d d gd!d#gd� W d Q R X d%}t jt|d�� tt d&�t d&d'�fd(d� W d Q R X d)}t jt|d�� td*d+d �� W d Q R X d S ),Nz)Empty data passed with indices specified.)rK r r )re z<Mixing dicts with non-Series may lead to ambiguous ordering.ru rv )ru rv rw )r� rA z:Shape of passed values is \(4, 3\), indices imply \(3, 3\)� )rx r2 rr r� �bazz 2000-01-01r2 )�periods)re r& rx rd rX z:Shape of passed values is \(1, 3\), indices imply \(1, 4\))r& re rn z:Shape of passed values is \(3, 1\), indices imply \(1, 4\)zMust pass 2-d input)r2 r2 r2 r� rA r� r' z:Shape of passed values is \(2, 3\), indices imply \(1, 3\)r1 z:Shape of passed values is \(2, 3\), indices imply \(2, 2\)z,2 columns passed, passed data had 10 columnsr� � )r| Ztwosz2If using all scalar values, you must pass an indexFT)rO rP rQ r r6 r� r� r\ rM r r7 rf r� r� r� )r+ rS r= r. r. r/ �test_constructor_error_msgsS sD $ &**$z5TestDataFrameConstructors.test_constructor_error_msgsc C s| |dd� t d�D ��|dd� t d�D ��d�}t|�}tdd� |�� D ��}t�||� ||�� �}t|�}t�||� d S )Nc s s | ]}|d | fV qdS )g $@Nr. )rY rZ r. r. r/ rm � s zKTestDataFrameConstructors.test_constructor_subclass_dict.<locals>.<genexpr>r� c s s | ]}|d | fV qdS )g 4@Nr. )rY rZ r. r. r/ rm � s )r~ r c S s i | ]\}}t |�� �|�qS r. )�dict�items)rY �col�valr. r. r/ r� � s zLTestDataFrameConstructors.test_constructor_subclass_dict.<locals>.<dictcomp>)rf r r� r) r* )r+ Z dict_subclassrn r, Zrefdfr. r. r/ � test_constructor_subclass_dict� s z8TestDataFrameConstructors.test_constructor_subclass_dictc C s| ddl m} i }tj|d d d�< x2|�� D ]&\}}|t�}|�|�� � |||< q,W t|�}|j |j d�}t � ||� d S )Nr )� defaultdictrA r� )r& ) � collectionsr� r6 r8 r� r� r� �to_dictr r� r& r) r* ) r+ r� r� rn �k�v�dctr� r- r. r. r/ � test_constructor_defaultdict� s z6TestDataFrameConstructors.test_constructor_defaultdictc C sL t �ddddgg�}tdgdgdgdgd�dddd gd �}t�|j|� d S ) Ng @g @g @g �?)r� rw rv ru r� rw rv ru )re )r6 r7 r r) Zassert_numpy_array_equalr9 )r+ r- r, r. r. r/ �test_constructor_dict_block� s z5TestDataFrameConstructors.test_constructor_dict_blockc C s� ddd�dddd�d�}t |td �}t|�d ks4t�|d jtjksHt�|d jtjks\t�t |�}t|�d kstt�|d jtjks�t�|d jtjks�t�d S ) Nr' r1 )r� r� r� r� �3)r� r� r )r� rA )r4 r2 rA r� )r r� ro r: r4 r6 r rs )r+ � test_datar� r. r. r/ �test_constructor_dict_cast� s z4TestDataFrameConstructors.test_constructor_dict_castc C s� t ttd�t�d���t ttd�tj�d���d�}d}tjt |d�� t |t d�}W d Q R X t |�dksnt �|d jtjks�t �|d jtjks�t �d S ) Nr� � )r� rA zBeither all columns will be cast to that dtype, or a TypeError will)rK )r4 r� rA )r� r� rf r) ZmakeStringIndexr6 r� �randnrz r{ r r� ro r: r4 rs r )r+ r rS r� r. r. r/ �test_constructor_dict_cast2� s z5TestDataFrameConstructors.test_constructor_dict_cast2c C s2 ddt jd�i}t|�}t|d d t�s.t�d S )NZCol1zA String)ZRow1�Row2r )r6 r8 r r` r� r: )r+ r� r, r. r. r/ �!test_constructor_dict_dont_upcast� s z;TestDataFrameConstructors.test_constructor_dict_dont_upcastc C s< t ddgddggddgddgd�}t|d d t�s8t�d S )Nr' r1 ru rv )r& re )r r` r� r: )r+ �dmr. r. r/ �"test_constructor_dict_dont_upcast2� s "z<TestDataFrameConstructors.test_constructor_dict_dont_upcast2c C s< ddd�}t |�}t dd� |�� D ��}tj||dd� d S )N)r' r1 r2 )rx rd rX )ru rv c S s i | ]\}}t |�|�qS r. )r� )rY r� r� r. r. r/ r� � s zMTestDataFrameConstructors.test_constructor_dict_of_tuples.<locals>.<dictcomp>F)� check_dtype)r r� r) r* )r+ rn rF r- r. r. r/ � test_constructor_dict_of_tuples� s z9TestDataFrameConstructors.test_constructor_dict_of_tuplesc C sF t d�t dd�d�}t|�}tdddgdddgd��}t�||� d S ) Nr2 rX )ru rv r r' r1 rx rd )rf r r) r* )r+ rn rF r- r. r. r/ � test_constructor_dict_of_ranges� s z9TestDataFrameConstructors.test_constructor_dict_of_rangesc C sL t td��ttd��d�}t|�}tdddgdddgd��}t�||� d S )Nr2 )ru rv r r' r1 )�iterrf �reversedr r) r* )r+ rn rF r- r. r. r/ �"test_constructor_dict_of_iterators� s z<TestDataFrameConstructors.test_constructor_dict_of_iteratorsc C s\ dd� t d�D �dd� tt d��D �d�}t|�}tdddgdddgd��}t�||� d S )Nc s s | ] }|V qd S )Nr. )rY r� r. r. r/ rm � s zPTestDataFrameConstructors.test_constructor_dict_of_generators.<locals>.<genexpr>r2 )ru rv r r' r1 )rf r r r) r* )r+ rn rF r- r. r. r/ �#test_constructor_dict_of_generators� s *z=TestDataFrameConstructors.test_constructor_dict_of_generatorsc C s dd� }dddd�ddd d�d d d d�d �}t |�� �}t|�}tdd� |D �t�dd� |D ��d�j}t�|j�|_|||� ddddd�|d< |�dd|d f� tdd� |D �tdd� |D �dd�d�j}t|jdd�|_t|�}|j |j |jd�}|||� d S )Nc S s t j| |ddddd�S )NT)r �check_index_type�check_column_typeZ check_names)r) r* )rF r- r. r. r/ �check� s zITestDataFrameConstructors.test_constructor_dict_multiindex.<locals>.checkr r' r1 ))r� r� )r� �j)r r� rX rd rx � � � ))ru ru )rv ru )rv rw c S s g | ] }|d �qS )r' r. )rY rZ r. r. r/ r[ s zNTestDataFrameConstructors.test_constructor_dict_multiindex.<locals>.<listcomp>c S s g | ] }|d �qS )r r. )rY rZ r. r. r/ r[ s )r& g �^@�o )rg )r� r� )r� r )r r� �zc S s g | ] }|d �qS )r' r. )rY rZ r. r. r/ r[ s c S s g | ] }|d �qS )r r. )rY rZ r. r. r/ r[ s F)Z tupleize_cols)re r& ) �sortedr� r r r� �Tr& �insertr r� re )r+ r r� Z_dr, r- r. r. r/ � test_constructor_dict_multiindex� s$ & (z:TestDataFrameConstructors.test_constructor_dict_multiindexc s� ddddg� � fdd�}|t j�}|dd� �}|t�}td d d d d �d d d d d �d d d d d �d d d d d �gdd� � D �d�}t|�}t|�}t|�}t�||� t�||� t�||� d S )Nz 1984-02-19z 1988-11-06z 1989-12-03z 1990-03-15c s � fdd�t ��D �S )Nc s" i | ]\}}� |�d | i|�qS )r1 r. )rY r� �s)ri r. r/ r� s ziTestDataFrameConstructors.test_constructor_dict_datetime64_index.<locals>.create_data.<locals>.<dictcomp>)� enumerate)ri )� dates_as_str)ri r/ � create_data s zUTestDataFrameConstructors.test_constructor_dict_datetime64_index.<locals>.create_datac S s t �| d�S )Nz%Y-%m-%d)r �strptime)rZ r. r. r/ rj rk zRTestDataFrameConstructors.test_constructor_dict_datetime64_index.<locals>.<lambda>r )r r' r1 r2 r1 rx rX c S s g | ] }t |��qS r. )r )rY r% r. r. r/ r[ " s zTTestDataFrameConstructors.test_constructor_dict_datetime64_index.<locals>.<listcomp>)r& )r6 rL r r r) r* ) r+ r" Zdata_datetime64Z data_datetimeZdata_Timestampr- Zresult_datetime64Zresult_datetimeZresult_Timestampr. )r! r/ �&test_constructor_dict_datetime64_index s" z@TestDataFrameConstructors.test_constructor_dict_datetime64_indexc s� ddddg� � fdd�}|dd� �}|d d� �}|d d� �}t d d d d d �d dd d d �d d dd d �d d d d d �gdd� � D �d�}t |�}t |�}t |�}t�||� t�||� t�||� d S )Nr' r1 r2 rx c s � fdd�t ��D �S )Nc s" i | ]\}}� |�d | i|�qS )r1 r. )rY r� r )ri r. r/ r� 1 s zjTestDataFrameConstructors.test_constructor_dict_timedelta64_index.<locals>.create_data.<locals>.<dictcomp>)r )ri )� td_as_int)ri r/ r" 0 s zVTestDataFrameConstructors.test_constructor_dict_timedelta64_index.<locals>.create_datac S s t �| d�S )Nr� )r6 � timedelta64)rZ r. r. r/ rj 3 rk zSTestDataFrameConstructors.test_constructor_dict_timedelta64_index.<locals>.<lambda>c S s t | d�S )N)�days)r )rZ r. r. r/ rj 4 rk c S s t | d�S )Nr� )r )rZ r. r. r/ rj 5 rk r )r r' r1 r2 rX c S s g | ]}t |d ��qS )r� )r )rY �tdr. r. r/ r[ > s zUTestDataFrameConstructors.test_constructor_dict_timedelta64_index.<locals>.<listcomp>)r& )r r) r* ) r+ r" Zdata_timedelta64Zdata_timedeltaZdata_Timedeltar- Zresult_timedelta64Zresult_timedeltaZresult_Timedeltar. )r% r/ �'test_constructor_dict_timedelta64_index, s" zATestDataFrameConstructors.test_constructor_dict_timedelta64_indexc C s� t jdddgdd�}t jdddgdd�}t||d ��}|d j|jksJt�|d j|jks^t�t|�t��� |�t��� d ��}|d j|jks�t�|d j|jks�t�d S ) Nz2012-01rH z2012-04rW )�freqz 2012-02-01z 2012-03-01r� )ru rv ru rv )r� Z PeriodIndexr r4 r: r� rB rD )r+ ru rv r, r. r. r/ � test_constructor_period_dictH s "z6TestDataFrameConstructors.test_constructor_period_dictc C sN |\}}t d|idgd�}|d j|ks,t�t dgdg|d�}t�||� d S )Nru r )r& )r& re rn )r r4 r: r) r* )r+ Zea_scalar_and_dtypeZ ea_scalarZea_dtyper, r- r. r. r/ �&test_constructor_dict_extension_scalarU s z@TestDataFrameConstructors.test_constructor_dict_extension_scalarz data,dtypez2020-01rd )�left�rightr# r. z 2011-01-01z US/Eastern)�tzc C sl t ddgddg|d�}|d j|ks(t�|d j|ks:t�tj|gd |d�}t ||d��}t�||� d S ) Nr r' ru rv )r& re rn r1 )r4 )ru rv )r r4 r: r� r7 r) r* )r+ rn r4 r, r= r- r. r. r/ �&test_constructor_extension_scalar_data^ s z@TestDataFrameConstructors.test_constructor_extension_scalar_datac C s� t jddd�}ttj�dd�|d�}i }x6|jD ],}x&|jD ] }|�||�|� |i �|< q<W q0W t||d�}t � ||� i }x6|jD ],}x&|jD ] }|�||�|� |i �|< q�W q�W t||d�j }t � ||� d S )Nz1/1/2000rd )r� r� )re )r& ) r� � period_ranger r6 r� r re r& r� � setdefaultr) r* r )r+ r� r, rn r� �rowrF r. r. r/ �"test_nested_dict_frame_constructoru s z<TestDataFrameConstructors.test_nested_dict_frame_constructorc C sj |dt d�}t|dddgddgd�}t|j�dks6t�t|j�d ksHt�t|d �dgddd gd�}t|j�d kstt�t|j�dks�t�t|dddgddgtjd �}|tjkr�|j j tjks�t�n8|� � � � � � s�t�|j j tj ks�t�t |j �� � s�t�d }tjt|d �� t|dddgdgd� W d Q R X d}tjt|d �� t|ddgddgd� W d Q R X tjtdd �� t|d�dddgdgd� W d Q R X t|�}tj|jttd��dd� tj|jttd ��dd� t|ddgd�}tj|jttd ��dd� t|dddgd�}tj|jttd��dd� t|d��}t|j�dk�sFt�t|d��}t|j�dk�sft�d S )N)r1 r2 )r4 r� rA r� r' r1 )re r& r2 )r2 )re r& r4 z:Shape of passed values is \(2, 3\), indices imply \(1, 3\))rK z:Shape of passed values is \(2, 3\), indices imply \(2, 2\)zMust pass 2-d input)r2 r2 r2 T)r� )r& )re )r r2 r )r2 r )r� r ro r& r: re r6 r# r| r9 r4 r r� r rO rP rQ r) r� r rf )r+ r� �matr� rS r. r. r/ �_check_basic_constructor� s@ $ z2TestDataFrameConstructors._check_basic_constructorc C s8 | � tj� tddgddgdgd�}t|�dks4t�d S )Nrr r� r r' r� )r& re r1 )r6 r6 r| r ro r: )r+ r� r. r. r/ �test_constructor_ndarray� s z2TestDataFrameConstructors.test_constructor_ndarrayc C s� | � tj� tjdtd�}d|d< d|d< t|ddd gd d gd �}d|d d ksVt�d|d d ksjt�tjdtd�}t|ddd gd d gd �}t�t�||k� �s�t�d S ) N)r1 r2 )r4 g �?)r r g @)r' r1 r� rA r� r' r1 )re r& ) r6 �ma� masked_allr� r r: r6 r� �asarray)r+ r5 r� r. r. r/ � test_constructor_maskedarray� s z6TestDataFrameConstructors.test_constructor_maskedarrayc C s" t jdtd�}t|dddgddgd�}t|j�dks8t�t|j�d ksJt�t� t� ||k� �sdt�t|dddgddgtj d �}|j j tj ks�t�t �|�}d|d < d|d < t|dddgddgd�}d|d d ks�t�d|d d ks�t�t jdd d�}t|dddgddgd�}t|j�dk�s&t�t|j�d k�s:t�t|�j � � �sNt�d}tjt|d��F t�� �2 tjdtdd� t|dddgddgtjd �}W d Q R X W d Q R X |j j tjk�s�t�t �|�}d|d < d|d < t|dddgddgd�}d|d �d�d k�st�d|d �d�d k�s*t�t jdtd�}t|dddgddgd�}t|j�dk�sdt�t|j�d k�sxt�t� t� ||k� ��s�t�t|dddgddgtd �}|j j tk�s�t�t �|�}d|d < d|d < t|dddgddgd�}|d d dk�st�|d d dk�s t�d S )N)r1 r2 )r4 r� rA r� r' r1 )re r& r2 )re r& r4 )r r )r' r1 zM8[ns]z'datetime64\[ns\] values and dtype=int64)rK �ignorez elementwise comparison failed)�category�messager3 TF)r8 r9 r� r ro r& r: re r6 r� r: r r9 r4 r� r r) rz r{ �warnings�catch_warnings�filterwarnings�DeprecationWarningr# r] r� rB )r+ r5 r� Zmat2rS r. r. r/ �%test_constructor_maskedarray_nonfloat� s^ . z?TestDataFrameConstructors.test_constructor_maskedarray_nonfloatc C s� t jdtd��� }t|ddgddgd�}ttjtjgtjtjgd�ddgddgtd �}t�||� t j dtd��� }t|ddgddgd�}td d gd d gd�ddgddgtd �}t�||� d S ) N)r1 r1 )r4 r� rA r' r1 )re r& )r� rA )re r& r4 g �?) r8 r9 r� Z harden_maskr r6 r8 r) r* r| )r+ Zmat_hardrF r- r. r. r/ �%test_constructor_maskedarray_hardened" s z?TestDataFrameConstructors.test_constructor_maskedarray_hardenedc s� t jjt jjdddgd�dgd d�� � �tj�� t�t �� t � t d�}W d Q R X t t jdt d�dd gd �}t� ||� t � fd d �� j jD �t d�}t� ||� d S ) Nrd )r z<f8)�pricez<f8)r4 F)�mask)rd r1 r rE )re c s i | ]}� | |�qS r. r. )rY r5 )rn r. r/ r� H s zSTestDataFrameConstructors.test_constructor_maskedrecarray_dtype.<locals>.<dictcomp>)r6 r8 r7 r� r] �mrecordsZ mrecarrayr) rz r{ r r� r* r4 r� )r+ rF r- Zaltr. )rn r/ �%test_constructor_maskedrecarray_dtype8 s z?TestDataFrameConstructors.test_constructor_maskedrecarray_dtypec C s� t jtjddd�}dt�ddg�fdt�ddg�fd t�d d g�fg}x:|d d � D ]*\}}|�d | tjj|d dgd�f� qTW |�dtj� d�f� |�dtjjddgd d�f� x�t � |d�D ]�}t |� \}}t j||d�}dd� |D �}t�t�� t|�} W d Q R X t||d�} || | � t�t�� t||d d d� d�} W d Q R X t||d d d� d�} || | � t�t�� t|ddgd�} W d Q R X t||ddgd�} || | � q�W d S ) NT)r r r� g �?g @r� r' r1 �strr �defZmasked1_F)rF r9 )r1 Z masked_noneg �?g @r2 )r� c S s( i | ] \}}t |d �r |�� n||�qS )�filled)�hasattrrK )rY r� r� r. r. r/ r� e s zHTestDataFrameConstructors.test_constructor_mrecarray.<locals>.<dictcomp>)re r� )r& )re r& )� functools�partialr) r* r6 r7 �appendr8 Z masked_arrayr9 � itertools� combinationsr� rG Z fromarraysrz r{ r ) r+ Zassert_fr_equalr_ r5 r= Zcombr� rn ZmrecsrF r- r. r. r/ �test_constructor_mrecarrayK s4 " z4TestDataFrameConstructors.test_constructor_mrecarrayc C s t g d�}|jjdkst�d S )N)r& )r r )r r9 �shaper: )r+ r, r. r. r/ � test_constructor_corner_shape{ s z7TestDataFrameConstructors.test_constructor_corner_shapez%data, index, columns, dtype, expectedr� ru rv r rr r� r c C s" t ||||�}|jj|ks t�d S )N)r r9 r4 r: )r+ rn r& re r4 r- r, r. r. r/ �test_constructor_dtype s z0TestDataFrameConstructors.test_constructor_dtypez data,input_dtype,expected_dtypeTF�booleang �?g @ZFloat64ZInt64rw �stringc C s( t d|i|d�}|d j|� ks$t�d S )Nru )r4 )r r4 r: )r+ rn Z input_dtypeZexpected_dtyper, r. r. r/ �0test_constructor_dtype_nullable_extension_arrays� s zJTestDataFrameConstructors.test_constructor_dtype_nullable_extension_arraysc C s� dddddd�}t |t�d�d�}|d jtjks6t�|d jtjksJt�|d jtjks^t�|d jtjksrt�|d jtj ks�t�d S )Nr' Tg @y @rr )r� r� r� �complexrB r� )r& r� r� r� rY rB ) r r6 r\ r4 r# r: r� r r� rs )r+ rn r, r. r. r/ �!test_constructor_scalar_inference� s z;TestDataFrameConstructors.test_constructor_scalar_inferencec C sh t tj�d�dd��}t |d jdgd d��}t�||� tjt dd�� t ddd�� W d Q R X d S )Nr� T)ru rv ru zmust pass an index)rK F) r r6 r� r r9 r) r* rO rP rQ )r+ r, r� r. r. r/ �#test_constructor_arrays_and_scalars� s z=TestDataFrameConstructors.test_constructor_arrays_and_scalarsc C s8 t |�}t�||� t |tjd�}|jjtjks4t�d S )N)r4 )r r) r* r6 r# r9 r4 r: )r+ r� r, Z df_castedr. r. r/ �test_constructor_DataFrame� s z4TestDataFrameConstructors.test_constructor_DataFramec C s� t j�d�}t|dgt �d�d�}|jjdks2t�t j�d�}t|�}|jjdksVt�|jjdksft�tddgt �d�d�}|jjdks�t�tddgd�}|jjd ks�t�tt �d�d �}|jjd ks�t�t j d d gt d�� dd�}d}t j t|d�� t|ddgdgtd� W d Q R X tt|j��}t�||� tt jdtd�t jdt jd�d�t �d�d �}t|j�dk�stt�|jjt jk�s�t�d S )Nr� r� )re r& r1 r rA )r� r1 )re )r r1 )r& )r� r rr r� )r4 r' z(could not convert string to float: 'foo')rK )r& re r4 )r� rA )r6 r� r r r\ r9 �ndimr: rS r7 rB rM rO rP rQ r� r� r) r* r| r� r ro re r4 )r+ r� r= r r5 rS r. r. r/ �test_constructor_more� s0 z/TestDataFrameConstructors.test_constructor_morec C sx t g g d�}t g d�}t�||� t g ddgd�}t i ddgd�}t�||� dd� }t |� ddgd�}t�||� d S )N)r& r� rA )re c s s d S )Nr. r. r. r. r/ � empty_gen� s zHTestDataFrameConstructors.test_constructor_empty_list.<locals>.empty_gen)r r) r* )r+ r, r- r_ r. r. r/ �test_constructor_empty_list� s z5TestDataFrameConstructors.test_constructor_empty_listc C s| t ddgddggddgd�}t|d �s,t�|d jtjks@t�t dt�d �i�}d d � td �D �}t |�}t� ||� d S ) Nr' ru r1 rv �numrI )rn re r r� c S s g | ]}t �|��qS r. )r6 r7 )rY rZ r. r. r/ r[ � s zLTestDataFrameConstructors.test_constructor_list_of_lists.<locals>.<listcomp>) r r r: r4 r6 rs r\ rf r) r* )r+ r, r- rn rF r. r. r/ � test_constructor_list_of_lists� s z8TestDataFrameConstructors.test_constructor_list_of_listsc C sj t d�t d�g}tddddgdddd gg|d �}t�|�}tddddgdddd gg|d �}t�||� d S ) N�abcdZcdefr' r1 r2 rx rd rX r )re )r� r r Z from_arraysr) r* )r+ r_ rF r� 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}tjt|d��& tddddgdd d d gg|d � W d Q R X d S ) Nr �cdez#3 columns passed, passed data had 4)rK r' r1 r2 rx rd rX r )re )r� rO rP rQ r )r+ r_ rS r. r. r/ �0test_constructor_wrong_length_nested_list_column s zJTestDataFrameConstructors.test_constructor_wrong_length_nested_list_columnc C sR t d�t d�g}d}tjt|d��& tddddgdd d d gg|d � W d Q R X d S ) Nrc re z all arrays must be same length)rK r' r1 r2 rx rd rX r )re )r� rO rP rQ r )r+ r_ rS r. r. r/ �2test_constructor_unequal_length_nested_list_column s zLTestDataFrameConstructors.test_constructor_unequal_length_nested_list_columnrn z 2021-01-01rZ c C s< t |dddgdgd�}t dtd�gd i�}t�||� d S )Nr r' r1 rZ )r& re z 2021-01-01r2 )r r r) r* )r+ rn rF r- r. r. r/ �&test_constructor_one_element_data_list s z@TestDataFrameConstructors.test_constructor_one_element_data_listc C sj G dd� dt j�}|ddg�|ddg�g}ddg}t||d �}tddgddgg|d �}tj||d d � d S ) Nc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )zPTestDataFrameConstructors.test_constructor_sequence_like.<locals>.DummyContainerc S s || _ d S )N)�_lst)r+ �lstr. r. r/ �__init__* s zYTestDataFrameConstructors.test_constructor_sequence_like.<locals>.DummyContainer.__init__c S s | j �|�S )N)ri � __getitem__)r+ �nr. r. r/ rl - s z\TestDataFrameConstructors.test_constructor_sequence_like.<locals>.DummyContainer.__getitem__c S s | j �� S )N)ri �__len__)r+ rm r. r. r/ rn 0 s zXTestDataFrameConstructors.test_constructor_sequence_like.<locals>.DummyContainer.__len__N)�__name__� __module__� __qualname__rk rl rn r. r. r. r/ �DummyContainer) s rr r' ru r1 rv ra rI )re F)r )r �Sequencer r) r* )r+ rr Zlst_containersre rF r- r. r. r/ � test_constructor_sequence_like% s z8TestDataFrameConstructors.test_constructor_sequence_likec C s� dd l }td|� dtd��i�}tdttd��i�}tj||dd� tttd��ttd��g�}t|� dtd��|� dtd��g�}tj||dd� d S )Nr r� r� r� F)r )r7 r rf r� r) r* )r+ r7 rF r- r. r. r/ � test_constructor_stdlib_array9 s $z7TestDataFrameConstructors.test_constructor_stdlib_arrayc C s, t td��}t ttd���}t�||� d S )Nr� )r rf r� r) r* )r+ rF r- r. r. r/ �test_constructor_rangeF s z0TestDataFrameConstructors.test_constructor_rangec C s@ t td�td�g�}t ttd��ttd��g�}t�||� d S )Nr� )r rf r� r) r* )r+ rF r- r. r. r/ � test_constructor_list_of_rangesL s z9TestDataFrameConstructors.test_constructor_list_of_rangesc C s< G dd� d�}t dddggd �}t |� �}t�||� d S )Nc @ s e Zd Zdd� ZdS )zATestDataFrameConstructors.test_constructor_iterable.<locals>.Iterc s s" x t d�D ]}dddgV q W d S )Nr� r' r1 r2 )rf )r+ r� r. r. r/ �__iter__T s zJTestDataFrameConstructors.test_constructor_iterable.<locals>.Iter.__iter__N)ro rp rq rx r. r. r. r/ �IterS s ry r' r1 r2 r� )r r) r* )r+ ry r- rF r. r. r/ �test_constructor_iterableQ s z3TestDataFrameConstructors.test_constructor_iterablec C s0 t ttd���}t ttd���}t�||� d S )Nr� )r r rf r� r) r* )r+ rF r- r. r. r/ �test_constructor_iterator\ s z3TestDataFrameConstructors.test_constructor_iteratorc C sH t ttd��ttd��g�}t ttd��ttd��g�}t�||� d S )Nr� )r r rf r� r) r* )r+ rF r- r. r. r/ �"test_constructor_list_of_iteratorsa s z<TestDataFrameConstructors.test_constructor_list_of_iteratorsc C s� dd� t d�D �}dd� t d�D �}ttt d��tt d��g�}t||g�}t�||� dd� t d�D �}t|�}tt d�dd��}tj||dd � d S ) Nc s s | ] }|V qd S )Nr. )rY r� r. r. r/ rm i s zGTestDataFrameConstructors.test_constructor_generator.<locals>.<genexpr>r� c s s | ] }|V qd S )Nr. )rY r� r. r. r/ rm j s c s s | ]}|d gV qdS )ru Nr. )rY r� r. r. r/ rm p s ru )r r' F)r )rf r r� r) r* )r+ Zgen1Zgen2r- rF �genr. r. r/ �test_constructor_generatorf s z4TestDataFrameConstructors.test_constructor_generatorc C s& t i g�}t dgd�}t�||� d S )Nr )r& )r r) r* )r+ rF r- r. r. r/ � test_constructor_list_of_dictsu s z8TestDataFrameConstructors.test_constructor_list_of_dicts� dict_typec C st t ddggddgd�}|� }dg|d< dg|d< t |�}t�||� |� }d|d< d|d< t |g�}t�||� d S )Nr1 r' rv ru )re )r r) r* )r+ r� r- rn rF 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< ddd�}t ddgddggddgd�}t ||g�}t�||� t ddgddgddggddgd�}t |||g�}t�||� d S )Nr1 rv r' ru )rv ru )re )r r) r* )r+ r� Zrow_oneZrow_twoZ row_threer- rF r. r. r/ �0test_constructor_ordered_dict_conflicting_orders� s "zJTestDataFrameConstructors.test_constructor_ordered_dict_conflicting_ordersc C s^ dd� t d�D �}t|�}tdddgdddgdddgd�dd d gd d d gd�}t�||� d S )Nc S s$ g | ] }t |d ddgt|�d��qS )rv ru rw )r& r5 )r rI )rY r� r. r. r/ r[ � s z[TestDataFrameConstructors.test_constructor_list_of_series_aligned_index.<locals>.<listcomp>r2 r r' r1 )rv ru rw rv ru rw �0r� r� )re r& )rf r r) r* )r+ �seriesrF r- r. r. r/ �-test_constructor_list_of_series_aligned_index� s zGTestDataFrameConstructors.test_constructor_list_of_series_aligned_indexc C sJ G dd� dt �}ddd�}||�g}|g}t|�}t|�}t�||� d S )Nc @ s e Zd ZdS )zTTestDataFrameConstructors.test_constructor_list_of_derived_dicts.<locals>.CustomDictN)ro rp rq r. r. r. r/ � CustomDict� s r� g �?r2 )ru rv )r� r r) r* )r+ r� r� Z data_customrn Z result_customrF 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�}tjtdd�� t|� W d Q R X d S )Nr� r )r� rA z%All arrays must be of the same length)rK )r6 r� r rO rP rQ r )r+ rn r. r. r/ �test_constructor_ragged� s z1TestDataFrameConstructors.test_constructor_raggedc C sF t td��}tddi|d�}tddddgi|d�}tj||dd� d S )Nr2 ru r )r& F)r )r rf r r) r* )r+ rE r, r- r. r. r/ �test_constructor_scalar� s z1TestDataFrameConstructors.test_constructor_scalarc C s" t |d |jdgd�}|�� d S )Nr� )r& re )r r& r� )r+ r� r, r. r. r/ � test_constructor_Series_copy_bug� s z:TestDataFrameConstructors.test_constructor_Series_copy_bugc C s� i }dddd�|d< t ddddgddd d gd �|d < t|�}|jjsHt�tjtd d�� tddgddd�d�� W d Q R X tddgt ddgddgd �d��}tddgddgd�ddgd �}t� ||� d S )Nr' r1 r2 )rr r� r� r� rx r� Zquxr� rr )r& rA zambiguous ordering)rK ru rv )ru rv )r� rA ) r r r& Z is_monotonicr: rO rP rQ r) r* )r+ rn rF r- r. r. r/ �&test_constructor_mixed_dict_and_Series� s "" z@TestDataFrameConstructors.test_constructor_mixed_dict_and_Seriesc C s8 ddgdg}t |�}t ddgddgg�}t�||� d S )Nr' r1 )r2 rx r2 rx )r r) r* )r+ rn rF r- r. r. r/ � test_constructor_mixed_type_rows� s z:TestDataFrameConstructors.test_constructor_mixed_type_rowsz tuples,lists)r. r. r2 rx rX c C s t |�}t |�}t�||� d S )N)r r) r* )r+ r� �listsrF r- r. r. r/ �test_constructor_tuple� s z0TestDataFrameConstructors.test_constructor_tuplec C s4 t dddgi�}t dtddg�i�}t�||� d S )Nr� )r' r1 )r2 rx )r r r) r* )r+ rF r- r. r. r/ � test_constructor_list_of_tuples� s z9TestDataFrameConstructors.test_constructor_list_of_tuplesc C s� ddl m} |dtd��}|dd�|dd�g}tddgddgd ��}t|�}t�||� tddgddgd ��}t|d d gd �}t�||� d S )Nr )� namedtupleZPandas�abr' r2 r1 rx )ru rv )rg r rg r )re )r� r� r� r r) r* )r+ r� Z named_tupler� r- rF r. r. r/ �$test_constructor_list_of_namedtuples� s z>TestDataFrameConstructors.test_constructor_list_of_namedtuplesc C sd ddl m} |ddtfdtfg�}|dd�|dd�g}tddgddgd��}t|�}t�||� d S ) Nr )�make_dataclass�PointrZ rg r2 r' )rZ rg )� dataclassesr� r� r r) r* )r+ r� r� rn r- rF r. r. r/ �$test_constructor_list_of_dataclasses s z>TestDataFrameConstructors.test_constructor_list_of_dataclassesc C s� ddl m} |ddtfdtfg�}|ddtfdtfdtfg�}|dd �|d d d �g}tdtjgd d gtjd gtjd gd ��}t|�}t�||� d S ) Nr )r� r� rZ rg �HLine�x0�x1r2 r' )rZ rg r� r� )r� r� r� r r6 r8 r) r* )r+ r� r� r� rn r- rF r. r. r/ �7test_constructor_list_of_dataclasses_with_varying_types s &zQTestDataFrameConstructors.test_constructor_list_of_dataclasses_with_varying_typesc C sb ddl m} |ddtfdtfg�}d}tjtt�|�d�� t|dd�ddd �g� W d Q R X d S ) Nr )r� r� rZ rg z0asdict() should be called on dataclass instances)rK r' )rZ rg ) r� r� r� rO rP � TypeError�re�escaper )r+ r� r� rS r. r. r/ �1test_constructor_list_of_dataclasses_error_thrown# s zKTestDataFrameConstructors.test_constructor_list_of_dataclasses_error_thrownc C s~ ddddd�dddd d �d d d dddd�g}t ddd gddd gdd dgddd gd d dgd d dgd��}t |�}t�||� d S )Nr' rx r r� )�First�Second�Third�Fourthrd r1 r� r )r� r� r� r� rX r2 r� r � r� )r� r� r� r� �YYY�XXX)r� r� r� r� r� r� )r r) r* )r+ rn r- rF r. r. r/ �#test_constructor_list_of_dict_order. s z=TestDataFrameConstructors.test_constructor_list_of_dict_orderc C sv t dddgdddgdd�}t|�}|jd dks4t�t�|j|j� tj� d �}t |dd �}t|�}td|i�}t� ||� t |t dd �d �}t|�}td |i�}t� ||� d}t j t|d�� t|ddgd� W d Q R X t g dtd�}t|�}|jd dk�st�t |dd �}t||g�j}t||d�ddgd�}t� ||� t||g�j}t||d�d dgd�}t� ||� d S )Nr' r1 r2 ru rv rw rZ )r& r5 r r� )r5 r� )r& z<Shape of passed values is \(10, 1\), indices imply \(10, 2\))rK )re )r5 r4 )rZ z Unnamed 0z Unnamed 0)r' r )r r re r: r) r� r& r6 r� r r* rf rO rP rQ rB r )r+ ru r, r= r r- rS �s1r. r. r/ � test_constructor_Series_namedB s4 z7TestDataFrameConstructors.test_constructor_Series_namedc C s� t td�dd�}t td�dd�}t�t|dgd�|�� � t�t|dgd�|�� � t|dgd�jsht�t|dgd�js|t�d S )Nrd r )r5 r' )re )r rf r) r* r �to_framer� r: )r+ �s0r� r. r. r/ �)test_constructor_Series_named_and_columnsh s zCTestDataFrameConstructors.test_constructor_Series_named_and_columnsc C s� t dddgdddgdd�}t dddgdddgd �}tddg�}t||d �}t|�|��}|jd dksjt�t�||� t||d �}t|�|��}|jd d ks�t�t�|j |� t�||� d S ) Nr' r1 r2 ru rv rw rZ )r& r5 )r& r ) r r r r� re r: r) r* r� r& )r+ r� �s2Z other_indexZdf1Zexp1r� Zexp2r. r. r/ �+test_constructor_Series_differently_indexedv s zETestDataFrameConstructors.test_constructor_Series_differently_indexedz#name_in1,name_in2,name_in3,name_out)rE rE rE rE )rE rE NN)rE NNN)�idx1�idx2NN)r� r� r� N)r� r� Zidx3N)NNNNc C s� t dddg|d�t dddg|d�t dddg|d�g}dd� t|d d d g�D �}t|�}t dddddg|d�}td d dtjtjgtjd d dtjgtjtjd d dgd�|d�} t�|| � d S )Nru rv rw )r5 r� �ec S s$ i | ] \}}t d ddg|d�|�qS )r r' r1 )r& )r )rY r� rw r. r. r/ r� � s zJTestDataFrameConstructors.test_constructor_index_names.<locals>.<dictcomp>rZ rg r r r' r1 )rZ rg r )r& )r r� r r6 r8 r) r* ) r+ Zname_in1Zname_in2Zname_in3Zname_out�indicesr� rF Zexp_indr- r. r. r/ � test_constructor_index_names� s z6TestDataFrameConstructors.test_constructor_index_namesc C s\ t |jd d� �}t |jd d� �}t|j||d�}t�|jt|�� t�|jt|�� d S )Nrd r2 )r& re )r� r& re r r^ r) r� r )r+ r� r& re rF r. r. r/ � test_constructor_manager_resize� s z9TestDataFrameConstructors.test_constructor_manager_resizec C s� t |d t|d �d�ddgd�}t�||jd d �ddgf � d}tjt|d��( t |d t|d �d d� d�� W d Q R X d S )Nr� rA )r� rA )re zdoes not match index length)rK ���)r r� r) r* r� rO rP rQ )r+ r� r, rS r. r. r/ �%test_constructor_mix_series_nonseries� s z?TestDataFrameConstructors.test_constructor_mix_series_nonseriesc C sB t tjdgddggtjd�}t tjdgddgg�}t�||� d S )Nr' r )r4 )r r6 r8 r# r) r* )r+ r, r- r. r. r/ �%test_constructor_miscast_na_int_dtype� s z?TestDataFrameConstructors.test_constructor_miscast_na_int_dtypec C s^ t ddggddgd�}t ddgg�}ddg|_t�||� t jdgddgd�}t�||� d S )Nr rd ru )re )r rd )r re r) r* Z from_records)r+ r, ZedfZidfr. r. r/ �"test_constructor_column_duplicates� s z<TestDataFrameConstructors.test_constructor_column_duplicatesc C s� t ddgddgtd�}t ddgddgtd�}t�||� t ddgddgtjd�}t�||� t ddgddgtjd�}t�||� t ddgddgdd�}t�||� d S )Nr r' )r& re r4 ZU5)r rB rI r) r* r6 Zstr_Zunicode_)r+ r- r, r. r. r/ �(test_constructor_empty_with_string_dtype� s zBTestDataFrameConstructors.test_constructor_empty_with_string_dtypec C s. t g dg|d�}t dg|d�}t�||� d S )N�c1)r& re r4 )re r4 )r r) r* )r+ Znullable_string_dtyper- r, r. r. r/ �,test_constructor_empty_with_string_extension� s zFTestDataFrameConstructors.test_constructor_empty_with_string_extensionc C sX t ddddgdddgd�}t�|t t�|j��d �|j|j�� t d dddgdddgd�}t�|t t�|j��d �|j|j�� t dddgddgd�}t�|t tj ddgddggt d �ddgddgd�� d }t j t |d�� t dddg� W d Q R X t j t |d�� t dddgd� W d Q R X d}t j t|d�� t dddgddgt� W d Q R X d S )Ng r' r1 r2 ru rv rw )r& re r r r# )r4 z*DataFrame constructor not properly called!)rK )re zincompatible data and dtype)r r) r* r6 r� rS r� r& re r7 rB rO rP rQ r� r� )r+ r, rS r. r. r/ � test_constructor_single_value� s* ""z7TestDataFrameConstructors.test_constructor_single_valuec C s� t �t j�j}t �t j�j}t �d�j}t �t j�j}tdddtd�tddddd�d �t � d �d �}|j }t t �d �gt �|�gd t �|�gd t d �d �}t �||� tdddddd|t jd|d�|t jd|d�it � d �d �}|j }t t �d�gt �d �g t �d�g t �d�g t �|�g ddd||gd �}t �||� tdddddd|t jdgd |d�|t jdgd |d�it � d �d �}|j }t t �d�gt �d �g t �d�g t �d�g t �|�g ddd||gd �}t �||� d S )NzM8[ns]r' rr r� Z20010101i� r1 r )r� rA r� r� �Er� )r& r# ZABCDEru g �?rv rw )r4 r rB )r6 r4 �int_r5 Zfloat_rs r r r r\ r� r r� r) r< r7 )r+ ZintnameZ floatnameZdatetime64nameZ objectnamer, rF r- r. r. r/ � test_constructor_with_datetimes sP * : :z9TestDataFrameConstructors.test_constructor_with_datetimesc C s6 t dddd�}dd� |D �}t|�}|jdks2t�d S )Nz 2000-01-01r� r� )�startr* r� c S s g | ] }|� � �qS r. )� to_pydatetime)rY �tsr. r. r/ r[ O s zNTestDataFrameConstructors.test_constructor_with_datetimes1.<locals>.<listcomp>zM8[ns])r r r4 r: )r+ �ind� datetimesZ datetime_sr. r. r/ � test_constructor_with_datetimes1K s z:TestDataFrameConstructors.test_constructor_with_datetimes1c C sv t dddd�}dd� |D �}dd� |D �}t|dgd �}||d < |j}tt�d �t�d �gdd gd �}t�||� d S )Nz 2000-01-01r� r� )r� r* r� c S s g | ] }|� � �qS r. )r� )rY r� r. r. r/ r[ V s zNTestDataFrameConstructors.test_constructor_with_datetimes2.<locals>.<listcomp>c S s g | ] }|� � �qS r. )r )rY r� r. r. r/ r[ W s r� )re �dateszdatetime64[ns]rB )r& )r r r� r r6 r4 r) r< )r+ r� r� r� r, rF r- r. r. r/ � test_constructor_with_datetimes2S s z:TestDataFrameConstructors.test_constructor_with_datetimes2c C s� t �d�}|�tddd��}td|idgd�}|jd |ks@t�t�|j t ddi�� td|ig�}|jd |ksvt�t�|j t ddi�� d S ) Nz US/Easterni� r' zEnd Dater )r& )r r zdatetime64[ns, US/Eastern]) �pytz�timezoneZlocalizer r �iatr: r) r< r� r )r+ r/ r% r, r. r. r/ � test_constructor_with_datetimes3a s z:TestDataFrameConstructors.test_constructor_with_datetimes3c C s� t ddd�}td|i�}|jd jd ks,t�t dddd�}td|i�}t|jd j�dks^t�t dddd�}td|i�}t|jd j�dks�t�d S ) N�20130101r2 )r� r( )r r �UTC)r� r/ z US/Eastern)r r r� r/ r: rI )r+ �drr, r. r. r/ � test_constructor_with_datetimes4s s z:TestDataFrameConstructors.test_constructor_with_datetimes4c C s^ t ddddd�}td|�� jdd�i�}t� }||d< t�||� td|i�}t�||� d S ) Nz1/1/2011rd �10sz US/Eastern)r� r* r/ ru T)�drop)r r � to_series� reset_indexr) r* )r+ r� r- r, r. r. r/ � test_constructor_with_datetimes5� s z:TestDataFrameConstructors.test_constructor_with_datetimes5c C sV t ddddd�}t dddd�}t||d��}t|�� jdd �|d��}t�||� d S ) Nz1/1/2011rd r� z US/Eastern)r� r* r/ )r� r* )ru rv T)r� )r r r� r� r) r* )r+ r� Zi_no_tzr, r- r. r. r/ � test_constructor_with_datetimes6� s z:TestDataFrameConstructors.test_constructor_with_datetimes6r= rH c C s* t |�j}tt�d�g�}t�||� d S )Nzdatetime64[ns])r r� r r6 r4 r) r< )r+ r= rF r- r. r. r/ �%test_constructor_datetimes_with_nulls� s z?TestDataFrameConstructors.test_constructor_datetimes_with_nulls�order�Kr� r� �Fr4 z datetime64[M]z datetime64[D]z datetime64[h]z datetime64[m]z datetime64[s]zdatetime64[ms]zdatetime64[us]zdatetime64[ns]c C s\ t jdddgdddgg||d�}t|�}tdddgdddgg�}|j|d�}t�||� d S ) Nz 2015-01-01z 2015-01-02z 2015-01-03z 2017-01-01z 2017-01-02z 2017-02-03)r4 r� )r4 )r6 r7 r r� r) r* )r+ r� r4 �nar, r- r. r. r/ �!test_constructor_datetimes_non_ns� s z;TestDataFrameConstructors.test_constructor_datetimes_non_nsztimedelta64[D]ztimedelta64[h]ztimedelta64[m]ztimedelta64[s]ztimedelta64[ms]ztimedelta64[us]ztimedelta64[ns]c C s� t jt �dd�t �dd�gt �dd�t �dd�gg||d�}t|��d�}ttdd�tdd�gtdd�tdd�gg�}t�||� d S )Nr' r� r1 rx rd )r4 r� ztimedelta64[ns])r6 r7 r& r r� r r) r* )r+ r� r4 r� r, r- r. r. r/ �!test_constructor_timedelta_non_ns� s z;TestDataFrameConstructors.test_constructor_timedelta_non_nsc C s2 t dd� td�D ��}|j}tt�d�gd �}t�||� t dd� td�D ��}|j}tt�d�gd �}t�||� t ddd gi�}|jjd t�d �ks�t �t d d g�}|jjd t�d �ks�t �t ddg�}|jjd t�d�ks�t �t dd d gi�}|jjd t�d �k�st �t dddgi�}|jjd t�d�k�sFt �t dd itd�d�}|jjd t�d �k�svt �t dditd�d�}|jjd t�d�k�s�t �t d d ddgddddgt d�dd� td�D �dd ddgd ��}|j}tt�d �t�d�t�d �t�d �t�d�gt d �d�}t�||� d S ) Nc S s g | ]}t �d ��qS )rd )r6 r\ )rY rZ r. r. r/ r[ � s zSTestDataFrameConstructors.test_constructor_for_list_with_dtypes.<locals>.<listcomp>rd r� c S s g | ]}t jt �d �dd��qS )rd r" )r4 )r6 r7 r\ )rY rZ r. r. r/ r[ � s r" ru l l r r# r' r1 g �?g @r r2 )r& rx r g333333�?gffffff@gffffff@g333333@rc c S s g | ]}t d dd��qS )i� r' )r )rY r� r. r. r/ r[ s g @)ru rv rw r� r� rB zdatetime64[ns]Zabcde) r rf r� r r6 r4 r) r< r; r: r� )r+ r, rF r- r. r. r/ �%test_constructor_for_list_with_dtypes� sJ z?TestDataFrameConstructors.test_constructor_for_list_with_dtypesc C s@ t |dd�}d|d< |d dk�� s(t�|d dk�� r<t�d S )NT)r� rd r� )r r� r: )r+ r� �copr. r. r/ �test_constructor_frame_copy# s z5TestDataFrameConstructors.test_constructor_frame_copyc C s\ t |j�}d|jd< |jd dk�� s*t�t |jdd�}d|jd< |jd dk�� rXt�d S )Nrd T)r� rX )r r9 r� r: )r+ r� r, r. r. r/ � test_constructor_ndarray_copy* s z7TestDataFrameConstructors.test_constructor_ndarray_copyc C sB |j }td|d idd�}d|d d d �< |d dk�� r>t�d S )Nr� T)r� rd )r� r r� r: )r+ r� r� r, r. r. r/ � test_constructor_series_copy6 s z6TestDataFrameConstructors.test_constructor_series_copyc C s� dd� }t dddgdddggdtjgd �}||� t dddgdddggd d tjgd �}||� t d dddgddddggtjd d tjgd �}||� t ddddgddddggtjd d tjgd �}||� t ddddgddddggtjdddgd �}||� d S )Nc S s� x(t t| j��D ]}| jd d �|f qW t�t| j��t| j� }t|�dkr�tjt dd�� | j d d �tj f W d Q R X nft|�dkr�t � | jd d �|d f | j d d �tj f � n*t �| jd d �|f | j d d �tj f � d S )Nr z^nan$)rK r' )rf ro re r; r6 r\ r rO rP �KeyErrorr� r8 r) r< r* )r, r� Zindexerr. r. r/ r C s 0zBTestDataFrameConstructors.test_constructor_with_nas.<locals>.checkr' r1 r2 rx rd rX )r& g�������?g������@)re r r g g @)r r6 r8 )r+ r r, r. r. r/ �test_constructor_with_nas? s "$,**z3TestDataFrameConstructors.test_constructor_with_nasc C s: t dtjdgi�}|d jtjks&t�|d d r6t�d S )Nru Fr' )r r6 r8 r4 rs r: )r+ r� r. r. r/ �&test_constructor_lists_to_object_dtypef s z@TestDataFrameConstructors.test_constructor_lists_to_object_dtypec C s\ t dddg�}t�|��dd�}t�|d�}t||jd�}t||||d��}t�||� d S ) Nr� rA r� r� r' )r2 rx )r4 )r r' r1 r2 ) r r6 r7 rM Z broadcast_tor r4 r) r* )r+ �catr= rF r- r. r. r/ �*test_constructor_ndarray_categorical_dtypel s zDTestDataFrameConstructors.test_constructor_ndarray_categorical_dtypec C s� t dtd�idd�}ttd�ddd�}t�|d |� ttd�dd�}|�� }ttd�ddd�}t�|d |� |jdd�}ttd�ddd�}t�|d |� t td�dd�}ttd�ddd�}t�|d |� d S ) Nr� r r= )r4 )r4 r5 r rr )r5 )r r� r r) r< r� )r+ r, r- r rF r. r. r/ � test_constructor_categoricalv s z6TestDataFrameConstructors.test_constructor_categoricalc C sX d}t jt|d�� tttd��g�}W d Q R X tdttd�dd�i�}t �||� d S )Nz%will be changed to match the behavior)rK r r r= )r4 )r) rz r{ r r r� r r* )r+ rS r, r- r. r. r/ �-test_construct_from_1item_list_of_categorical� s zGTestDataFrameConstructors.test_construct_from_1item_list_of_categoricalc C sx d}t jt|d��" tttd��ttd��g�}W d Q R X tttd�dd�ttd�dd�d�dd gd �}t �||� d S ) Nz%will be changed to match the behavior)rK r Zabdr= )r4 )r r' r r' )re )r) rz r{ r r r� r r* )r+ rS r, r- r. r. r/ �(test_construct_from_list_of_categoricals� s & zBTestDataFrameConstructors.test_construct_from_list_of_categoricalsc C sl d}t jt|d�� tttd��td�g�}W d Q R X tttd�dd�td�d�dd gd �}t �||� d S ) Nz%will be changed to match the behavior)rK r rJ r= )r4 )r r' r r' )re )r) rz r{ r r r� r r* )r+ rS r, r- r. r. r/ �%test_from_nested_listlike_mixed_types� s "$z?TestDataFrameConstructors.test_from_nested_listlike_mixed_typesc C sf d� ddg�}d}tjt|d��< tjt|d��" ttt d��tt d��g� W d Q R X W d Q R X d S )NrJ z:Shape of passed values is \(6, 2\), indices imply \(3, 2\)z;Passed arrays should have the same length as the rows Indexz%will be changed to match the behavior)rK r Zabdefg) rN rO rP rQ r) rz r{ r r r� )r+ rS Zmsg2r. r. r/ �0test_construct_from_listlikes_mismatched_lengths� s zJTestDataFrameConstructors.test_construct_from_listlikes_mismatched_lengthsc C s� ddddg}t |��d�}t |dd�}t�||� ddddg}t |��d�}t |dd�}t�||� td dd �}t ttjtjtjgdddgd ��}||_t d |i�}t d t dddgdd�i|d �}t� ||� d S )Nr' r1 r2 r= )r4 ru rv rw Z20000101)r� )r9 � categoriesrZ )r& ) r r� r) r< r r r6 r8 r& r r* )r+ r� r� �resr& r- r, r. r. r/ �#test_constructor_categorical_series� s z=TestDataFrameConstructors.test_constructor_categorical_seriesc C s, t dddgidg|d�}|jj|ks(t�d S )Nru r' r1 rv )re r4 )r rv r4 r: )r+ r4 rn r. r. r/ �%test_check_dtype_empty_numeric_column� s z?TestDataFrameConstructors.test_check_dtype_empty_numeric_columnc C s. t dddgidg|d�}|jjjdks*t�d S )Nru r' r1 rv )re r4 rB )r rv r4 r5 r: )r+ r4 rn r. r. r/ �$test_check_dtype_empty_string_column� s z>TestDataFrameConstructors.test_check_dtype_empty_string_columnc C sL t dtd��� j}t dti�}t�||� tt dtd��j}t�||� d S )Nr )r5 r4 )r rB r� r� r) r< r )r+ rF r- r. r. r/ � test_to_frame_with_falsey_names� s z9TestDataFrameConstructors.test_to_frame_with_falsey_namesr r= c C sb t ddddddgi|pdd�}t td �dg|d �}t�||� t dtd �i|d�}t�||� d S ) Nr� r r' r1 r2 rx r# )r4 rd )re r4 )r rf r) r* )r+ r4 r- rF r. r. r/ � test_constructor_range_dtype� s z6TestDataFrameConstructors.test_constructor_range_dtypec C s\ G dd� dt �}tdddgdddgg�}t||dddg�|dddg�g��}t�||� d S ) Nc @ s e Zd ZdS )zETestDataFrameConstructors.test_frame_from_list_subclass.<locals>.ListN)ro rp rq r. r. r. r/ �List s r� r' r1 r2 rx rd rX )r� r r) r* )r+ r� r- rF r. r. r/ � test_frame_from_list_subclass s $z7TestDataFrameConstructors.test_frame_from_list_subclass� extension_arrZaabbcz1/1/2017z1/1/2018)r� �endr* c C s$ t t|��}t |�}t�||� d S )N)r r r) r* )r+ r� r- rF r. r. r/ �%test_constructor_with_extension_array s z?TestDataFrameConstructors.test_constructor_with_extension_arrayc C s^ t �� }||f}t|ttd�td�d�i|gd�}tdddgtt|g��d�}t�||� d S )Nr2 )r& )re r r' r1 )r �todayr r rf r r) r* )r+ r� �tuprF r- r. r. r/ �*test_datetime_date_tuple_columns_from_dict s " zDTestDataFrameConstructors.test_datetime_date_tuple_columns_from_dictc C s� t dddgtdddg�d�}t dd dd d gtd dd ddg�d�}t||g�}tt�tjdtjddgdddddgg�d dd ddgd�}t�||� d S )N�' rX rx ZfemaleZmale�unknown)r& r1 � �� � �frV g �C@g @g @g @g c@g @n@g �b@)re )r r r r6 r7 r8 r) r* )r+ r� r� rF r- r. r. r/ �/test_construct_with_two_categoricalindex_series$ s zITestDataFrameConstructors.test_construct_with_two_categoricalindex_seriesc C s� t ddddd�t ddddd�fg}tj|ddgd �}t|�}|d d |d d ksXt�|d d |d d kstt�t|d �}t|d t�s�t�|d |d d ks�t�d S ) Ni� r r r r r� )�DatezM8[us])� ForecastingzM8[us])r4 r r r' )r r6 r7 r r: r r` r )r+ r� r= r, r r. r. r/ �test_from_M8_structured4 s " z1TestDataFrameConstructors.test_from_M8_structuredc C s> G dd� dt �}td|ddddd�gi�}|j jdks:t�d S )Nc @ s e Zd ZdS )zOTestDataFrameConstructors.test_from_datetime_subclass.<locals>.DatetimeSubclassN)ro rp rq r. r. r. r/ �DatetimeSubclassB s r r i� r' zdatetime64[ns])r r r4 r: )r+ r rn r. r. r/ �test_from_datetime_subclass@ s z5TestDataFrameConstructors.test_from_datetime_subclassc C s@ t dddd�}d}tjt|d�� t|td�d� W d Q R X d S ) Nz 2016-01-01r2 z US/Pacific)r� r/ z@Shape of passed values|Passed arrays should have the same length)rK rx )r& )r rO rP rQ r rf )r+ �dtirS r. r. r/ �(test_with_mismatched_index_length_raisesH s zBTestDataFrameConstructors.test_with_mismatched_index_length_raisesc C sR t dddd�}t�|�}ttj�t|��|d��}t�|d jt�d��sNt �d S )Nz1/1/2000 00:00:00z1/1/2000 1:59:50r� )r* )r� rA rA zM8[ns]) r r6 r: r r� r ro Z issubdtyper4 r: )r+ r� r� r, r. r. r/ �!test_frame_ctor_datetime64_columnO s z;TestDataFrameConstructors.test_frame_ctor_datetime64_columnc C s� ddddgddddgg}t tj�dd�dd� |D �d�}t|jt�sHt�t|jt�rXt�t tj�dd�|d �}t|jt�s~t�d S ) Nru rv rZ rg rx c S s g | ]}t �|��qS r. )r6 r7 )rY rZ r. r. r/ r[ [ s zYTestDataFrameConstructors.test_dataframe_constructor_infer_multiindex.<locals>.<listcomp>)r& )re ) r r6 r� r r` r& r r: re )r+ Z index_lists�multir. r. r/ �+test_dataframe_constructor_infer_multiindexV s zETestDataFrameConstructors.test_dataframe_constructor_infer_multiindex� input_valsr� r� z1/1/2011�H)r� r* )r� r* r/ c C s6 t d|i|d�}t d|i��d|i�}t�||� d S )Nr� )r4 )r r� r) r* )r+ r r� rF r- r. r. r/ �test_constructor_list_strc s z3TestDataFrameConstructors.test_constructor_list_strc C s< t dddd gi|d�}t dddd gitd�}t�||� d S )Nr� g �?g @)r4 z1.0z2.0)r rB r) r* )r+ r� rF r- r. r. r/ � test_constructor_list_str_nav s z6TestDataFrameConstructors.test_constructor_list_str_nar� c sp t jddg|d�� t jddg|d���jjdkr4d S tjddg|d��t� ��d�|d��d d � �� ����fd d �}|s�|� d �jd< d �jd< |s�t�fdd��jj D ��dks�t �d �d < |�rX� jjdk�r� d � j� dd�ks�t ��d �j� dd�k�sBt �n4� d � j� d�k�s(t ��d �j� d�k�sBt ��jd dk�slt �n�jd d k�slt �d S )Nr' r1 )r4 r2 rx )�S�U)ru rv rw )r� c S s0 t | tj�r| jS t | jtj�r(| jjS t�d S )N)r` r6 Zndarray�baser4 Z_ndarrayr� )rb r. r. r/ �get_base� s z<TestDataFrameConstructors.test_dict_nocopy.<locals>.get_basec sn t �fdd��jjD ��dks"t�t � �fdd��jjD ��dksFt�t ��fdd��jjD ��dksjt�d S )Nc 3 s | ]}|� kV qd S )Nr. )rY rZ )rw r. r/ rm � s zRTestDataFrameConstructors.test_dict_nocopy.<locals>.check_views.<locals>.<genexpr>r' c 3 s( | ] }t |jtj�r�|�� kV qd S )N)r` r4 r6 )rY rZ )ru r r. r/ rm � s c 3 s( | ] }t |jtj�r�|�� kV qd S )N)r` r4 r6 )rY rZ )rv r r. r/ rm � s )�sumr^ r_ r: r. )ru rv rw r, r r. r/ � check_views� s " z?TestDataFrameConstructors.test_dict_nocopy.<locals>.check_viewsr )r r )r r' c 3 s | ]}|� kV qd S )Nr. )rY rZ )rw r. r/ rm � s z=TestDataFrameConstructors.test_dict_nocopy.<locals>.<genexpr>rW rI )r r1 ) r6 r7 r4 rU r� r r; r r^ r_ r: �type)r+ r� Zany_nullable_numeric_dtypeZany_numpy_dtyper r. )ru rv rw r, r r/ �test_dict_nocopy| s. " z*TestDataFrameConstructors.test_dict_nocopyc C s2 t tddd�dgd�}t dgd�}t�||� d S )Nr' rr )r5 r� )re )r r r) r* )r+ rF r- r. r. r/ �'test_from_series_with_name_with_columns� s zATestDataFrameConstructors.test_from_series_with_name_with_columnsc C sd t dddgdddggdddgdd d ggd �}t dddgdddggt�d d dg�d �}t�||� d S )Nr' r1 r2 rx rd rX r� ru rv rw )re )r� ru )r� rv )r� rw )r r r� r) r* )r+ rF r- r. r. r/ �test_nested_list_columns� s *z2TestDataFrameConstructors.test_nested_list_columnsc C s t jddd�}|j�t��dd�}t|�}|jdks8t�|j |j k� � sLt�||k� � � � s`t�t j � dddd g�}|j�t��dd�}t|�}|jdks�t�|j |j k� � s�t�||k� � � � s�t�tj||||f j}t|�}t||||d ��} t�|| � d S ) Nz 2016-04-05r2 )r� r' r� )r' r2 rx rd rX )r r' r1 r2 )r� r1 �_datar� rB rM r rS r: r� r4 r� Z IntervalIndexZ from_breaksr6 Zr_r r) r* ) r+ �pirn r, �iiZdata2r� Zdata3r� r- r. r. r/ �1test_from_2d_object_array_of_periods_or_intervals� s zKTestDataFrameConstructors.test_from_2d_object_array_of_periods_or_intervals)�ro rp rq r0 r> r@ rG rT rO �mark� parametrizerc rh rp r rq rt r} r� r� r( Z skip_array_manager_invalid_testr� r� Zskipifr r� r� r� r� r� r� r� r6 r7 rB r� r� r� r� r� r� r� r� r� r8 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 r r r0 r4 r6 r7 r; rC rD rH ZslowrR rT r� rf rs r4 r� rU r� Z BooleanDtypeZ Float64DtypeZ Int64DtypeZ StringDtyperX rZ r[ r\ r^ r` rb rd rf rg rh rt ru rv rw rz 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� r� r� r� r� r� r� r� r� r� r� r� r r? rL rH r� r� r� r� r� Z&skip_array_manager_not_yet_implementedr� r� r� r� r� r� r� r� r� r� r� r) ZALL_INT_DTYPESZALL_EA_INT_DTYPESZ FLOAT_DTYPESZCOMPLEX_DTYPESZDATETIME64_DTYPESZTIMEDELTA64_DTYPESZ BOOL_DTYPESr� Z STRING_DTYPESZ BYTES_DTYPESZ OBJECT_DTYPESr� r� Zarm_slowr� r� r r r r r1 r� r� r r r r r r r r r r r r r r. r. r. r/ r$ F s " * & & $" 4 " 9J0 0 ' D & G ; ' 8 & ( * O r$ c @ s e Zd Zdd� ZdS )�)TestDataFrameConstructorWithDtypeCoercionc C s� t j�dd�}d}tjt|d�� t|dd� W d Q R X t�d �� t|�� dd� W d Q R X t j|d< t�d �� t|dd� W d Q R X d S )Nr� rd z!if they cannot be cast losslessly)rK r3 )r4 )r r ) r6 r� r r) rz r{ r �roundr8 )r+ r= rS r. r. r/ �"test_floating_values_integer_dtype� s zLTestDataFrameConstructorWithDtypeCoercion.test_floating_values_integer_dtypeN)ro rp rq r! r. r. r. r/ r � s r c @ s� e Zd Zej�dddg�dd� �Zej�dddg�d d � �Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zej�d eeg�d d � �Zd d!� Zd"S )#�&TestDataFrameConstructorWithDatetimeTZr/ z US/Easternzdateutil/US/Easternc C s� t dddd�}|�|�}td|d�|d�}td|j�}|d j|ksHt�d d � |D �}d d � |D �}td |i�}||d< ||d< ||d< |j}tt �d�t|d�t �d�t|d�gd dddgd�} t � || � d S )Nz2011/1/1z2012/1/1zW-FRI)r* rr )r� rA )r& rI rA c S s g | ] }|� � �qS r. )r� )rY r� r. r. r/ r[ s zeTestDataFrameConstructorWithDatetimeTZ.test_construction_preserves_tzaware_dtypes.<locals>.<listcomp>c S s g | ] }|� � �qS r. )r� )rY r� r. r. r/ r[ s r� �dr_tz�datetimes_naive�datetimes_with_tzzdatetime64[ns])r/ ) r rC r r �tzinfor4 r: r� r r6 r) r< ) r+ r/ r� r# r, Z tz_expectedr$ r% rF 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�� td|gidd�}W d Q R X td|gi�}t�||� tjtdd�� td|idgdd�}W d Q R X t�||� tjtdd�� t|gdd�}W d Q R X t�||� tjtdd�� ttj|gt d�dd�}W d Q R X t�||� t�t�� t|dgdgdd �}W d Q R X t�||� tjtdd�� tt |g�gdd�}W d Q R X t�||� tjtdd�� t|ggdgdd �}W d Q R X t� ||� d S ) N�2019)r/ r zdatetime64[ns])r4 F)�check_stacklevel)r& r4 )r& re r4 )re r4 ) r r� r) rz r{ r r* r6 r7 rB r Z assert_equal) r+ Ztz_aware_fixturer( r/ r� �ts_naiverF r- r, r. r. r/ �'test_constructor_data_aware_dtype_naive' s6 " zNTestDataFrameConstructorWithDatetimeTZ.test_constructor_data_aware_dtype_naivec C s� t tdddd�dd�}tddd�}t||d ��}|d jsBtd ��|d jd ksTt�t�|d t|d d�� t�|d t|d d�� d S ) Nr� r2 z US/Eastern)r� r/ rr )r5 Z20130110)r� )r� rA r� zM8[ns, US/EasternrA ) r r r r4 r: r5 r) r< r )r+ rE r� r, r. r. r/ �test_from_dictN s z5TestDataFrameConstructorWithDatetimeTZ.test_from_dictc C s� t ddddd�}t|�}t�|d t|dd�� tt|��}t�|d t|dd�� t dddd�}t|�}t�|d t|dd�� tt|��}t�|d t|dd�� d S ) Nr� r2 z US/Easternrr )r� r/ r5 )r5 )r� r/ r )r r r) r< r )r+ r� r� r. r. r/ �test_from_index] s z6TestDataFrameConstructorWithDatetimeTZ.test_from_indexc C sB t ddd�}t||d�}td|d�|d� td|jd�|d� d S )Nz1/1/2012r� )r� )r& rr )ru rv )r r r r9 )r+ r� r r. r. r/ �+test_frame_dict_constructor_datetime64_1680l s zRTestDataFrameConstructorWithDatetimeTZ.test_frame_dict_constructor_datetime64_1680c C s@ t dddd�}t|�}td|d�|d�}|d �� �� s<t�d S ) Nz2011/1/1z2012/1/1zW-FRI)r* rr )r� rA )r& rA )r r r r r� r: )r+ r� r� r� r. r. r/ �+test_frame_datetime64_mixed_index_ctor_1681t s zRTestDataFrameConstructorWithDatetimeTZ.test_frame_datetime64_mixed_index_ctor_1681c C sX t ddddd�}t|dgd�}tdtddd�td dd�td dd�gi�}t�||� d S ) Nz20130101T10:00:00r2 r z US/Eastern)r� r� r* r/ Z timestamps)re )r/ z20130101T10:01:00z20130101T10:02:00)r r r r) r* )r+ r� rF r- r. r. r/ � test_frame_timeseries_column| s zCTestDataFrameConstructorWithDatetimeTZ.test_frame_timeseries_columnc C sj ddg}ddd�dddd �d �}t |d d d g|d�}t ddtjtjfg|td d d g�d�}t�||� d S )N�Nevada�Ohiog333333@g333333@)i� i� g �?g333333�?g������ @)i� i� i� )r2 r3 i� i� i� )r& re )g333333@g333333�?)g333333@g������ @)re r& )r r6 r8 r r) r* )r+ re �poprF r- r. r. r/ � test_nested_dict_construction� s zDTestDataFrameConstructorWithDatetimeTZ.test_nested_dict_constructionc C sd t dddd�}|j�t��dd�}t|�}|jdks8t�|j|j k� � sLt�||k� � � � s`t�d S )Nz2016-04-05 04:30r2 r� )r� r/ r' r� )r' r2 ) r r r� rB rM r rS r: r� r4 r� )r+ r rn r, r. r. r/ � test_from_tzaware_object_array� s zETestDataFrameConstructorWithDatetimeTZ.test_from_tzaware_object_arrayc C s� t jtd�td�td�gtddd�tjtddd�gtdd d�tjtd d d�ggtd �j}t|d d dgd�}dddg}|j|k� � s�t �d S )Nz2013-01-01 00:00:00z2013-01-02 00:00:00z2013-01-03 00:00:00z2013-01-01 00:00:00-0500z US/Eastern)r/ z2013-01-03 00:00:00-0500z2013-01-01 00:00:00+0100ZCETz2013-01-03 00:00:00+0100)r4 r� rA r� )re zdatetime64[ns]zdatetime64[ns, US/Eastern]zdatetime64[ns, CET]) r6 r7 r r� rH rB r r r� r� r: )r+ r= r� Zexpected_dtypesr. r. r/ �$test_from_tzaware_mixed_object_array� s zKTestDataFrameConstructorWithDatetimeTZ.test_from_tzaware_mixed_object_arrayc C s: t �d��d�}t|dd�}t|��d�}t�||� d S )Nr� )rd r1 zdatetime64[ns, UTC])r4 )r6 r\ rM r r� r) r* )r+ Z array_dim2r, r- r. r. r/ � test_from_2d_ndarray_with_dtype� s zFTestDataFrameConstructorWithDatetimeTZ.test_from_2d_ndarray_with_dtyper� c C sh |dddh�}d|j � d�}tjt|d�� td|i� W d Q R X tjt|d�� t|� W d Q R X d S )Nr' r1 r2 �'z' type is unordered)rK ru )ro rO rP r� r r )r+ r� r9 rS r. r. r/ �!test_construction_from_set_raises� s zHTestDataFrameConstructorWithDatetimeTZ.test_construction_from_set_raisesc C s> t jdddd��dd�}t|�}tdd� |jjD ��s:t�d S ) Nr r� zdatetime64[ns])r4 rx r2 c s s | ]}t |t�V qd S )N)r` r )rY r= r. r. r/ rm � s zeTestDataFrameConstructorWithDatetimeTZ.test_construction_from_ndarray_datetimelike.<locals>.<genexpr>)r6 r\ rM r r� r^ r_ r: )r+ r= r, r. r. r/ �+test_construction_from_ndarray_datetimelike� s zRTestDataFrameConstructorWithDatetimeTZ.test_construction_from_ndarray_datetimelikeN)ro rp rq rO r r r' r, r- r. r/ r0 r1 r5 r6 r7 r8 �set� frozensetr: r; r. r. r. r/ r" s ' r" c C s" t | t�r| jd S | jd S d S )Nr )r r )r` r r; )rb r. r. r/ �get1� s r> c @ s� e Zd Zejeedgd�dd� �Zej� dddg�dd � �Z d d � Z d d � Z dd� Z ej� dejejg�dd� �Zej� deejg�dd� �Zej� deejg�dd� �Zdd� ZdS )�TestFromScalarN)�paramsc s� |j }dtd�i� �tkr$dg� d< |d kr:tj�f� �S |tkrh�tkrX� �fdd�S � �fdd�S n$�tkr~� �fdd�S � �fd d�S d S ) Nr& r1 r� re c s �| | d�f� |��S )N)r r' r. )rZ �kwargs)�extrarR r. r/ rj � s z,TestFromScalar.constructor.<locals>.<lambda>c s �d| if� |��S )Nr� r. )rZ rA )rB rR r. r/ rj � rk c s �| | gf� |��S )Nr. )rZ rA )rB rR r. r/ rj � rk c s �d| | gif� |��S )Nr� r. )rZ rA )rB rR r. r/ rj � s )�paramrf r rM rN r� r )r+ �requestrR �boxr. )rB rR r/ ri � s zTestFromScalar.constructorr4 zM8[ns]zm8[ns]c C s8 |t j|d�}t�|j|k�s"t�t�|�� �s4t�d S )N)r4 )r� rH r6 r� r� r: r )r+ r4 ri rb r. r. r/ �test_from_nat_scalar� s z#TestFromScalar.test_from_nat_scalarc C s( t d�}||dd�}t|�|ks$t�d S )Nr' zm8[ns])r4 )r r> r: )r+ ri r( rb r. r. r/ �*test_from_timedelta_scalar_preserves_nanos s z9TestFromScalar.test_from_timedelta_scalar_preserves_nanosc C s0 t �� td� }||dd�}t|�|ks,t�d S )Nr' zM8[ns])r4 )r r? r r> r: )r+ ri r� rb r. r. r/ �*test_from_timestamp_scalar_preserves_nanos s z9TestFromScalar.test_from_timestamp_scalar_preserves_nanosc C s4 t d�}|�� }||td�}tt|�tj�s0t�d S )Nr' )r4 )r Zto_timedelta64rB r` r> r6 r& r: )r+ ri r( Ztd64rb r. r. r/ �#test_from_timedelta64_scalar_object s z2TestFromScalar.test_from_timedelta64_scalar_object�clsc C s� |j }|jj}|d tkr>|d tkr>tjjdd�}|�|� |dd�}t j dt j di| }d } |t j krxd � d d g�} tj t| d �� |||d � W d Q R X |dd�}tj t| d �� |||d � W d Q R X d S )NrR ri z4DataFrame incorrectly allows mismatched datetimelike)r� rH rI zm8[ns]zM8[ns]z Cannot castrJ z?dtype datetime64\[ns\] cannot be converted to timedelta64\[ns\])rK )r4 rx )�nodeZcallspecr@ r r� rO r ZxfailZ add_markerr6 rL r& rN rP r� ) r+ ri rJ rD rK r@ r r� r4 rS r. r. r/ �(test_from_scalar_datetimelike_mismatched s$ z7TestFromScalar.test_from_scalar_datetimelike_mismatchedc C sB t ddd�}|tjkr"t�|d�}||�}tt|��|ks>t�d S )Ni' r' r� )r r6 rL r r> r: )r+ ri rJ r� rF r. r. r/ � test_from_out_of_bounds_datetime5 s z/TestFromScalar.test_from_out_of_bounds_datetimec C sN t ddd�t ddd� }|tjkr.t�|d�}||�}tt|��|ksJt�d S )Ni' r' i� r� )r r6 r& r r> r: )r+ ri rJ r� rF r. r. r/ �!test_from_out_of_bounds_timedelta> s z0TestFromScalar.test_from_out_of_bounds_timedeltac C sj d}t d|d�}t d�}tjtddd�� ||dd�}W d Q R X t�|jdk�sTt�t�||k�sft�d S ) Nz US/Easternr) )r/ zData is timezone-awareF)rK r* zM8[ns])r4 )r r) rz r{ r6 r� r� r: )r+ ri r/ r� r+ rF r. r. r/ � test_tzaware_data_tznaive_dtypeG s z.TestFromScalar.test_tzaware_data_tznaive_dtype)ro rp rq rO Zfixturer� r� ri r r rF rG rH rI r6 rL r& rL r rM r rN rO r. r. r. r/ r? � s r? )>r� r r r r r rM rP r� r? �numpyr6 Znumpy.mar8 Znumpy.ma.mrecordsrG rO r� Z pandas.compatr Z pandas.util._test_decorators�utilZ_test_decoratorsr( Zpandas.core.dtypes.commonr Zpandas.core.dtypes.dtypesr r r r Zpandasr� r r r r r r r r r r r r r r Zpandas._testingZ_testingr) Z pandas.arraysr r r r r� r� r$ r r" r> r? r. r. r. r/ �<module> sh @ A U
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка