U
|
ý°d ã @ s¼ d dl mZ d dlZd dlZd dlZd dlmZ e
|
d¡Z ej d¡ej ddd eg¡dd Zej d¡d
|
d Zej d¡d d Zej d¡dd Zdd Zdd Zdd ZdS )é )ÚPathNÚ
|
pyreadstatz,ignore::pandas.errors.ChainedAssignmentErrorÚ
|
path_klassc C s | S )N© )Úpr r úPd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/io/test_spss.pyÚ<lambda> ó r c C s | |dddd}t j|dd}t jddid gd
|
}t |d ¡|d< t ||¡ t j|d d}t jdd id gd
|
}t ||¡ d S ) NÚioÚdataÚspssúlabelled-num.savT©Úconvert_categoricalsÚVAR00002ú This is oner )ÚindexFç ð?©ÚpdÚ read_spssÚ DataFrameÚ CategoricalÚtmÚassert_frame_equal)r ÚdatapathÚfnameÚdfÚexpectedr r r Útest_spss_labelled_num s r c C s~ | dddd}t j|dd}t ddd gi¡}t |d ¡|d< t ||¡ t j|d d}t dd
|
tjgi¡}t ||¡ d S ) Nr
|
r r zlabelled-num-na.savTr r r Fr )r r r r r r ÚnpÚnan©r r r r r r r Útest_spss_labelled_num_na s r# c C s| | dddd}t j|dd}t ddd gi¡}t |d ¡|d< t ||¡ t j|d
|
d}t dd d gi¡}t ||¡ d S ) Nr
|
r r zlabelled-str.savTr ZgenderZMaleZFemaleFÚMÚFr r" r r r Útest_spss_labelled_str- s r& c C s | dddd}t j|dd}t ddd dd
|
gi¡}t |d ¡|d< t ||¡ t j|d d}t dd d d dgi¡}t ||¡ d S )Nr
|
r r ú umlauts.savTr Úvar1u the ä umlautu the ü umlautu the ö umlautFr ç @ç @r r" r r r Útest_spss_umlauts< s ÿ r+ c C s: | dddd}t jtdd tj|dd W 5 Q R X d S ) Nr
|
r r r zusecols must be list-like.©Úmatchr )Zusecols)ÚpytestÚraisesÚ TypeErrorr r )r r r r r Útest_spss_usecolsM s r1 c s | dddd}t j|d|d}t jddd dd
|
gid d |d krtt d ¡ddlm t fddjD ¡t |¡ d S )Nr
|
r r r' F)r Ú dtype_backendr( r r) r* ZInt64)ZdtypeZpyarrowr )ÚArrowExtensionArrayc s$ i | ] }| j | d dqS )T)Z from_pandas)Úarray)Ú.0Úcol©r3 r Úpar r Ú
|
<dictcomp>b s ÿ z3test_spss_umlauts_dtype_backend.<locals>.<dictcomp>)
|
r r r r. Ú importorskipZ pandas.arraysr3 Úcolumnsr r )r r2 r r r r7 r Ú test_spss_umlauts_dtype_backendU s
|
þÿr< c C s0 d} t jt| d tjddd W 5 Q R X d S )NzPdtype_backend numpy is invalid, only 'numpy_nullable' and 'pyarrow' are allowed.r, ÚtestÚnumpy)r2 )r. r/ Ú
|
ValueErrorr r )Úmsgr r r Útest_invalid_dtype_backendk s ÿrA )Úpathlibr r> r r. Zpandasr Zpandas._testingZ_testingr r: r ÚmarkÚfilterwarningsZ parametrizer r# r& r+ r1 r< rA r r r r Ú<module> s"
|
|
|
|
|
|
|
|
|
