U
|
¬ý°dä ã @ s d dl Zd dlZd dlZd dlmZ d dlm Z d dl
|
m Z d dl m Z mZmZ ejeje jgddd Zdd Zd
|
d Zd d Zdd Zdd Zej dddgdejgfdgejgfdejgejejgfejejgejejgfg¡dd Z ej dddgdddej ddd e dg¡ddgd d ggejd digg¡d d! Z d"d# Z d$d% Z d&d' Z!d(d) Z"ej d*d+d,gd dge e fd+d,gd dgd-e fd+d,ejgd dejge e fg¡d.d/ Z#ej d0ejdgd1d2defe dejg¡defe dejg¡d3e fg¡d4d5 Z$dS )6é N)Ú
|
is_integer)Ú IntegerArray)Ú Int8DtypeÚ
|
Int32DtypeÚ
|
Int64Dtype)Úparamsc C s | j S )zjFixture returning parametrized IntegerArray from given sequence.
|
|
Used to test dtype conversions.
|
)Úparam)Úrequest© r
|
údd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/integer/test_construction.pyÚ constructor s r c C s* t jdd gt d} | d t jks&td S )Né ©Údtype)ÚpdÚarrayr ÚNAÚAssertionError)Úar
|
r
|
r Útest_uses_pandas_na s r c C sÀ | j }t | ¡}tj| jtjddt|d}t ||¡ t | ¡}tjt | ¡
|
¡ t|d}t ||¡ t | ¡ ¡ j dd}t | ¡ ¡ t |j¡¡}tj|t|d}t ||¡ d S )NÚfloat)Zna_valuer r T)Zdrop)r r ÚSeriesZto_numpyÚnpÚnanÚstrÚtmÚassert_series_equalr ÚtolistZdropnaZ reset_indexÚastypeÚtype)Údatar ÚexpectedÚresultZdroppedr
|
r
|
r Útest_from_dtype_from_float s
|
|
r# c C sê t d| i¡}|d d¡}t jtjtjdgtddd}t ||¡ |d d¡j
|
}tjt j dgtd}t ||¡ t ||D ]d\}}t |¡r¢t |¡sätqt|rÄ||ks¶tt|sätq||ksÐtt|t|kstqd S )NÚAÚobjectr r )Úname)r Z DataFramer r r r r r% r r Úvaluesr Zassert_numpy_array_equalÚzipZisnullr r r )Z data_missingZdfr" r! ÚrÚer
|
r
|
r Útest_conversions2 s
|
r+ c C s t jddddgdd} t jddddgd d}t| |}tjdddt jgd
|
d}t ||¡ d }tjt |d t|
|
¡ | W 5 Q R X tjt |d t| |
|
¡ W 5 Q R X tjt |d t| t ¡| W 5 Q R X d }tjt |d t| W 5 Q R X d S )Nr é é é Úint64r FTÚboolÚInt64z@.* should be .* numpy array. Use the 'pd.array' function instead©Úmatchz;__init__\(\) missing 1 required positional argument: 'mask') r r r r r r Ú assert_extension_array_equalÚpytestÚraisesÚ TypeErrorr r r )r' Úmaskr" r! Úmsgr
|
r
|
r Ú test_integer_array_constructorK s
|
r: c C s t jddddgdd} t jddddgd d}t| |}|j| ksDt|j|ksRtt| |dd
|
}|j| k snt|j|k s|td S ) Nr r, r- r. r/ r FTr0 )Úcopy)r r r Ú_datar Z_mask)r' r8 r" r
|
r
|
r Ú#test_integer_array_constructor_copya s
|
r= za, br c C s, t j| dd}t j|dd}t ||¡ d S )Nr1 r )r r r r4 )r Úbr" r! r
|
r
|
r Ú!test_to_integer_array_none_is_nann s
|
r? r' ZfooÚbarç ð?Z20130101r, )Zperiodsr- r. r c C sl d dddddg¡}tjttf|d tj| dd W 5 Q R X tjttf|d t | ¡ W 5 Q R X d S )
|
Nú|z#cannot be converted to IntegerDtypez)invalid literal for int\(\) with base 10:z values must be a 1D list-likezCannot pass scalarz!int\(\) argument must be a stringr2 r1 r ) Újoinr5 r6 Ú
|
ValueErrorr7 r r r Ú_from_sequence)r' r9 r
|
r
|
r Útest_to_integer_array_error} s ûÿ rF c C sl | t jddgdd}|jt ks&t| t jddgdd}|jt ksLt| ddg}|jt kshtd S )Nr r, Úint8r Úint32)r r r r r r r ©r r" r
|
r
|
r Ú$test_to_integer_array_inferred_dtype s rJ c C sN | ddgdd}|j t ks t| tjddgdddd}|j t ksJtd S )Nr r, ZInt8r rG ZInt32)r r r r r r rI r
|
r
|
r Ú#test_to_integer_array_dtype_keyword¨ s rK c C s t ddg¡} tjddgdd}t | |¡ tjtdd t d dg¡ W 5 Q R X t t jddgd
|
d¡} | j
|
t ks|t d S ) NrA g @r r, r1 r z!cannot safely cast non-equivalentr2 g ø?Úfloat32) r rE r r r r4 r5 r6 r7 r r r r ©r" r! r
|
r
|
r Útest_to_integer_array_float± s rN c C s t ddd g¡} tjddtjgdd}t | |¡ tj t
|
dd t ddd g¡ W 5 Q R X tj t
|
dd t d
|
d g¡ W 5 Q R X d S ) NÚ1Ú2r r, r1 r z,invalid literal for int\(\) with base 10: .*r2 Ú z1.5z2.0) r rE r r r r r r4 r5 r6 rD rM r
|
r
|
r Útest_to_integer_array_str¾ s ÿ ÿrR z5bool_values, int_values, target_dtype, expected_dtypeFTr1 c C s8 | ||d}|j |ksttj||d}t ||¡ d S ©Nr )r r r r r r4 )r Z bool_valuesZ
|
int_valuesZ target_dtypeZexpected_dtyper" r! r
|
r
|
r Útest_to_integer_array_boolÎ s rT z values, to_dtype, result_dtyper/ r rG c C s> t j| |d}|j| ks ttj| | d}t ||¡ d S rS )r rE r r r r r r4 )r' Zto_dtypeZ result_dtyper" r! r
|
r
|
r Útest_to_integer_arrayß s
|
rU )%Únumpyr r5 Zpandasr Zpandas._testingZ_testingr Zpandas.api.typesr Zpandas.core.arraysr Zpandas.core.arrays.integerr r r Zfixturer rE r r r# r+ r: r= ÚmarkZ parametrizer r? Z
|
date_rangerF rJ rK rN rR rT rU r
|
r
|
r
|
r Ú<module> sn
|
üþ
|
|
øþ
|
ýþ
|
ýþ
|
