U
|
ý°d ã @ sV d dl mZ d dlZd dlZd dlmZmZmZm Z m
|
Z
|
d dl m Z G dd dZdS )é )Ú OrderedDictN)Ú DataFrameÚIndexÚ
|
MultiIndexÚ
|
RangeIndexÚSeriesc @ sÊ e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Ze j
|
d i de dfddiddigde dgddfeddggde ddgddfddigde dgddfg¡dd Zd d Zd d Zd d! Zd"S )#Ú TestFromDictc
|
C sÀ t ddgddgddgddggt ddgddgddggt ddgddggt t ddgddgddggt ddgddgddggg}t|}tjtttt||d d
|
}t || |j
|
¡¡ d S ) NÚaç ø?Úbé Úcé Údé Úindex©Úorient) r r Ú from_dictÚdictÚzipÚrangeÚlenÚtmÚassert_frame_equalÚreindexr ©ÚselfÚdataÚresultÚexpected© r! úed:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/constructors/test_from_dict.pyÚ test_constructor_list_of_odicts s ú ÿz,TestFromDict.test_constructor_list_of_odictsc C s\ t ddgddgddgddggg}t|}tjttd g|d
|
d |j¡}t ||¡ d S ) Nr r
|
r r r r r r r r r ) r r r r r r r r r r r! r! r" Útest_constructor_single_row$ s "ÿz(TestFromDict.test_constructor_single_rowc
|
C s¶ t ddgddgddggt ddgddgddggg}t tdd g|}tdddg}tdd
|
d g|d dd tdd
|
dg|d dg}t|}tj|dd}t ||¡ tdd
|
d g|d dd tdd
|
dg|g}t|}t tddg|}tj|dd}t ||¡ t ddgdd
|
gdd gddggt ddgdd
|
gddggt ddgddggt t ddgdd
|
gdd ggt dd
|
gdd gddggg}dd |D }t|}t ttt ||}tj|dd}t ||
|
|j ¡¡ t|t j dt jdd}t ||¡ tttdg}tdgd}t ||¡ t ddgddgddggt ddgddgddggg}t ttt ||}tdddg}tdd
|
d g|d dtdd
|
dg|g}t|}tj|dd}t ||¡ d S )Nr r
|
r g @r g @g @ÚxÚyr r ÚO)ÚdtypeÚnamer )r) r r z Unnamed 0r c S s g | ] }t |qS r! )r )Ú.0r r! r! r" Ú
|
<listcomp>R s z@TestFromDict.test_constructor_list_of_series.<locals>.<listcomp>©r( ©r r )r r r r r r r r r r r r ÚnpZarangeÚint64Úobject)r r ZsdictÚidxZdata2r r Zresult2r! r! r" Ú test_constructor_list_of_series- sX þþ þ ú þ$z,TestFromDict.test_constructor_list_of_seriesc C s~ |j j}tj|dd}|j|jd}t ||¡ dddgdddgd}tj|dd}t |¡j t| ¡ ¡}t ||¡ d S )Nr r r- é r é )ÚhiZthere)
|
ÚTZ_seriesr r r r r r ÚlistÚkeys)r Zfloat_string_frameÚ data_dictZreconsr r ÚrsZxpr! r! r" Útest_constructor_orientl s z$TestFromDict.test_constructor_orientc C s` t dt ddgfdt ddgfdt dd gfg}tj|d
|
d j}tj|d d }t ||¡ d S ) NÚone)Úcol_aZfoo1)Úcol_bZbar1Útwo)r= Zfoo2)r> Zbar2Zthree)r= Zfoo3)r> Zbar3Úcolumnsr r )r r r r6 r r )r r r r r! r! r" Ú"test_constructor_from_ordered_dictx s ýÿz/TestFromDict.test_constructor_from_ordered_dictc C sØ t jtdddgfdddgfgddd gd
|
}t ddgddggddgdd gd }t ||¡ d }tjt|d ( t jddgddgdddd gd
|
W 5 Q R X tjt|d & t jddgddgddd gd W 5 Q R X d S )NÚAé é ÚBr r4 r r< r? )r r@ )r r@ z2cannot use columns parameter with orient='columns'©Úmatch)rB rE r@ )r@ )r r r r r ÚpytestÚraisesÚ
|
ValueError)r r r Úmsgr! r! r" Ú test_from_dict_columns_parameter
s ý" ýz-TestFromDict.test_from_dict_columns_parameterzdata_dict, orient, expectedr r ©r rC rD r@ F)Z tupleize_cols)rM rC )©r rD rN )r r c C s" t ||¡}|j}t ||¡ d S )N)r r r@ r Zassert_index_equal)r r9 r r Zdfr r! r! r" Ú!test_constructor_from_dict_tuples s z.TestFromDict.test_constructor_from_dict_tuplesc C st t ddddgt ddddg¡d }t ddddgt ddd
|
d g¡d }t td }t|||d t |||d¡ d S )NrC rD r r )rC rD )rC r )rD rD )rD r r- )r rD )r r r, )ÚfooÚbarÚbaz©rP rR rQ )r r Ú from_tuplesr0 r r )r Ús1Ús2Zs3r! r! r" Ú(test_frame_dict_constructor_empty_series² s
|
ÿ
|
ÿ
|
z5TestFromDict.test_frame_dict_constructor_empty_seriesc C s6 d}t jt|d t tdddg¡ W 5 Q R X d S )Nz2If using all scalar values, you must pass an indexrF )r é )r r4 )r r )rH rI rJ r r r ©r rK r! r! r" Ú%test_from_dict_scalars_requires_index¿ s z2TestFromDict.test_from_dict_scalars_requires_indexc C s8 d}t jt|d tjdddddd W 5 Q R X d S ) NzNExpected 'index', 'columns' or 'tight' for orient parameter. Got 'abc' insteadrF rC r rD rS Úabcr )rH rI rJ r r rY r! r! r" Ú test_from_dict_orient_invalidÄ s ÿz*TestFromDict.test_from_dict_orient_invalidN)Ú__name__Ú
|
__module__Ú __qualname__r# r$ r2 r; rA rL rH ÚmarkZ parametrizer r r rO rW rZ r\ r! r! r! r" r s0 ? ý ýôþ
|
r )Ú collectionsr Únumpyr. rH Zpandasr r r r r Zpandas._testingZ_testingr r r! r! r! r" Ú<module> s
|
|
