Ë
|
oñúh ã ó d Z ddlZddlZddlZddlmZ ddlZddlm Z ddl
|
m Z ddl mZ ddlmZ d Zej$ d « Zej$ d « Zej$ d
|
« Zej$ d « Zej$ d « Zej$ d « Z G d dej2 « Z G d dej6 « Z y)aj
|
This file contains a minimal set of tests for compliance with the extension
|
array interface test suite, and should contain no other tests.
|
The test suite for the full functionality of the array is located in
|
`pandas/tests/arrays/`.
|
|
The tests in this file are inherited from the BaseExtensionTests, and only
|
minimal tweaks should be applied to get the tests passing (by overwriting a
|
parent method).
|
|
Additional tests should either be added to one of the BaseExtensionTests
|
classes (if they are relevant for the extension interface for all dtypes), or
|
be added to the array-specific tests in `pandas/tests/arrays/`.
|
|
é N)Úusing_string_dtype©Ú Categorical©ÚCategoricalDtype)Úbasec ó° t j j d« j t t
|
j « d¬« } | d | d k7 r | S V)Né éd )Úsizer é )ÚnpÚrandomÚ default_rngÚchoiceÚlistÚstringÚ ascii_letters)Úvaluess úZH:\Change_password\venv_build\Lib\site-packages\pandas/tests/extension/test_categorical.pyÚ make_datar sT Ø
|
Ü×&Ñ& qÓ)×0Ñ0´´f×6JÑ6JÓ1KÐRUÐ0ÓVð !9q Ò !Ø Ø Mð ó c ó t « S ©Nr © r r Údtyper ) s ä Ó Ð r c ó( t t « « S )zLength-100 array for this type.
|
|
* data[0] and data[1] should both be non missing
|
* data[0] and data[1] should not be equal
|
)r r r r r Údatar . s ô y{Ó #Ð#r c ó8 t t j dg« S )z Length 2 array with [NA, Valid]ÚA)r r Únanr r r Ú data_missingr" 8 s ô }Ó %Ð%r c ó& t g d¢g d¢d¬« S )N)r ÚBÚC)r% r r$ T©Ú
|
categoriesÚorderedr r r r Údata_for_sortingr) > s ä ²?ÈDÔ QÐQr c ó& t g d¢ddgd¬« S )N)r Nr$ r$ r Tr& r r r r Údata_missing_for_sortingr+ C s ä Ò'°S¸#°JÈÔ MÐMr c ó t g d¢« S )N)Úar- NNÚbr. r- Úcr r r r Údata_for_groupingr0 H s ä ÒAÓ BÐBr c ó e Zd Zej j d¬« fd« Zd Zd Zej j d¬« fd« Z
|
ej j d¬« fd « Z d
|
Z ej j d d d g« d « Z fdZ fdZdej$ f fd Zej j d¬« ej j dddg« fd« « Zej j d¬« ej j dddg« fd « « Z xZS ) ÚTestCategoricalzMemory usage doesn't match©Úreasonc ó$ t | |« y r )ÚsuperÚtest_memory_usage©Úselfr Ú __class__s r r7 z!TestCategorical.test_memory_usageN s ø ô Ñ! $Õ'r c óö |j j }||j « }|d |v sJ |d |v sJ ||v sJ ||vsJ t j D ] }||u r||vsJ t « r||v r J y )Nr )r Úna_valueÚisnaÚtmÚ NULL_OBJECTSr )r9 r r" r< Ú na_value_objs r Ú test_containsz TestCategorical.test_containsS s© ð
|
::×&Ñ&àTYY[LÑ!ð Aw$Ð ØA ,Ñ.Ð.Ð.ð <Ñ'Ð'Ð'ØtÑ#Ð#Ð#ô OOò 4LØxÑ'ØØ tÑ+Ð +Ð+ä%Õ'Ø# |Ò3Ð3Ð3ñ 4r c ó |j « }|j d|¬« }t ||« sJ |j t g « k( sJ y )N)é )r )Úconstruct_array_typeÚ_emptyÚ
|
isinstancer r )r9 r ÚclsÚresults r Ú
|
test_emptyzTestCategorical.test_emptym sL Ø×(Ñ(Ó*ØD¨Ó.ä& #Ô&Ð&Ð&ð ||Ô /°Ó 3Ò3Ð3Ñ3r zBackwards compatibilityc ó$ t | |« y r )r6 Útest_getitem_scalarr8 s r rK z#TestCategorical.test_getitem_scalarv s ø ô
|
Ñ# DÕ)r z Unobserved categories includedc ó$ t | ||« S r )r6 Útest_value_counts)r9 Úall_dataÚdropnar: s r rM z!TestCategorical.test_value_counts} s ø äwÑ(¨°6Ó:Ð:r c ó2 |d« \ }}t j |« }t j |« }|j |d « }t j t t |« t |« « D cg c]
|
\ }}||z c}}« } t j || « |j d }
|
|j |
|
d « }t j t |« D cg c] }||
|
z c}« } t j || « y c c}}w c c}w )Nr
|
c ó | |z S r r ©Úx1Úx2s r ú<lambda>z2TestCategorical.test_combine_add.<locals>.<lambda> s
|
¨r°B©w r r c ó | |z S r r rR s r rU z2TestCategorical.test_combine_add.<locals>.<lambda> s
|
°°R± r )ÚpdÚSeriesÚcombineÚzipr r> Úassert_series_equalÚiloc) r9 Ú data_repeatedÚ
|
orig_data1Ú
|
orig_data2Ús1Ús2rH r- r. ÚexpectedÚvals r Útest_combine_addz TestCategorical.test_combine_add sã ñ "/¨qÓ!1Ñ
|
JÜ YYzÓ "Ü YYzÓ "ØBÑ 6Ó7Ü99Ü!$¤T¨*Ó%5´t¸JÓ7GÓ!H× Iv1QUÓ Ió
|
ô × Ñ v xÔ0àggajØCÑ!7Ó8Ü99¬t°JÓ/?Ö @¨!a #gÒ @ÓAÜ
|
× Ñ v xÕ0ùó Jùò As Á9D
|
Ã$ DÚ na_actionNÚignorec óX |j d |¬« }t j ||« y )Nc ó | S r r )Úxs r rU z*TestCategorical.test_map.<locals>.<lambda> s A r )re )Úmapr> Ú assert_extension_array_equal)r9 r re rH s r Útest_mapzTestCategorical.test_map s$ à+°Ó;Ü
|
×'Ñ'¨°Õ5r c ó |}|dk( r/|j t j j d¬« « t | ||« y ©NÚ__rmod__z/rmod never called when string is first argumentr3 )Ú applymarkerÚpytestÚmarkÚxfailr6 Ú test_arith_frame_with_scalar©r9 r Úall_arithmetic_operatorsÚrequestÚop_namer: s r rt z,TestCategorical.test_arith_frame_with_scalar sI ø à*Ø jÒ Ø × Ñ Ü ×!Ñ!ØLð "ó ô ô
|
Ñ,¨T°7Õ;r c ó |}|dk( r/|j t j j d¬« « t | ||« y rn )rp rq rr rs r6 Ú test_arith_series_with_scalarru s r rz z-TestCategorical.test_arith_series_with_scalar¢ sI ø Ø*Ø jÒ Ø × Ñ Ü ×!Ñ!ØLð "ó ô ô
|
Ñ-¨d°GÕ<r Úserc óÈ d|j d}|dvr0d}t j t |¬« 5 |||« d d d « y t | ||||« S # 1 sw Y y xY w)NÚ__)Ú__eq__Ú__ne__z7Unordered Categoricals can only compare equality or not)Úmatch)Ú__name__rq ÚraisesÚ TypeErrorr6 Ú_compare_other)r9 r{ r ÚopÚotherrx Úmsgr: s r r z TestCategorical._compare_other¬ sl ø Ør{{m 2Ð&Ø Ð.Ñ .ØKCÜ y°Ô4ñ Ù4÷ ð ô 7Ñ)¨#¨t°R¸Ó?Ð ?÷ ð ús ³
|
AÁA!z Categorical overrides __repr__r ÚbigÚsmallc ó& t | ||« y r )r6 Útest_array_repr)r9 r r r: s r r z TestCategorical.test_array_reprµ s ø ô Ñ dÕ+r ÚTBDÚas_indexTFc ó& t | ||« y r )r6 Útest_groupby_extension_agg)r9 r r0 r: s r r z*TestCategorical.test_groupby_extension_aggº s ø ô Ñ*¨8Ð5FÕGr )r Ú
|
__module__Ú __qualname__rq rr rs r7 rA rI ÚskiprK rM rd Ú parametrizerl rt rz rW rX r r r Ú __classcell__)r: s @r r2 r2 M sW ø Ø [[×ÑÐ :ÐÓ;ó(ó <ð(ò4ò44ð [[×ÑÐ 6ÐÓ7ó*ó 8ð*ð [[×ÑÐ >ÐÓ?ó;ó @ð;ò1ð" [[× Ñ [¨4°Ð*:Ó;ñ6ó <ð6ô <ô=ð@ "§)¡)õ @ð [[×ÑÐ >ÐÓ?Ø [[× Ñ V e¨WÐ%5Ó6ó,ó 7ó @ð,ð [[×ÑeÐÓ$Ø [[× Ñ Z¨$°¨ Ó7óHó 8ó %ôHr r2 c ó e Zd Zd Zy)Ú Test2DCompatc óÈ t |j dd« « }|j d« dk( sJ t |j dd« « }|j d« dk( sJ y )Nr éÿÿÿÿz
|
Categories)ÚreprÚreshapeÚcount)r9 r Úress r Ú test_repr_2dzTest2DCompat.test_repr_2dÁ s\ ô 4<< 2Ó&Ó'Øyy Ó(¨AÒ-Ð-Ð-ä4<< AÓ&Ó'Øyy Ó(¨AÒ-Ð-Ñ-r N)r r r r r r r r r À s ó.r r ) Ú__doc__r Únumpyr rq Úpandas._configr ÚpandasrW r Úpandas._testingÚ_testingr> Úpandas.api.typesr Úpandas.tests.extensionr r Úfixturer r r" r) r+ r0 ÚExtensionTestsr2 ÚNDArrayBacked2DTestsr r r r ú<module>r© së ðñó ã Û å -ã Ý Ý Ý -Ý 'òð ñ ó ð ð ñ$ó ð$ð ñ&ó ð&ð
|
ñRó ðRð ñNó ðNð ñCó ðCôpHd×)Ñ)ô pHôf.4×,Ñ,õ .r
|
