U
|
ý°dþ ã @ s d dl Z d dlZd dlZd dlZd dlmZ d dlZd dl m
|
Z d dl m Z mZ d dlmZ ejjeddZedd Zd d
|
Zd d Zeej d ddg¡ej dddg¡dd Zedd Zejje dddd Zeej dddg¡ej dd d d d d d!d gfd"d d d d d!d#gfd"dd dd d!d#gfd"ejd dd d!d#gfd"d$gd d d d d d#gfd"d$gd d%gd d d d%d#gfd"d$gd dgd d d dd#gfd"d$gd ejgd d d dd#gfd d gd d%gd d d d!d%gfd d gd d%gd%d d d!d gfe d$dd d d d d d gfe d&d'd d%gd d d d%d#gfe d$d"d d d%gd d d%d d#gfe dd d d d d d gfdddddgd d%gd d d d%d#gfg¡d(d) Z ed*d+ Z ed,d- Z dS ).é N)Úpa_version_under7p0)Ú StringArrayÚ StringDtype)ÚArrowStringArrayz9pyarrow>=7.0.0 is required for PyArrow backed StringArray)Úreasonc C sH t jt jt jgtdd} | | k}t jt jt jgdd}t ||¡ d S )NÚpyarrow©Údtypezboolean[pyarrow])ÚpdÚarrayÚNAr ÚtmZ assert_extension_array_equal)ÚaÚresultÚexpected© r údd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/string_/test_string_arrow.pyÚtest_eq_all_na s r c C sl t d| ¡4 t j| ks tt ddg¡}|jj| ks<tW 5 Q R X t| ¡ ddg¡}t
|
||¡ d S )NÚstring_storager Úb) r
|
Úoption_contextr ÚstorageÚAssertionErrorr r Zconstruct_array_typeÚ_from_sequencer Ú assert_equal)r r r r r r Ú test_config s ÿr c C s2 t d¡} tjt| d dtjj_W 5 Q R X d S )Nz#Value must be one of python|pyarrow©ÚmatchÚfoo) ÚreÚescapeÚpytestÚraisesÚ
|
ValueErrorr
|
ÚoptionsÚmoder ©Úmsgr r r Ú test_config_bad_storage_raises+ s
|
r( ÚchunkedTFr Únumpyr c C s dd l }| dkr|nt} | dddg¡}|rH| tkr>t d¡ | |¡}| tkrVd}n
|
t d¡}tjt |d t
|
| W 5 Q R X d S )
|
Nr r é é é z%chunked not applicable to numpy arrayzBUnsupported type '<class 'numpy.ndarray'>' for ArrowExtensionArrayzBArrowStringArray requires a PyArrow (chunked) array of string typer ) r Únpr r! ÚskipÚ chunked_arrayr r r" r# r )r r) ÚpaÚarrr' r r r Ú'test_constructor_not_string_type_raises1 s
|
|
ÿr3 c
|
C s¸ t dd¡ tjdd dgdd W 5 Q R X t dd¡ tjdd dgdd W 5 Q R X tjtd d tjdd dgd d W 5 Q R X tjdd dgd
|
d tjtd d4 t dd¡ tjdd dgt d W 5 Q R X W 5 Q R X t dd¡ tjdd dgt d W 5 Q R X tjtd d tjdd dgtdd W 5 Q R X tjdd dgtdd t dd¡ tjdd dgdd W 5 Q R X t dd¡ tjdd dgdd W 5 Q R X tjdd dgd d tjtd d tjdd dgd
|
d W 5 Q R X t dd¡ tjdd dgt d W 5 Q R X tjtd d4 t dd¡ tjdd dgt d W 5 Q R X W 5 Q R X tjdd dgtdd tjtd d tjdd dgtdd W 5 Q R X d S ) Nr Úpythonr ÚcÚstringr r r zstring[python]ústring[pyarrow]) r
|
r r r r! r" r r r r r r r Ú%test_from_sequence_wrong_dtype_raisesH s< * " *r8 zpyarrow is installedc C s~ t d¡} tjt| d tdd W 5 Q R X tjt| d tg W 5 Q R X tjt| d t dd dg¡ W 5 Q R X d S )Nz-pyarrow>=7.0.0 is required for PyArrow backedr r )r r r )r r r! r" Ú ImportErrorr r r r& r r r Ú!test_pyarrow_not_installed_raisesy s
|
r: Úmultiple_chunkszkey, value, expectedéÿÿÿÿZXXr r r5 Údr+ Úer- ZYYr, é c C s dd l }| td¡}| |¡}| r`| |d d
|dd
g¡}| |d d
|dd
g¡}t|}t|}|||< t ||¡ d S )Nr Úabcder- )r r Úlistr0 r r r )r; ÚkeyÚvaluer r1 r r r r Ú test_setitem s
|
rD c C s dd l } t| td¡}tjtd d d|d< W 5 Q R X tjtd d d|d< W 5 Q R X tjtd d d|ddg< W 5 Q R X tjtd d d|ddg< W 5 Q R X tjtd d d|dddg< W 5 Q R X tjtd d dd d
|
g|dd g< W 5 Q R X d S ) Nr r@ r r é iúÿÿÿTFÚbarZbazr+ )r r r rA r! r" Ú
|
IndexErrorr# )r1 r2 r r r Ú#test_setitem_invalid_indexer_raises± s rH c C st t jtddd} | d¡}t | ¡}t |¡}t|t|ksDtt |¡}t
|
|| ¡ t |¡}t
|
||¡ d S )Né
|
r7 r r, ) r
|
ZSeriesÚrangeÚheadÚpickleÚdumpsÚlenr Úloadsr Zassert_series_equal)r Zexpected_slicedZ full_pickledZsliced_pickledr Z result_slicedr r r Útest_pickle_roundtripÊ s
|
|
|
|
|
rP ) rL r r* r. r! Z pandas.compatr Zpandasr
|
Zpandas._testingZ_testingr Zpandas.core.arrays.string_r r Z pandas.core.arrays.string_arrowr ÚmarkZskipifZskip_if_no_pyarrowr r r( Z parametrizer3 r8 r: r ÚslicerD rH rP r r r r Ú<module> sj þ
|
|
0þ
|
ñþ
|
|
