3
���h�
� @ sz d dl mZ d dlZd dlZd dlZd dljZ ej
d�Z ej j
ddd� eg�dd� �Zd d
� Zd d � Zd
d� Zdd� ZdS )� )�PathN�
pyreadstat�
path_klassc C s | S )N� )�pr r �;/tmp/pip-build-5_djhm0z/pandas/pandas/tests/io/test_spss.py�<lambda> s r c C s� | |dddd��}t j|dd�}t jddid gd
�}t j|d �|d< tj||� t j|d d�}t jdd id gd
�}tj||� d S )
N�io�data�spsszlabelled-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~ | dddd�}t j|dd�}t jddd gi�}t j|d �|d< tj||� t j|d d�}t jdd
tjgi�}tj||� d S ) Nr r
r zlabelled-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| | dddd�}t j|dd�}t jddd gi�}t j|d �|d< tj||� t j|d
d�}t jdd d gi�}tj||� d S )
Nr r
r zlabelled-str.savT)r ZgenderZMaleZFemaleF�M�F)r r r r r r )r r r r r r r �test_spss_labelled_str) s r c C s� | dddd�}t j|dd�}t jddd dd
gi�}t j|d �|d< tj||� t j|d d�}t jdd d
d dgi�}tj||� d S )Nr r
r z umlauts.savT)r Zvar1u
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: | dddd�}t jtdd�� tj|dd� W d Q R X d S ) Nr r
r zlabelled-num.savzusecols must be list-like.)�matchr
)Zusecols)�pytestZraises� TypeErrorr r )r r r r r �test_spss_usecolsG s r$ )Zpathlibr Znumpyr r" Zpandasr Zpandas._testingZ_testingr Z importorskipr �markZ parametrizer r r r r$ r r r r �<module> s
|