U
    sVc!                     @   sF   d dl mZ d dlZd dlmZ d dlmZ G dd dZdd Z	dS )    )StringION)read_sasc                   @   s   e Zd Zdd Zdd ZdS )TestSasc              	   C   s2   t d}d}tjt|d t| W 5 Q R X d S )N zVIf this is a buffer object rather than a string name, you must specify a format stringmatch)r   pytestraises
ValueErrorr   )selfbmsg r   @/tmp/pip-unpacked-wheel-xj8nt62q/pandas/tests/io/sas/test_sas.pytest_sas_buffer_format
   s
    zTestSas.test_sas_buffer_formatc              
   C   s@   d}t d(}tjt|d t| W 5 Q R X W 5 Q R X d S )Nz$unable to infer format of SAS file.+Ztest_file_no_extensionr   )tmZensure_cleanr   r	   r
   r   )r   r   pathr   r   r   $test_sas_read_no_format_or_extension   s    z,TestSas.test_sas_read_no_format_or_extensionN)__name__
__module____qualname__r   r   r   r   r   r   r   	   s   r   c                 C   s@   | dddd}t |}| dddd}t |dd}t|| d S )NioZsasdatazairline.sas7bdatzairline.sas7bdat.gzZsas7bdat)format)r   r   Zassert_frame_equal)datapathZfname_uncompressedZdf_uncompressedZfname_compressedZdf_compressedr   r   r   test_sas_archive   s
    r   )
r   r   r   Zpandasr   Zpandas._testingZ_testingr   r   r   r   r   r   r   <module>   s
   