U
|
ý°dI$ ã @ s d Z ddlZddlZddlZddlmZ ddlZddl m
|
Z ddl m Z ddlmZ ddlmZ ddlmZ dd Zejd
|
d gd d d Zejdd Zejdd Zejdd Zejdd Zejdd Zejdd Z ejdd Z G d d d ej Z G d d d ej Z!G d!d" d"ej"Z#G d#d$ d$ej$Z%G d%d& d&ej&Z'G d'd( d(ej(Z)G d)d* d*ej*Z+G d+d, d,ej,Z-G d-d. d.ej.Z/G d/d0 d0ej0Z1G d1d2 d2ej2Z3G d3d4 d4ej4Z5G d5d6 d6ej6Z7G d7d8 d8ej8Z9G d9d: d:ej:Z;G d;d< d<ej<Z=d=d> Z>dS )?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)ÚPerformanceWarning)Úis_string_dtype)ÚArrowStringArray©Ú StringDtype)Úbasec C s& | j jdkrt d¡ dd }|| S )NÚpyarrowz-only applicable for pyarrow chunked array n/ac S sV dd l }| j}t|d }| |d |
j||d
j¡}|jdksJtt| |S )Nr é )r Ú_dataÚlenZ chunked_arrayÚchunksZ
|
num_chunksÚAssertionErrorÚtype)ÚarrÚpaZ arrow_arrayÚsplit© r úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/extension/test_string.pyÚ _split_array# s ÿz!split_array.<locals>._split_array)ÚdtypeÚstorageÚpytestÚskip)r r r r r Ú split_array s
|
r TF)Úparamsc C s | j S ©N)Úparam)Úrequestr r r Úchunked1 s r c C s
|
t | dS )N©r r )Zstring_storager r r r 6 s r c C s\ t jjttjdd}|d |d kr>t jjttjdd}q| ¡ |¡}|rXt|S |S )Néd )Úsizer é ) ÚnpÚrandomÚchoiceÚlistÚstringÚ ascii_lettersÚconstruct_array_typeÚ_from_sequencer )r r Ústringsr r r r Údata; s
|
r, c C s$ | ¡ tjdg¡}|r t|S |S )z Length 2 array with [NA, Valid]ÚA©r) r* ÚpdÚNAr ©r r r r r r Ú data_missingE s r2 c C s$ | ¡ dddg¡}|r t|S |S )NÚBÚCr- )r) r* r r1 r r r Údata_for_sortingL s r5 c C s& | ¡ dtjdg¡}|r"t|S |S )Nr3 r- r. r1 r r r Údata_missing_for_sortingR s r6 c C s t jS r )r/ r0 r r r r Úna_valueX s r7 c
|
C s2 | ¡ ddtjtjddddg¡}|r.t|S |S )Nr3 r- r4 r. r1 r r r Údata_for_grouping] s ÿr8 c s$ e Zd Z fddZdd Z ZS )Ú TestDtypec s& |d|j dkstt |¡ d S )Nzstring[ú])r r ÚsuperÚtest_eq_with_str©Úselfr ©Ú __class__r r r< f s zTestDtype.test_eq_with_strc C s t |s td S r )r r r= r r r Útest_is_not_string_typej s z!TestDtype.test_is_not_string_type)Ú__name__Ú
|
__module__Ú __qualname__r< rA Ú __classcell__r r r? r r9 e s r9 c s e Zd Z fddZ ZS )Ú TestInterfacec s( |j jdkrtjdd t |¡ d S ©Nr ú/2D support not implemented for ArrowStringArray©Úreason)r r r r r; Ú test_view©r> r, r r? r r rK q s zTestInterface.test_view)rB rC rD rK rE r r r? r rF p s rF c @ s e Zd Zdd Zdd ZdS )ÚTestConstructorsc C s d S r r )r> r, r r r Útest_from_dtypex s z TestConstructors.test_from_dtypec C sH t jddd tjdgtddd}t|jts4t|jjdksDtd S )Nr z1.0.0)Z
|
minversionÚEr ©r ) r Z importorskipr/ ÚSeriesr Ú
|
isinstancer r r )r> Úresultr r r Útest_constructor_from_list| s z+TestConstructors.test_constructor_from_listN)rB rC rD rN rT r r r r rM w s rM c s e Zd Z fddZ ZS )Ú TestReshapingc s( |j jdkrtjdd t |¡ d S rG )r r r r r; Útest_transposerL r? r r rV
s z TestReshaping.test_transpose)rB rC rD rV rE r r r? r rU s rU c @ s e Zd ZdS )Ú TestGetitemN©rB rC rD r r r r rW s rW c s e Zd Z fddZ ZS )Ú TestSetitemc s( |j jdkrtjdd t |¡ d S rG )r r r r r; Ú test_setitem_preserves_viewsrL r? r r rZ s z(TestSetitem.test_setitem_preserves_views)rB rC rD rZ rE r r r? r rY s rY c @ s e Zd ZdS )Ú TestIndexNrX r r r r r[ s r[ c s, e Zd Zdd Zdd Z fddZ ZS )Ú TestMissingc C s" | ¡ }|dg }| ||¡ d S )Nr" )ZdropnaÚ assert_extension_array_equal)r> r2 rS Úexpectedr r r Útest_dropna_array s
|
z TestMissing.test_dropna_arrayc C s || ¡ }|d }| |¡}||k s,t| ||¡ t t|jjdk¡ |jdd}W 5 Q R X ||k spt| ||¡ d S )Nr r Zbackfill)Úmethod) ÚisnaZfillnar r] ÚtmÚmaybe_produces_warningr r r )r> r, ZvalidrS r r r Ú test_fillna_no_op_returns_copy s
|
|
ÿ z*TestMissing.test_fillna_no_op_returns_copyc s> t jt|d k o|jjdkdd t ||¡ W 5 Q R X d S )Nr F)Zcheck_stacklevel)rb rc r r r r; Útest_fillna_series_method)r> r2 Z fillna_methodr? r r re ¯ s ýz%TestMissing.test_fillna_series_method)rB rC rD r_ rd re rE r r r? r r\ s r\ c @ s& e Zd Zej dddg¡dd ZdS )Ú TestNoReduceÚskipnaTFc C sD |}|dkrd S t |¡}t t¡ t|||d W 5 Q R X d S )N)ÚminÚmax)rg )r/ rQ r ÚraisesÚ TypeErrorÚgetattr)r> r, Zall_numeric_reductionsrg Úop_nameÚserr r r Útest_reduce_series_numeric¹ s
|
z'TestNoReduce.test_reduce_series_numericN)rB rC rD r ÚmarkÚ parametrizero r r r r rf ¸ s rf c @ s e Zd Zdd ZdS )Ú TestMethodsc C s |d d
¡ }t || ¡ ¡}tj||jd}|jdd ¡ }tjdt | gt | |j
|
dd}t |jdd d
|
kr| d ¡}n
|
| d ¡}| ||¡ d S ) Né
|
rP T)Ú normalizer" Z
|
proportion©ÚindexÚnamer Ú r zdouble[pyarrow]ZFloat64)Úuniquer# Úarrayra r/ rQ r Z value_countsZ
|
sort_indexr rv rl ÚastypeÚassert_series_equal)r> r, Úvaluesrn rS r^ r r r Ú test_value_counts_with_normalizeÆ s ÿ
|
z,TestMethods.test_value_counts_with_normalizeN)rB rC rD r~ r r r r rr Å s rr c @ s e Zd ZdS )Ú TestCastingNrX r r r r r Ø s r c @ s e Zd Zdd Zdd ZdS )ÚTestComparisonOpsc C sZ d|j d}t|||}|jjdkr,dnd}t| t¡|| |¡}| ||¡ d S )NÚ__r zboolean[pyarrow]Úboolean)rB rl r r r{ Úobjectr| ) r> rn r, ÚopÚotherrm rS r r^ r r r Ú_compare_otherÝ s
|
z TestComparisonOps._compare_otherc C s t |¡}| |||d¡ d S )NÚabc)r/ rQ r )r> r, Z comparison_oprn r r r Útest_compare_scalarä s
|
z%TestComparisonOps.test_compare_scalarN)rB rC rD r r r r r r r Ü s r c @ s e Zd ZdS )Ú TestParsingNrX r r r r r é s r c @ s e Zd ZdS )Ú TestPrintingNrX r r r r r í s r c sB e Zd Zej dddg¡dd Zej d¡ fddZ Z S ) Ú TestGroupByÚas_indexTFc
|
C s¢ t ddddddddg|d¡}|jd|dj ¡ }t j|dd \}}|r|t j|dd
|
}t jd d d g|dd}| ||¡ n"t |d d d gd¡}| ||¡ d S )Nr" r é é )r- r3 r3 )r T)Úsort)rw g @g ð?g @r- ru )r3 r- )
|
r/ Z DataFrameÚgroupbyr- ZmeanZ factorizeZIndexrQ r| Zassert_frame_equal) r> r r8 ZdfrS Ú_Zuniquesrv r^ r r r Útest_groupby_extension_aggò s z&TestGroupBy.test_groupby_extension_aggz4ignore:Falling back:pandas.errors.PerformanceWarningc s t ||¡ d S r )r; Ú test_groupby_extension_apply)r> r8 Zgroupby_apply_opr? r r r s z(TestGroupBy.test_groupby_extension_apply)
|
rB rC rD r rp rq r Úfilterwarningsr rE r r r? r r ñ s
|
|
r c @ s e Zd Zejdddd ZdS )Ú Test2DCompatT)Zautousec C s t |trtjdd d S )NrH rI )rR r r r rL r r r Úarrow_not_supported s
|
z Test2DCompat.arrow_not_supportedN)rB rC rD r Úfixturer r r r r r s
|
r c C s^ | \}}}| dddg¡}tj|d< |r2t |¡}d}tjt|d | |¡ W 5 Q R X d S )Nr r r" éÿÿÿÿzOsearchsorted requires array to be sorted, which is impossible with NAs present.)Úmatch)Ztaker/ r0 rQ r rj Ú
|
ValueErrorZ searchsorted)r5 Z as_seriesÚbÚcÚar Úmsgr r r Ú test_searchsorted_with_na_raises s
|
|
|
ÿr )?Ú__doc__r' Únumpyr# r Z pandas.errorsr Zpandasr/ Zpandas._testingZ_testingrb Zpandas.api.typesr Zpandas.core.arraysr Zpandas.core.arrays.string_r Zpandas.tests.extensionr r r r r r, r2 r5 r6 r7 r8 ZBaseDtypeTestsr9 ZBaseInterfaceTestsrF ZBaseConstructorsTestsrM ZBaseReshapingTestsrU ZBaseGetitemTestsrW ZBaseSetitemTestsrY ZBaseIndexTestsr[ ZBaseMissingTestsr\ ZBaseNoReduceTestsrf ZBaseMethodsTestsrr ZBaseCastingTestsr ZBaseComparisonOpsTestsr ZBaseParsingTestsr ZBasePrintingTestsr ZBaseGroupbyTestsr ZDim2CompatTestsr r r r r r Ú<module> sX
|
|
|
|
|
|
|
|
|
