U
|
ý°dÀ ã @ s` d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dl
|
m Z m Z m Z mZmZmZmZmZmZmZ d dlmZmZmZmZ d dlZd dlm Z m Z m Z m Z m Z m!Z! d dl"m#Z$ d dl%m&Z&m'Z' G dd d Z(G d
|
d d e(Z)G d d d e(Z*G dd de(Z+G dd de(Z,G dd dZ-ej. /deeeeg¡dd Z0ej. /ddde fde fded d fd ed fd e fd!ed"d#fg¡d$d% Z1ej. /de2d&fe3d&fe4d'fej5d'fe 6d(d)g¡d&fe d*d+gd&fe 6d'd&g¡d'fe d'd&gd'fe&d'd&gd'fe'e4d'fg
|
¡d,d- Z7d.d/ Z8ej. /d0e eeeeee ee g ¡d1d2 Z9d3d4 Z:d5d6 Z;d7d8 Z<dS )9é N)ÚNpyDatetimeUnit)Ú _registry)
|
Ú is_bool_dtypeÚis_categorical_dtypeÚis_datetime64_any_dtypeÚis_datetime64_dtypeÚis_datetime64_ns_dtypeÚis_datetime64tz_dtypeÚis_dtype_equalÚis_interval_dtypeÚis_period_dtypeÚis_string_dtype)ÚCategoricalDtypeÚDatetimeTZDtypeÚ IntervalDtypeÚ PeriodDtype)Ú CategoricalÚCategoricalIndexÚ DatetimeIndexÚ IntervalIndexÚSeriesÚ
|
date_range)Ú SparseArrayÚ SparseDtypec @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
ÚBasec C s t | d S ©N)Úhash©ÚselfÚdtype© r úVd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/dtypes/test_dtypes.pyÚ test_hash. s zBase.test_hashc C s |dkr t t|tjr t d S )NÚfoo)ÚAssertionErrorr
|
ÚnpÚint64r r r r! Útest_equality_invalid1 s zBase.test_equality_invalidc C sR d ddg¡}tjt|d t |¡ W 5 Q R X |tjkr@ttj|krNtd S )Nú|zdata type not understoodz$Cannot interpret '.*' as a data type©Úmatch)ÚjoinÚpytestÚraisesÚ TypeErrorr% r Zstr_r$ ©r r Úmsgr r r! Útest_numpy_informed5 s ÿzBase.test_numpy_informedc C sL t | ¡ t|jrtt |¡}t|ts<t|jr<t||ksHtd S r ) ÚtypeÚ reset_cacheÚlenÚ _cache_dtypesr$ ÚtmÚround_trip_pickleÚ
|
isinstancer ©r r Úresultr r r! Ú test_pickleA s
|
|
zBase.test_pickleN)Ú__name__Ú
|
__module__Ú __qualname__r" r' r1 r; r r r r! r - s r c @ s| e Zd Zejdd Zdd Zdd Zdd Zd d
|
Z e
|
d d gd dZ e
|
ddgddZ e ddge d dZej ddddde
|
gdd d gd de gedde e geddgdde gg¡dd Zej ddd d gd e gdd d gde gddd e gg¡dd Zd d Zd d Zd d! Zd"d# Zej d$d dgd fd ddgd fd dd d%gdfddgdfg¡d&d' Zd(d) Zd*d+ Zd,d- Zd.d/ ZdS )0ÚTestCategoricalDtypec C s t S )zG
|
Class level fixture of dtype for TestCategoricalDtype
|
)r ©r r r r! r S s zTestCategoricalDtype.dtypec C s6 t }||kst||ks tt|t|ks2td S r )r r$ r )r r Údtype2r r r! Útest_hash_vs_equalityZ s z*TestCategoricalDtype.test_hash_vs_equalityc C s¨ |dks t t|dst d|ks&t td|s4t |t ksBt t|t sRt t |ks`t tt |spt |dks|t t|drt d|kst td|r¤t d S ©NÚcategoryr# )r$ r
|
r r r r r! Ú test_equality` s z"TestCategoricalDtype.test_equalityc C sD t d¡}t||std}tjt|d t d¡ W 5 Q R X d S )NrD z0Cannot construct a 'CategoricalDtype' from 'foo'r) r# )r Úconstruct_from_stringr
|
r$ r, r- r. )r r r: r0 r r r! Ú test_construction_from_stringp s
|
|
z2TestCategoricalDtype.test_construction_from_stringc C s* d}t jt|d td W 5 Q R X d S )Nz(Parameter 'categories' must be list-liker) rD ©r, r- r. r ©r r0 r r r! Útest_constructor_invalidw s z-TestCategoricalDtype.test_constructor_invalidÚaÚbT©ÚorderedÚxÚyFr é )r Zfastpathz,values, categories, ordered, dtype, expectedNc C s t ||||¡}||ks td S r )r Ú_from_values_or_dtyper$ )r ÚvaluesÚ
|
categoriesrN r Úexpectedr: r r r! Útest_from_values_or_dtype s
|
ÿz.TestCategoricalDtype.test_from_values_or_dtypez"values, categories, ordered, dtypec C s2 d}t jt|d t ||||¡ W 5 Q R X d S )Nz?Cannot specify `categories` or `ordered` together with `dtype`.r) )r, r- Ú
|
ValueErrorr rR )r rS rT rN r r0 r r r! Ú test_from_values_or_dtype_raises s z5TestCategoricalDtype.test_from_values_or_dtype_raisesc C s2 d}t jt|d t d d d t¡ W 5 Q R X d S )Nz;Cannot not construct CategoricalDtype from <class 'object'>r) )r, r- rW r rR ÚobjectrI r r r! Ú'test_from_values_or_dtype_invalid_dtype s z<TestCategoricalDtype.test_from_values_or_dtype_invalid_dtypec C sN t |¡stt d¡s tt t ¡s,tt d¡r:tt tj¡rJtd S rC )r Úis_dtyper$ r% Úfloat64r r r r! Ú test_is_dtype¡ s
|
z"TestCategoricalDtype.test_is_dtypec C s` t |s ttddddddddg}t|dd}t |js>tt |sJtt t d¡r\td S )NrK rL ÚcÚA©Únamer\ )r r$ r r r r% )r r ZfactorÚsr r r! Ú
|
test_basic¨ s z TestCategoricalDtype.test_basicc C s( dddg}t |}t|j|ks$td S )N)rQ rK )é rL )é r^ )r ÚallrT r$ )r rT r: r r r! Útest_tuple_categories´ s
|
z*TestCategoricalDtype.test_tuple_categorieszcategories, expectedzb'c C s> t |}|jj|kstt||ks(tt|j|ks:td S r )r r Z _is_booleanr$ r )r rT rU Úcatr r r! Útest_is_boolean¹ s
|
z$TestCategoricalDtype.test_is_booleanc C s( d}t ttg jj}||ks$td S )Núdatetime64[ns])Ústrr r rT r r$ )r rU r: r r r! Ú%test_dtype_specific_categorical_dtypeÈ s z:TestCategoricalDtype.test_dtype_specific_categorical_dtypec C s t t rtd S r )r r r$ r@ r r r! Útest_not_stringÍ s z$TestCategoricalDtype.test_not_stringc C s6 t td¡}t|dd}t|}d}||ks2td S )Nre F©rT rN z7CategoricalDtype(categories=range(0, 3), ordered=False))ÚpdÚIndexÚranger Úreprr$ )r Úrngr r: rU r r r! Útest_repr_range_categoriesÑ s
|
z/TestCategoricalDtype.test_repr_range_categoriesc C s8 t dg tdgdd¡}t dgdd}||ks4td S ©NrK rL TrM )r Ú update_dtyper r$ )r r: rU r r r! Útest_update_dtypeÙ s z&TestCategoricalDtype.test_update_dtype) r< r= r> r, Úfixturer rB rE rG rJ r Údtype1rA r r^ ÚmarkÚ parametrizerV rX rZ r] rc rg ri rl rm rt rw r r r r! r? R sV
|
üþ
|
|
ýþ
|
|
|
üþ
|
r? c @ sÜ e Zd Zejdd Zdd Zdd Zdd Zd d
|
Z d d Z
|
d d Z dd Z dd Z ej dddddg¡dd Zdd Zd d Zd d Zd d! Zd"d# Zej d$d%d&g¡ej d'd(d)g¡d*d+ Zd,d- Zd.d/ Zd0S )1ÚTestDatetimeTZDtypec C s
|
t ddS )zF
|
Class level fixture of dtype for TestDatetimeTZDtype
|
Únsú
|
US/Eastern)r r@ r r r! r á s zTestDatetimeTZDtype.dtypec C s& t jtdd td W 5 Q R X d S )NzPassing a dtype aliasr) zdatetime64[ns, US/Central]©r, r- rW r r@ r r r! Útest_alias_to_unit_raisesè s z-TestDatetimeTZDtype.test_alias_to_unit_raisesc C sH t jtdd td W 5 Q R X t jtdd td W 5 Q R X d S )NÚ r) zthis is a bad stringzdatetime64[ns, US/NotATZ]©r, r- r. r r@ r r r! Ú#test_alias_to_unit_bad_alias_raisesí s z7TestDatetimeTZDtype.test_alias_to_unit_bad_alias_raisesc C s t dd}t |}||ks t||ks*t||ks6tt|t|ksJtt|t|ks^tt dd}||ksttt|t|kstd S )Nr} r~ z
|
US/Central)r r$ r )r r rA Údtype3Zdtype4r r r! rB õ s
|
|
z)TestDatetimeTZDtype.test_hash_vs_equalityc C sL t dd}|jdkst|jtjjks*t|jdks8tt|dksHtd S )NÚmsr~ z|M8[ms]zdatetime64[ms, US/Eastern])r Úunitr$ Z_cresor Z NPY_FR_msÚvaluerk )r Úresr r r! Ú test_construction_non_nanosecond s
|
|
z4TestDatetimeTZDtype.test_construction_non_nanosecondc C s, d}t jt|d tdd W 5 Q R X d S )Nz1DatetimeTZDtype only supports s, ms, us, ns unitsr) ÚDr~ r rI r r r! Útest_day_not_supported
|
s z*TestDatetimeTZDtype.test_day_not_supportedc C sD t d¡}t d¡}tt|t|s*ttt|t|s@td S )Núdatetime64[ns, US/Eastern]zdatetime64[ns, CET])r rF Ú
|
issubclassr2 r$ ©r rK rL r r r! Ú test_subclass s
|
|
z!TestDatetimeTZDtype.test_subclassc C sd t |s tt dstt|s$ttds0tt|s<ttdsHtt|rTttdr`td S ©Nr )r r$ r r r r r r r! Ú test_compat s z TestDatetimeTZDtype.test_compatc C s t d¡}t||std S r )r rF r
|
r$ r9 r r r! rG s
|
z1TestDatetimeTZDtype.test_construction_from_stringÚstringr# údatetime64[ns, notatz]zdatetime64[ps, UTC]z datetime64[ns, dateutil/invalid]c C s: d| d}t jtt |¡d t |¡ W 5 Q R X d S )Nz+Cannot construct a 'DatetimeTZDtype' from 'ú'r) )r, r- r. ÚreÚescaper rF ©r r r0 r r r! Ú)test_construct_from_string_invalid_raises$ s z=TestDatetimeTZDtype.test_construct_from_string_invalid_raisesc C s. d}t jt|d t dg¡ W 5 Q R X d S )Nz<'construct_from_string' expects a string, got <class 'list'>r) r )r, r- r. r rF rI r r r! Ú,test_construct_from_string_wrong_type_raises4 s z@TestDatetimeTZDtype.test_construct_from_string_wrong_type_raisesc C sn t d ¡rtt |¡s tt d¡s*tt d¡s8tt d¡rFtt t dd¡sZtt tj¡rjtd S )Nr úM8[ns, US/Eastern]r# r} ú
|
US/Pacific)r r[ r$ r% r\ r r r r! r] 9 s z!TestDatetimeTZDtype.test_is_dtypec C s t |dstt |ds tt |tdds0tt |dr>tt |tddrRtt tddtddrltt t d¡d st|dkstd S )
|
Nr r r} r~ r# ZCETr zM8[ns]rj )r
|
r$ r r% r r r r r! rE B s ÿz!TestDatetimeTZDtype.test_equalityc C sb t |s ttdddd}t|dd}t |js4tt |s@tt t d¡rRtt dr^td S ) NÚ20130101re r~ ©ÚperiodsÚtzr_ r` r\ ç ð?)r r$ r r r r% )r r Údrrb r r r! rc Q s z TestDatetimeTZDtype.test_basicc C s` t dddd}t|dd}t|s&tt dddd}t|dd}t|sLt|j|jks\td S )Nú
|
2013-01-01re r~ r r_ r` z
|
2013-08-01)r r r r$ r )r Zdr1Ús1Zdr2Ús2r r r! Útest_dst] s z TestDatetimeTZDtype.test_dstr ÚUTCr~ Ú constructorZM8Ú
|
datetime64c C s4 | d| d}t |¡}t d|}||ks0td S )Nz[ns, ú]r} )r rF r$ )r r r§ Zdtz_strr: rU r r r! Ú test_parserg s
|
|
z TestDatetimeTZDtype.test_parserc C s$ t jtdd t W 5 Q R X d S )NzA 'tz' is required.r) r r@ r r r! Ú
|
test_emptyp s z TestDatetimeTZDtype.test_emptyc C sT t d¡}tdddd}td|j}|j|ks2ttd|d j}|j|ksPtd S )Nr~ r¢ re r r} r )ÚpytzÚtimezoner r r r$ )r r r¡ r r r r! Útest_tz_standardizet s
|
z'TestDatetimeTZDtype.test_tz_standardizeN)r< r= r> r, rx r r r rB r r r r rG rz r{ r r r] rE rc r¥ rª r« r® r r r r! r| à s:
|
|
úþ
|
|
r| c @ s e Zd Zejdd Zdd Zdd Zdd Zd d
|
Z d d Z
|
d d Z ej ddddddg¡dd Zdd Zdd Zdd Zd d Zd d Zd!S )"ÚTestPeriodDtypec C s t dS )zB
|
Class level fixture of dtype for TestPeriodDtype
|
r )r r@ r r r! r s zTestPeriodDtype.dtypec C s t d}t |}||ks t||ks(t||ks4t||ks@t||ksLt||ksXtt|t|ksltt|t|kstd S ©Nr )r r$ r )r r rA r r r r! rB s z%TestPeriodDtype.test_hash_vs_equalityc C sÆ t jtdd td W 5 Q R X dD ].}t|}|jtjj ¡ ksHt t
|
|s&t q&dD ]0}t|}|jtjj d¡ks~t t
|
|sZt qZdD ]0}t|}|jtjj d¡ks´t t
|
|st qd S ) NzInvalid frequency: xxr) Úxx)ú period[D]z Period[D]r )ú
|
period[3D]z
|
Period[3D]Ú3Dre )z period[26H]z Period[26H]Z26Hz period[1D2H]z Period[1D2H]Z1D2Hé ) r, r- rW r Úfreqro ZtseriesÚoffsetsZDayr$ r ZHour)r rb Údtr r r! Útest_construction s z!TestPeriodDtype.test_constructionc C s@ t d}t d}tt|t|s&ttt|t|s<td S )Nr² r³ )r r r2 r$ r r r r! r s z TestPeriodDtype.test_subclassc C s| t dt dkstt dt dks(tt dt dks<tt dt dksPtt dt dksdtt dt dksxtd S )Nr² r³ z period[1S1U]zperiod[1000001U]©r r$ r@ r r r! Ú test_identity´ s z TestPeriodDtype.test_identityc C s4 t |r tt drtt|r$ttdr0td S )Nr² )r r$ r r r r r! r ¾ s zTestPeriodDtype.test_compatc C s\ t d}t||stt d¡}t||s.ttjtdd t dddg¡ W 5 Q R X d S )Nr² Úlistr) rQ rd re )r r
|
r$ rF r, r- r. r9 r r r! rG Ä s
|
z-TestPeriodDtype.test_construction_from_stringr r# z period[foo]zfoo[D]rj r c C s: d| d}t jtt |¡d t |¡ W 5 Q R X d S )Nz'Cannot construct a 'PeriodDtype' from 'r r) )r, r- r. r r r rF r r r r! Ú/test_construct_dtype_from_string_invalid_raisesÍ s z?TestPeriodDtype.test_construct_dtype_from_string_invalid_raisesc C sö t |¡stt d¡s tt d¡s*tt t d¡s<tt d¡sJtt d¡sXtt t d¡sjtt t d¡s|tt d¡rtt d¡rtt d¡r¦tt d¡r´tt d ¡rÂtt tj¡rÒtt tj¡râtt tj¡ròtd S )
|
Nr² r³ r´ z period[U]z period[S]ÚUÚSr r# )r r[ r$ r% Úobject_r& r\ r r r r! r] Ü s z TestPeriodDtype.test_is_dtypec C sp t |dstt |tds tt |tds2tt tdtdsHtt |drVtt tdtdrltd S )Nr² r Z2D)r
|
r$ r r r r r! rE ï s z TestPeriodDtype.test_equalityc C s~ t |s ttjdddd}t |js*tt |s6tt|dd}t |jsPtt |s\tt t d¡rntt drztd S ) Nz2013-01-01 09:00é ÚH)r r¶ r_ r` r\ r )r r$ ro Z period_ranger r r% )r r Zpidxrb r r r! rc ø s zTestPeriodDtype.test_basicc C s0 t }d}tjt|d t| W 5 Q R X d S )Nz!object has no attribute 'freqstr'r) )r r, r- ÚAttributeErrorrk )r r¸ r0 r r r! r« s zTestPeriodDtype.test_emptyc C s t tdrtd S r° )r r r$ r@ r r r! rm s z TestPeriodDtype.test_not_stringN)r< r= r> r, rx r rB r¹ r r» r rG rz r{ r½ r] rE rc r« rm r r r r! r¯ ~ s.
|
|
ûþ
|
|
r¯ c @ sæ e Zd Zejdd Zdd Zej dddde d¡g¡d d
|
Z
|
ej dddde d¡g¡d d Z d d Z ej ddddg¡dd Z ej deeddeeddeedddg¡dd Zej dd d d g¡d d! Zd"d# Zd$d% Zd&d' Zej d(d)d*d+dg¡d,d- Zej d(d.d/d g¡d0d1 Zd2d3 Zd4d5 Zd6d7 Zej dddddd8d9d:d;d<e d=g
|
¡d>d? Z ej ddd8d9d:d;d<e d=g¡d@dA Z ej ddddg¡dBdC Z dDdE Z dFdG Z!dHdI Z"dJdK Z#dLdM Z$dS )NÚTestIntervalDtypec C s
|
t ddS )zD
|
Class level fixture of dtype for TestIntervalDtype
|
r& Úright)r r@ r r r! r s zTestIntervalDtype.dtypec C s& t dd}t |}||ks t||ks*t||ks6t||ksBt||ksNt||ksZtt|t|ksntt|t|kstt d}t |}t d}||ks¦t||ks²t||ks¾t||ksÊt||ksÖt||ksâtt|t|ksött|t|ks tt|t|ks"td S )Nr& rÅ Úinterval)r r$ r )r r rA r ry r r r! rB s,
|
z'TestIntervalDtype.test_hash_vs_equalityÚsubtypeúinterval[int64]zInterval[int64]r& c C s0 t |dd}|jt d¡ks tt|s,td S )NrÅ ©Úclosedr& )r rÇ r% r r$ r ©r rÇ Úir r r! r¹ 5 s z#TestIntervalDtype.test_constructionc C s t |}|jd kstd S r )r rÊ r$ ©r rÇ r r r r! Ú$test_construction_allows_closed_none= s z6TestIntervalDtype.test_construction_allows_closed_nonec C s, d}t jt|d tdd W 5 Q R X d S )Nz?'closed' keyword does not match value specified in dtype stringr) zinterval[int64, left]rÅ )r, r- rW r rI r r r! Útest_closed_mismatchF s z&TestIntervalDtype.test_closed_mismatchNrÆ ZIntervalc C s& t |}|jd kstt|s"td S r )r rÇ r$ r rË r r r! Útest_construction_genericK s z+TestIntervalDtype.test_construction_genericÚabcFÚwxyzTz<U10zinterval[category]zinterval[object]c C s* d}t jt|d t| W 5 Q R X d S )NzIcategory, object, and string subtypes are not supported for IntervalDtyper) ©r, r- r. r ©r rÇ r0 r r r! Ú test_construction_not_supportedR s ÿz1TestIntervalDtype.test_construction_not_supportedr± Ú IntervalAz Interval[foo]c C s* d}t jt|d t| W 5 Q R X d S )Nz!could not construct IntervalDtyper) rÓ rÔ r r r! Útest_construction_errorsg s z*TestIntervalDtype.test_construction_errorsc C s: t tjd}d}tjt|d t |dd W 5 Q R X d S )NÚleftz&dtype.closed and 'closed' do not matchr) ÚbothrÉ )r r% r\ r, r- rW r/ r r r! Útest_closed_must_matchm s z(TestIntervalDtype.test_closed_must_matchc C s* t jtdd ttjd W 5 Q R X d S )Nzclosed must be one ofr) r# )r, r- rW r r% r\ r@ r r r! Útest_closed_invalidu s z%TestIntervalDtype.test_closed_invalidc C s2 t d}t||stt d¡}t||s.td S ©Núinterval[int64, right])r r
|
r$ rF r9 r r r! rG y s
|
z/TestIntervalDtype.test_construction_from_stringr r g
ëQ¸ @©rK rL c C s< dt | }tjtt |¡d t |¡ W 5 Q R X d S )Nz.'construct_from_string' expects a string, got r) )r2 r, r- r. r r r rF r r r r! Ú$test_construction_from_string_errors s z6TestIntervalDtype.test_construction_from_string_errorsr# z
|
foo[int64]c C s, d}t jt|d t |¡ W 5 Q R X d S )NzIncorrectly formatted string passed to constructor. Valid formats include Interval or Interval\[dtype\] where dtype is numeric, datetime, or timedeltar) )r, r- r. r rF r r r r! Ú+test_construction_from_string_error_subtype s ÿz=TestIntervalDtype.test_construction_from_string_error_subtypec C s@ t d}t d}tt|t|s&ttt|t|s<td S rÜ )r r r2 r$ r r r r! r s z TestIntervalDtype.test_subclassc C s t |¡stt d¡s tt t d¡s.tt t d¡s@tt t tj¡sTtt t dd¡shtt t dd¡s|tt t tjd¡stt d¡r tt d¡r®tt d ¡r¼tt d
|
¡rÊtt d ¡rØtt d ¡rætt tj¡rött tj¡rtt tj¡rtd 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] s" z TestIntervalDtype.test_is_dtypec C sæ t |dstt |tdds"tt tddtdds<tt |drJtt |tdr\tt tddtdrttt |drtt tddtddrtt tddtddr¶ttdd}td d}||ksÖt||ksâtd S )
|
NrÝ r& rÅ rÈ Zneitherr\ rÙ rØ r )r
|
r$ r )r r ry rA r r r! rE ® s2 ÿ ÿ ÿ ÿ
|
|
z TestIntervalDtype.test_equalityZuint64r\ Ú
|
complex128r¨ Z timedelta64ÚQc C s> |d k r dnd }t ||d}t|ds*tt|t s:td S )NrÅ rÉ rÆ )r r
|
r$ )r rÇ rÊ r r r r! Útest_equality_genericÉ s z'TestIntervalDtype.test_equality_genericc C sP |d k r dnd }t ||d}d| d| d}t||ks>t|jdksLtd S )NrÅ rÉ z interval[z, r© rÆ ©r rk r$ ra )r rÇ rÊ r rU r r r! Útest_name_reprß s
|
z TestIntervalDtype.test_name_reprc C s* t |}t|dkst|jdks&td S ©NrÆ rä rÍ r r r! Útest_name_repr_genericó s z(TestIntervalDtype.test_name_repr_genericc C s^ t |s tt td¡}t |js(tt |s4tt|dd}t |jsNtt |sZtd S )Nre r_ r` )r r$ r Ú from_breaksrq r r )r r Úiirb r r r! rc ú s z TestIntervalDtype.test_basicc C s¤ t ds tt t dg¡s tt t t d¡¡s8tt t tddd¡sRtt dr^tt drjtt d rvtt tjrtt tj rtt tj
|
r td S )
|
Nzinterval[int64, both])r rQ é r re ©r r¾ r¿ r# ) r r$ r Ú from_tuplesrè r% Úaranger rÀ r& r\ r@ r r r! Útest_basic_dtype s ÿ z"TestIntervalDtype.test_basic_dtypec C sf t ¡ t dd}tt jdks$tt d tt jdks>tt ¡ t |¡ tt jdksbtd S )Nr& rÅ rQ rÆ rd r )r r3 r4 r5 r$ r6 r7 r r r r! Ú test_caching s
|
|
z TestIntervalDtype.test_cachingc C s t t rtd S r )r r r$ r@ r r r! rm ! s z!TestIntervalDtype.test_not_stringc C s$ t d}|jd kstt |¡ d S ræ )r Ú_closedr$ r6 r7 r r r r! Ú test_unpickling_without_closed% s z0TestIntervalDtype.test_unpickling_without_closed)%r< r= r> r, rx r rB rz r{ r% r¹ rÎ rÏ rÐ r r¼ rY rk rÕ r× rÚ rÛ rG rß rà r r] rE r rã rå rç rc rî rï rm rñ r r r r! rÄ s
|
ÿ
|
ÿ
|
|
ùþ
|
|
|
|
öþ
|
ùþ
|
|
rÄ c @ sä e Zd Zej dede d¡dddddd gd d
|
ge d d d g¡dd Z
|
dd Z ej dd
|
dg¡dd Z dd Z dd Zej dddd gddd gfddd gd ddgfg¡d d Zd d Zd!d" Zd#d$ Zej d%d d
|
dg¡ej d&d d
|
dg¡d'd( Zd)d* Zej ded+dg¡ej d,d-d.g¡d/d0 Zd1d2 Zd3d4 Zd5d6 Zd7d8 Zd9d: Zd;d< Zd=d> Z d?d@ Z ej dAed+edBedCdg¡ej dDd d
|
dg¡dEdF Z dGdH Z ej dIdJe ej!e"dKg¡dLdM Z#dS )NÚ TestCategoricalDtypeParametrizedrT Zabcdiè rK rL é
|
rd gÍÌÌÌÌÌô?TFZ2017rê rë c C s2 t ||d}t |jt |¡¡ |j|ks.td S )NrM ©r r6 Úassert_index_equalrT ro rp rN r$ )r rT rN Úc1r r r! rc / s z+TestCategoricalDtypeParametrized.test_basicc C sH ddg}t |dd}t |dd}t |d d}||k s8t||k sDtd S )NrK rL TrM F©r r$ )r rT rö Úc2Úc3r r r! Útest_order_matters> s z3TestCategoricalDtypeParametrized.test_order_mattersrN Nc C s8 t ddg|d}t ddg|d}t|t|ks4td S )NrK rL rM ©r r r$ )r rN rö rø r r r! Útest_unordered_sameF s z4TestCategoricalDtypeParametrized.test_unordered_samec C s: t dddg}t |jt dddg¡¡ |jdks6td S )NrK rL r^ Frô )r r: r r r! Útest_categoriesL s z0TestCategoricalDtypeParametrized.test_categoriesc C s8 t dddg}t dddg}||k s(t||ks4td S )NrQ rd re r g @g @r÷ ©r rö rø r r r! Útest_equal_but_differentQ s z9TestCategoricalDtypeParametrized.test_equal_but_differentzv1, v2rQ re c C s@ t |dd}t |dd}t |d d}||k s0t||k s<td S )NFrM Tr÷ )r Zv1Zv2rö rø rù r r r! Útest_order_hashes_differentW s
|
z<TestCategoricalDtypeParametrized.test_order_hashes_differentc C s2 d}t jt|d tddtjg W 5 Q R X d S )Nz%Categorical categories cannot be nullr) rQ rd )r, r- rW r r% ÚnanrI r r r! Útest_nan_invalid_ s z1TestCategoricalDtypeParametrized.test_nan_invalidc C s0 d}t jt|d tdddg W 5 Q R X d S )Nz%Categorical categories must be uniquer) rQ rd )r, r- rW r rI r r r! Útest_non_unique_invalidd s z8TestCategoricalDtypeParametrized.test_non_unique_invalidc C s0 t ddgdd}t ddgdd}||k s,td S ru r÷ rþ r r r! Ú$test_same_categories_different_orderi s zETestCategoricalDtypeParametrized.test_same_categories_different_orderÚordered1Úordered2c C sî t td|}t td|}||k}t|t|k}||ks@tt td|}t td|}||k}t|dkozt|dk}||kstt dddg|}||ks¤tt td|}t d |}t d |}||ksÒt||ksÞt||ksêtd S )NrÑ ZcabFrQ rd re )r r¼ Úboolr$ )r r r rö rø r: rU rù r r r! Útest_categorical_equalityn s$
|
|
z:TestCategoricalDtypeParametrized.test_categorical_equalityc C s t ddg}t }t dd}||ks(t||ks4t||ks@t||ksLt||ksXt||ksdt||kspt||ks|td S ru r÷ )r ÚfirstÚsecondÚthirdr r r! Ú3test_categorical_dtype_equality_requires_categories s
|
zTTestCategoricalDtypeParametrized.test_categorical_dtype_equality_requires_categoriesrÑ ÚotherrD znot a categoryc C s* t ||}||k}|dk}||ks&td S )NrD r÷ )r rT rN r rö r: rU r r r! Ú!test_categorical_equality_strings s
|
zBTestCategoricalDtypeParametrized.test_categorical_equality_stringsc C sP t jtdd tddgdd W 5 Q R X t jtdd td W 5 Q R X d S ) NrN r) rK rL r# rM z 'categories' must be list-likerD rH r@ r r r! Útest_invalid_raises¨ s z4TestCategoricalDtypeParametrized.test_invalid_raisesc C s8 t ddddg}t ddddg}t|t|ks4td S )NrK rL rQ rd Ú1Ú2rû r r r r! Ú
|
test_mixed¯ s z+TestCategoricalDtypeParametrized.test_mixedc C s2 t ddgdddgdd}t |¡}||ks.td S )NrQ rd re Trn ©r r Z_from_categorical_dtyper$ rþ r r r! Ú$test_from_categorical_dtype_identity´ s
|
zETestCategoricalDtypeParametrized.test_from_categorical_dtype_identityc C sF t ddgdddgdd}tj|ddgd}|tddgddksBtd S )NrQ rd re Trn )rT rM r ©r rö r: r r r! Ú&test_from_categorical_dtype_categoriesº s zGTestCategoricalDtypeParametrized.test_from_categorical_dtype_categoriesc C sD t ddgdddgdd}tj|dd}|tdddgddks@td S ©NrQ rd re Trn FrM r r r r r! Ú#test_from_categorical_dtype_orderedÀ s zDTestCategoricalDtypeParametrized.test_from_categorical_dtype_orderedc C sH t ddgdddgdd}tj|ddgdd}|tddgddksDtd S r r r r r r! Ú test_from_categorical_dtype_bothÆ s ÿzATestCategoricalDtypeParametrized.test_from_categorical_dtype_bothc C sD t ddg|d}t|dks td}t |j|dt|¡s@td S )NrK rL rM rD z8CategoricalDtype\(categories=\[.*\], ordered={ordered}\))r rk r$ r r* Úformatrr )r rN rö Úpatr r r! Útest_str_vs_reprÎ s z1TestCategoricalDtypeParametrized.test_str_vs_reprc C sT t tddg}t |jt ddg¡¡ t tddg}t |jt ddg¡¡ d S )NrK rL )r r r6 rõ rT ro rp r )r rö r r r! Útest_categorical_categoriesÕ s z<TestCategoricalDtypeParametrized.test_categorical_categoriesÚnew_categoriesZcbarÒ Ú new_orderedc
|
C sf t d}t||}t||}| |¡}t |p0|¡}|d k r@|n|j} t |j|¡ |j| ksbt d S )NrÑ )
|
r¼ r rv ro rp rN r6 rõ rT r$ )
|
r rN r r Zoriginal_categoriesr Z new_dtyper: Úexpected_categoriesÚexpected_orderedr r r! rw Ü s
|
|
|
z2TestCategoricalDtypeParametrized.test_update_dtypec C sD t td|}|j}|j}| d¡}t |j|¡ |j|ks@td S )NrÑ rD )r r¼ rT rN rv r6 rõ r$ )r rN r r r! r: r r r! Útest_update_dtype_stringì s
|
z9TestCategoricalDtypeParametrized.test_update_dtype_stringÚ bad_dtyper# râ c C s: t tdd}d}tjt|d | |¡ W 5 Q R X d S )NrÑ Fz8a CategoricalDtype must be passed to perform an update, r) )r r¼ r, r- rW rv )r r# r r0 r r r! Útest_update_dtype_errorsô s z9TestCategoricalDtypeParametrized.test_update_dtype_errors)$r< r= r> r, rz r{ r¼ r% rí r rc rú rü rý rÿ r r r r r r r r r r r r r r r rw r" rY r& r r$ r r r r! rò . sX
|
ûþ
|
|
|
0
|
ÿ rò r c C s | t jkstd S r )ÚregistryZdtypesr$ ©r r r r! Ú test_registryü s r' zdtype, expected)r& NrÆ zinterval[int64, neither]z interval[datetime64[ns], left]rj rØ r² r rD r r} r~ c C s t | ¡|kstd S r )r% Úfindr$ )r rU r r r! Útest_registry_find s r) FTrK rL rQ rd c C s t | }||kstd S r )r r$ )r rU r: r r r! Útest_is_bool_dtype s r* c C s$ t ttddg} | dks td S )NTF)r r r r$ )r: r r r! Útest_is_bool_dtype_sparse' s r+ Úcheckc C sV t dddgi¡}t d ¡ | | W 5 Q R X t d ¡ | |d W 5 Q R X d S )Nr_ rQ rd )ro Ú DataFramer6 Zassert_produces_warning)r, Údatar r r! Útest_is_dtype_no_warning, s
|
r/ c C s. t dd} | dkdkst| dkdks*td S )NÚM)r¶ z period[M]TFrº r& r r r! Ú#test_period_dtype_compare_to_stringC s
|
r1 c C sr t t d¡ tj¡¡} d}tjt|d | | t ¡k W 5 Q R X tjt|d |
|
| t ¡¡ W 5 Q R X d S )NrÁ z>'<' not supported between instances of 'complex' and 'complex'r) ) ro r- r% rí Zastyperá r, r- r. rY Últ)Údfr0 r r r! Útest_compare_complex_dtypesJ s r4 c C sT t dgdd¡} t dgdd¡}d| ddg< t | |¡ d| d< t | |¡ d S )Ng rÞ r rK rL )ro r- r6 Zassert_frame_equal)r3 rU r r r! Ú"test_multi_column_dtype_assignmentV s r5 )=r Únumpyr% r, r¬ Zpandas._libs.tslibs.dtypesr Zpandas.core.dtypes.baser r% Zpandas.core.dtypes.commonr r r r r r r
|
r r r Zpandas.core.dtypes.dtypesr r r r Zpandasro r r r r r r Zpandas._testingZ_testingr6 Zpandas.core.arrays.sparser r r r? r| r¯ rÄ rò rz r{ r' r) rk Úintr Zbool_Úarrayr* r+ r/ r1 r4 r5 r r r r! Ú<module> s 0 % O
|
ÿ
|
|
ùþ
|
|
öþ
|
÷þ
|
|
