U
|
ý°dt ã @ sN d dl m Z d dlZd dlZd dlmZmZmZ d dlm Z
|
G dd dZ dS )é )ÚdatetimeN)ÚIndexÚ
|
RangeIndexÚSeriesc @ s6 e Zd Zej dddg¡ej ddi dddfd i dddfd
|
i ddd fd i dddfd i dddfdddidddfdddidddfg¡dd Zdd Zej deddge ddge
|
ddg¡g e
|
dd¡e
|
dg¡dgg¡dd Z ej ddedddddg¡d d Zd d Zd d! Zd"d# Zd$d% Zd&d' ZdS )(ÚTestRangeIndexConstructorsÚnameNZfooz args, kwargs, start, stop, step)é r r é )r r )r r é r
|
)r )r r © ÚstartÚstopc C st t |d|i|}ttj|||tjd|d}t|t s<t|j|ksJt|jt |||ks`tt
|
j ||dd d S )Nr ©Údtype©r Úequiv©Úexact) r r ÚnpÚarangeÚint64Ú
|
isinstanceÚAssertionErrorr Ú_rangeÚrangeÚtmÚassert_index_equal) ÚselfÚargsÚkwargsr r Ústepr ÚresultÚexpectedr r údd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/ranges/test_constructors.pyÚtest_constructor s z+TestRangeIndexConstructors.test_constructorc C sr d}t jt|d t W 5 Q R X t jt|d tdd W 5 Q R X d}t jt|d td W 5 Q R X d S )Nz1RangeIndex\(\.\.\.\) must be called with integers©ÚmatchZFoor zKIndex\(\.\.\.\) must be called with a collection of some kind, 0 was passedr )ÚpytestÚraisesÚ TypeErrorr r )r Úmsgr r r# Ú test_constructor_invalid_args$ s ÿz8TestRangeIndexConstructors.test_constructor_invalid_argsr ÚaÚbé
|
c C s6 dt |j }tjt|d t| W 5 Q R X d S )Nz+Value needs to be a scalar value, was type r% )ÚtypeÚ__name__r' r( r) r ©r r r* r r r# Ú(test_constructor_additional_invalid_args4 s zCTestRangeIndexConstructors.test_constructor_additional_invalid_argsiÐ c C s: dt | d| }tjt|d t| W 5 Q R X d S )Nz Wrong type z for value r% )r/ r' r( r) r r1 r r r# Ú(test_constructor_invalid_args_wrong_typeE s zCTestRangeIndexConstructors.test_constructor_invalid_args_wrong_typec C s t ddd}t |dd}| |¡s&tt |dd}tj||dd t |}tj||dd tjtdd t |d
|
d W 5 Q R X d S ) Nr r r
|
F©ÚcopyTr úCIncorrect `dtype` passed: expected signed integer, received float64r% Úfloat64r )r Z identicalr r r r' r( Ú
|
ValueError)r Úindexr! r r r# Útest_constructor_sameK s þz0TestRangeIndexConstructors.test_constructor_samec C s0 t tddd}t ddd}tj||dd d S )Nr r r
|
Tr )r r r r )r r! r" r r r# Ú test_constructor_range_object] s z8TestRangeIndexConstructors.test_constructor_range_objectc C s t tddd¡}t ddd}tj||dd t tdd¡}t ddd}tj||dd t tdd¡}t ddd}tj||dd t td¡}t ddd}tj||dd ttddd}t ddd}tj||dd d}tjt|d t jtd
|
dd W 5 Q R X d S ) Nr r r
|
Tr é r zI(RangeIndex.)?from_range\(\) got an unexpected keyword argument( 'copy')?r% r. r4 ) r Z
|
from_ranger r r r r' r( r) )r r! r" r* r r r# Útest_constructor_rangeb s& ÿz1TestRangeIndexConstructors.test_constructor_rangec C s t d}d|_t |}d|_|jdks*t|jdks8tt|}|jdksNtd|_|jdksbt|jdkspt|jdks~td S )Nr. Úoriginalr5 Únew)r r r r )r Úorigr5 r? r r r# Útest_constructor_name~ s z0TestRangeIndexConstructors.test_constructor_namec C sÈ t jddddgtd}tdd}|jjt jks2tt| d¡}t
|
j ||dd t j td
|
d td d d W 5 Q R X t j tdd tddd W 5 Q R X t j tdd tdddd W 5 Q R X d S )Nr r
|
é é r r r r r zWrong type \<class 'str'\>r% Ú1Ú10z Wrong type \<class 'float'\>gñ?gffffff$@gÍÌÌÌÌÌô?r6 r7 )r ÚarrayÚobjectr Úvaluesr r r r Zastyper r r' r( r) r8 )r Zarrr9 r" r r r# Útest_constructor_corner s
|
þz2TestRangeIndexConstructors.test_constructor_corner)r0 Ú
|
__module__Ú __qualname__r' ÚmarkZ parametrizer$ r+ r r r rF r r2 r r3 r: r; r= rA rI r r r r# r sB ùþ
|
|
|
|
ùþ
|
|
r ) r Únumpyr r' Zpandasr r r Zpandas._testingZ_testingr r r r r r# Ú<module> s
|
|
