U
|
ý°d° ã @ sP d dl Zd dlZd dlZd dlmZ d dlmZ d dl m
|
Z
|
G dd de
|
Z dS )é N)ÚExtensionArray)Ú EABackedBlock)ÚBaseExtensionTestsc @ s e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Ze j
|
d ddg¡dd Z dd Z dd Zdd Zdd Zdd Zd d Zd d Zd S )!ÚBaseConstructorsTestsc C sP t |j||jd}| ||¡ |d d
}t |j||jd}| ||¡ d S )N©Údtyper )ÚtypeÚ_from_sequencer Ú assert_extension_array_equal©ÚselfÚdataÚresult© r ú_d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/extension/base/constructors.pyÚtest_from_sequence_from_cls s
|
z1BaseConstructorsTests.test_from_sequence_from_clsc C s6 |d |d |d g}| |¡}t|t|s2td S )Nr é é )r Ú
|
isinstancer ÚAssertionError)r r Zscalarsr r r r Útest_array_from_scalars s
|
z-BaseConstructorsTests.test_array_from_scalarsc C s¤ t j|dd}|j|jks tt|t|ks2tt|jdrTt|jjd t sTt|jj
|
|ksdtt |¡}|j|jks~tt|jdr t|jjd t s td S )NF)ÚcopyÚblocksr ) ÚpdÚSeriesr r ÚlenÚhasattrÚ_mgrr r r Úarray)r r r Úresult2r r r Útest_series_constructor s
|
z-BaseConstructorsTests.test_series_constructorc C st t jdddg|d}t j|gd dddg|d}| ||¡ t jg |d}t jg t jg dd|d}| ||¡ d S )Nr r é ©Úindexr Úobjectr )r r Úassert_series_equalZIndex©r r Zna_valuer Úexpectedr r r Ú*test_series_constructor_no_data_with_index& s z@BaseConstructorsTests.test_series_constructor_no_data_with_indexc C sB t j|dddg|d}t j|gd dddg|d}| ||¡ d S )Nr r r! r" ©r r r% r& r r r Ú,test_series_constructor_scalar_na_with_index0 s zBBaseConstructorsTests.test_series_constructor_scalar_na_with_indexc C s| |d }t j|dddg|d}t j|gd dddg|d}| ||¡ t j|dg|d}t j|gdg|d}| ||¡ d S )Nr r r r! r" Zfoor) )r r r Zscalarr r' r r r Ú)test_series_constructor_scalar_with_index5 s z?BaseConstructorsTests.test_series_constructor_scalar_with_indexÚ from_seriesTFc C s |rt |¡}t d|i¡}|jd |jks0t|jt|dfksFtt|j drht
|
|j j d t shtt
|
|j j d ts~td S )NÚAr r r )r r Ú DataFrameÚdtypesr r Úshaper r r r r r Úarraysr )r r r, r r r r Ú$test_dataframe_constructor_from_dict? s
|
z:BaseConstructorsTests.test_dataframe_constructor_from_dictc C sv t t |¡¡}|jd |jks$t|jt|dfks:tt|j dr\t
|
|j j d t s\tt
|
|j j d tsrtd S )Nr r r )r r. r r/ r r r0 r r r r r r r1 r r r r r Útest_dataframe_from_seriesJ s z0BaseConstructorsTests.test_dataframe_from_seriesc C sB d}t jt|d& tj|d d
dddddgd W 5 Q R X d S ) Nz;Length of values \(3\) does not match length of index \(5\))Úmatchr! r r r é ©r# )ÚpytestZraisesÚ
|
ValueErrorr r )r r Úmsgr r r Ú)test_series_given_mismatched_index_raisesR s z?BaseConstructorsTests.test_series_given_mismatched_index_raisesc C s¤ |j }t |¡}tjt||d}| ||¡ tjt|t|d}| ||¡ t |¡ |¡}tjt||d}| ||¡ tjt|t|d}| ||¡ d S ©Nr ) r r r Úlistr% Ústrr. ZastypeÚassert_frame_equal)r r r r' r r r r Útest_from_dtypeW s
|
z%BaseConstructorsTests.test_from_dtypec C s t |¡}| ||¡ d S )N)r r r
|
r r r r Útest_pandas_arrayk s
|
z'BaseConstructorsTests.test_pandas_arrayc C s: t j|t t¡d}t j tj|td¡}| ||¡ d S r; ) r r Únpr r$ r1 Z PandasArrayZasarrayZ assert_equal)r r r r' r r r Útest_pandas_array_dtypep s z-BaseConstructorsTests.test_pandas_array_dtypec C sB t jdg|d}t jdt jg |dit d¡d}| ||¡ d S )NÚa)Úcolumnsr r r r6 )r r. r Z
|
RangeIndexr> )r r r r' r r r Ú test_construct_empty_dataframev s ÿz4BaseConstructorsTests.test_construct_empty_dataframec C s¬ | ¡ }|jd|d}t||s$t|j|ks2t|jdks@t| d¡}t||sXt|j|ksft|jdkstt| d¡}t||st|j|kst|jdks¨td S )N)r5 r r5 )Zconstruct_array_typeÚ_emptyr r r r0 Úempty)r r Úclsr r r r r Ú
|
test_empty~ s
|
|
z BaseConstructorsTests.test_emptyN)Ú__name__Ú
|
__module__Ú __qualname__r r r r( r* r+ r7 ÚmarkZ parametrizer2 r3 r: r? r@ rB rE rI r r r r r
|
s
|
|
|
|
r ) ÚnumpyrA r7 Zpandasr Zpandas.api.extensionsr Z pandas.core.internals.blocksr Z pandas.tests.extension.base.baser r r r r r Ú<module> s
|
