a
    j=ic
                     @   s|   d dl mZ d dlZd dlZd dlZd dlmZ	 e
dZe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   j/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/pandas/tests/io/test_spss.py<lambda>       r   c                 C   s   | |dddd}t j|dd}t jddid	gd
}t |d |d< t|| t j|dd}t jddid	gd
}t|| d S )Niodataspsslabelled-num.savTZ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~   | dddd}t j|dd}t ddd gi}t |d |d< t|| t j|d	d}t dd
tjgi}t|| d S )Nr
   r   r   zlabelled-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|   | dddd}t j|dd}t ddd	gi}t |d |d< t|| t j|d
d}t dddgi}t|| d S )Nr
   r   r   zlabelled-str.savTr   ZgenderZMaleZFemaleFMFr   r!   r   r   r   test_spss_labelled_str)   s    r%   c                 C   s|   | dddd}t j|dd}t dg di}t |d |d< t|| t j|d	d}t dg d
i}t|| d S )Nr
   r   r   zumlauts.savTr   Zvar1)   the ä umlautu   the ü umlautr&   u   the ö umlautF)r   g       @r   g      @r   r!   r   r   r   test_spss_umlauts7   s    
r'   c                 C   sN   | dddd}t jtdd tj|dd W d    n1 s@0    Y  d S )	Nr
   r   r   r   zusecols must be list-like.)matchr   )Zusecols)pytestZraises	TypeErrorr   r   )r   r   r   r   r   test_spss_usecolsG   s    r+   )pathlibr   numpyr   r)   Zpandasr   Zpandas._testingZ_testingr   Zimportorskipr   markZparametrizer   r"   r%   r'   r+   r   r   r   r   <module>   s   

