U
|
ý°dá$ ã @ sp d Z ddlZddlZddlZddlmZ ddl m
|
Z
|
m Z ddl m Z mZ ejdddZejjG dd d ZdS )
|
z test feather-format compat é N)ÚArrowStringArrayÚ StringArray)Ú read_featherÚ
|
to_featherÚpyarrowz1.0.1)Z
|
minversionc @ s¸ e Zd Zdd Zdd Zdi fddZdd Zd
|
d Zd d Zdd Z dd Z
|
dd Z dd Z dd Z dd Zdd Zd d Zd d Zejjejd d!d"d#d$ Zd%d& Zd'd( ZdS ))Ú TestFeatherc
|
C s< t j||d$ t ¡ }t|| W 5 Q R X W 5 Q R X d S )N©Úmatch)ÚpytestÚraisesÚtmÚ ensure_cleanr )ÚselfÚdfÚexcÚerr_msgÚpath© r úSd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/io/test_feather.pyÚcheck_error_on_write s
|
z TestFeather.check_error_on_writec
|
C s8 t t¡$ t ¡ }t|| W 5 Q R X W 5 Q R X d S ©N)r Zexternal_error_raisedÚ Exceptionr r )r r r r r r Ú check_external_error_on_write s
|
z)TestFeather.check_external_error_on_writeNc K sJ |d kr |}t ¡ ,}t||f| t|f|}t ||¡ W 5 Q R X d S r )r r r r Úassert_frame_equal)r r ÚexpectedÚ write_kwargsZ read_kwargsr Úresultr r r Úcheck_round_trip# s
|
z TestFeather.check_round_tripc C sH d}t dddg¡ddt d¡t dddg¡fD ]}| |t|¡ q0d S )Nz'feather only support IO with DataFramesé é é ÚfooÚ20130101)ÚpdÚSeriesÚ TimestampÚnpÚarrayr Ú
|
ValueError)r ÚmsgÚobjr r r Ú
|
test_error- s ûzTestFeather.test_errorc C s( t tdttddt dd¡ d¡tjddd d
|
d tjdgd d d gd tjd gt td¡t j tt j
|
dddd dt j tt j
|
ddddd dt d¡t j t d¡gt j tt j
|
ddddd dd ¡}t j dddd|d< t jddd|d < t d dd¡|d < |jjjjdkst| |¡ d S ) NÚabcr é r é Úu1g @g @Úfloat64©Údtypeg ð?TFr" ©Úperiods)Úfreqz
|
US/Eastern)r4 ÚtzZ20130103Úns)r4 r5 ) ÚstringÚintZuintÚfloatZfloat_with_nullÚboolZbool_with_nullÚcatÚdtÚdttzZ dt_with_nullZdtnsZ2013ÚM)r5 r4 r4 z1 dayZ
|
timedeltasr Z intervals)r# Ú DataFrameÚlistÚranger& ÚarangeZastypeÚnanZ CategoricalZ DatetimeIndexÚ
|
date_ranger% ZNaTZ period_rangeZtimedelta_rangeZinterval_ranger> r2 r6 ÚzoneÚAssertionErrorr ©r r r r r Ú
|
test_basic8 s@
|
|
ÿþý ÿëÿzTestFeather.test_basicc C s2 t jt d¡ dd¡tdd ¡ }| |¡ d S )Né r- r Zaaa©Úcolumns)r# r@ r& rC ÚreshaperA Úcopyr rH r r r Útest_duplicate_columns[ s $z"TestFeather.test_duplicate_columnsc C s2 t t d¡ dd¡¡ ¡ }d}| |t|¡ d S )NrJ r- r ú%feather must have string column names)r# r@ r& rC rM rN r r( )r r r) r r r Útest_stringify_columnsa s z"TestFeather.test_stringify_columnsc C sP t tdttddtdttddd¡}ddg}| j||| |d d S )
|
Nr, r r- Zxyzé )Úcol1Zcol2Úcol3Zcol4rS rT )r rL )r# r@ rA rB r )r r rL r r r Útest_read_columnsf s üÿz TestFeather.test_read_columnsc C sB t ddgddgddgd¡}|dd g }| j||dd gd
|
d S ) Nr r ÚxÚyTF)ÚAÚBÚCrY rX rK )r# r@ r )r r r r r r Ú!test_read_columns_different_orders s z-TestFeather.test_read_columns_different_orderc C s" t ddddgi¡}| |¡ d S )NÚar ç @)r# r@ r rH r r r Útest_unsupported_othery s z"TestFeather.test_unsupported_otherc C s4 t dt d¡i¡}| j|dd | j|dd d S )NrX i T)Z use_threadsF)r# r@ r& rC r rH r r r Útest_rw_use_threads~ s z TestFeather.test_rw_use_threadsc C sÊ t ddddgi¡}| |¡ d}dddgt jdddtd dddgt j d
|
d d g¡fD ]}||_| |t |¡ qXd ddg|_d|j_
|
d}| |t |¡ d ddg|_t j d
|
g¡|_ d}| |t |¡ d S )NrX r r r zsfeather does not support serializing .* for the index; you can \.reset_index\(\) to make the index into column\(s\)r- r" r3 r, )r\ r )r\ r )Úbr r r! z=feather does not serialize index meta-data on a default indexrP ) r# r@ r rE rA Z
|
MultiIndexÚ from_tuplesÚindexr r( ÚnamerL )r r r) rb r r r Útest_write_with_index s(
|
ÿ û z!TestFeather.test_write_with_indexc C s* t ¡ ¡ }t |jt¡}t ||¡ d S r )r Ú makeDataFrameÚ reset_indexZround_trip_pathlibr r r ©r r r r r r Útest_path_pathlib¢ s z TestFeather.test_path_pathlibc C s* t ¡ ¡ }t |jt¡}t ||¡ d S r )r re rf Zround_trip_localpathr r r rg r r r Útest_path_localpath§ s z TestFeather.test_path_localpathc C s" t ¡ ¡ }| j|ddid d S )NÚversionr )r )r re rf r rH r r r Útest_passthrough_keywords¬ s z%TestFeather.test_passthrough_keywordsúkhttps://raw.githubusercontent.com/pandas-dev/pandas/main/pandas/tests/io/data/feather/feather-0_3_1.featherT)ÚurlZcheck_before_testc C s$ d}t |}t |}t ||¡ d S )Nrl )r r r )r Z feather_filerm r Úresr r r Útest_http_path° s
|
ÿzTestFeather.test_http_pathc sò t d¡t tjdtjdgddtjdddgddtjdtjdgd dtjdd
|
dgd dd d d gd d d gd ddgd dd gd¡}|dkrÂttjd ddgtj d}ttjd dtj
|
gtj d}n(t d ddg¡}t d dd g¡}t ¡ 4}t|| t d|¡ t||d}W 5 Q R X W 5 Q R X t tjdtjdgddtjdddgddtjdtjdgd dtjdd
|
dgd dtjd d tj
|
gddtjd d d gdd||d¡|dkrâddlm t fddjD ¡t |¡ d S )Nr r r ZInt64r1 r g ø?g @ZFloat64r] TFr\ r` Úc)r\ r` rp ÚdÚeÚfÚgÚhÚpythonzmode.string_storage©Ú dtype_backendÚbooleanr )ÚArrowExtensionArrayc s$ i | ] }| j | d dqS )T)Z from_pandas)r' )Ú.0Úcol©rz r Úpar r Ú
|
<dictcomp>ð s ÿ z?TestFeather.test_read_feather_dtype_backend.<locals>.<dictcomp>)r
|
Ú importorskipr# r@ r$ r& rD r r' Zobject_ZNAr r r r Zoption_contextr Z pandas.arraysrz rL r )r Zstring_storagerx r Z string_arrayZstring_array_nar r r r} r Ú test_read_feather_dtype_backend sP
|
øÿ
|
|
øÿ
|
þÿz+TestFeather.test_read_feather_dtype_backendc
|
C sf d}t dttddi¡}t d¡6}| |¡ tjt |d t
|
|dd W 5 Q R X W 5 Q R X d S ) NzPdtype_backend numpy is invalid, only 'numpy_nullable' and 'pyarrow' are allowed.r9 r r- z tmp.featherr Únumpyrw ) r# r@ rA rB r r r r
|
r r( r )r r) r r r r r Útest_invalid_dtype_backendø s ÿ
|
z&TestFeather.test_invalid_dtype_backend)Ú__name__Ú
|
__module__Ú __qualname__r r r r+ rI rO rQ rU r[ r^ r_ rd rh ri rk r
|
ÚmarkÚnetworkr ro r r r r r r r s.
|
# û
|
6r )Ú__doc__r r& r
|
Zpandasr# Zpandas._testingZ_testingr Zpandas.core.arraysr r Zpandas.io.feather_formatr r r r r Z
|
single_cpur r r r r Ú<module> s
|
