3
Ud
                 @   sz   d dl mZ d dlZd dlZd dlZd dljZ	ej
dZejjddd egdd Zd	d
 Zdd Zdd Zdd ZdS )    )PathN
pyreadstat
path_klassc             C   s   | S )N )pr   r   M/var/www/html/virt/lib64/python3.6/site-packages/pandas/tests/io/test_spss.py<lambda>   s    r   c             C   s   | |dddd}t j|dd}t jddid	gd
}t j|d |d< tj|| t j|dd}t jddid	gd
}tj|| d S )Niodataspsszlabelled-num.savT)convert_categoricalsVAR00002zThis is oner   )indexFg      ?)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~   | dddd}t j|dd}t jddd gi}t j|d |d< tj|| t j|d	d}t jdd
tjgi}tj|| d S )Nr	   r
   r   zlabelled-num-na.savT)r   r   zThis is oneFg      ?)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|   | dddd}t j|dd}t jddd	gi}t j|d |d< tj|| t j|d
d}t jdddgi}tj|| d S )Nr	   r
   r   zlabelled-str.savT)r   ZgenderZMaleZFemaleFMF)r   r   r   r   r   r   )r   r   r   r   r   r   r   test_spss_labelled_str)   s    r   c             C   s   | dddd}t j|dd}t jddd	dd
gi}t j|d |d< tj|| t j|dd}t jdddddgi}tj|| d S )Nr	   r
   r   zumlauts.savT)r   Zvar1u   the ä umlautu   the ü umlautu   the ö umlautFg      ?g       @g      @)r   r   r   r   r   r   )r   r   r   r   r   r   r   test_spss_umlauts7   s    r    c             C   s:   | dddd}t jtdd tj|dd W d Q R X d S )	Nr	   r
   r   zlabelled-num.savzusecols must be list-like.)matchr   )Zusecols)pytestZraises	TypeErrorr   r   )r   r   r   r   r   test_spss_usecolsG   s    r$   )pathlibr   Znumpyr   r"   Zpandasr   Zpandas._testingZ_testingr   Zimportorskipr   markZparametrizer   r   r   r    r$   r   r   r   r   <module>   s   

