o
    Ebj                     @   sN   d Z ddlmZ ddlZddlZddlZddlm	Z	 ddl
ZG dd dZdS )zK
Ensure that we can use pathlib.Path objects in all relevant IO functions.
    )PathN)tempdirc                   @   sv   e Zd ZedejZdd Zdd Z	dd Z
dd	 Zd
d Zdd Zdd Zdd Zdd Zdd Zdd ZdS )	TestPaths   c                 C   sX   t  }t|d }tj|d| ji | sJ W d    d S 1 s%w   Y  d S Ndata.matdata)r   r   scipyiosavematr   is_file)selftemp_dirpath r   ;/usr/lib/python3/dist-packages/scipy/io/tests/test_paths.pytest_savemat   s
   "zTestPaths.test_savematc                 C   sr   t  ,}t|d }tjt|d| ji tj|}|d | jk s'J W d    d S 1 s2w   Y  d S r   )	r   r   r	   r
   r   strr   Zloadmatall)r   r   r   Zmat_contentsr   r   r   test_loadmat   s   "zTestPaths.test_loadmatc                 C   sl   t  )}t|d }tjt|d| ji tj|}|d dks$J W d    d S 1 s/w   Y  d S )Nr   r   r   )r   )   r   int64)r   r   r	   r
   r   r   r   Zwhosmat)r   r   r   contentsr   r   r   test_whosmat    s   "zTestPaths.test_whosmatc                 C      t tjd }tj| d S )Nzdata/scalar_string.sav)r   __file__parentr	   r
   Zreadsavr   r   r   r   r   test_readsav)      zTestPaths.test_readsavc                 C   |   t  1}tjtjd}t|d }tjt|| tj	|}||kj
dks,J W d    d S 1 s7w   Y  d S )N   data.hbr   )r   r	   sparse
csr_matrixeyer   r
   hb_writer   Zhb_readnnzr   r   r   r   Zdata_newr   r   r   test_hb_read-      "zTestPaths.test_hb_readc                 C   sf   t  &}tjtjd}t|d }tj|| | s!J W d    d S 1 s,w   Y  d S )Nr!   r"   )	r   r	   r#   r$   r%   r   r
   r&   r   r   r   r   r   r   r   r   test_hb_write7   s   "zTestPaths.test_hb_writec                 C   r    )Nr!   data.mtxr   )r   r	   r#   r$   r%   r   r
   mmwriter   Zmmreadr'   r(   r   r   r   test_mmio_read>   r*   zTestPaths.test_mmio_readc                 C   sZ   t   }tjtjd}t|d }tj|| W d    d S 1 s&w   Y  d S )Nr!   r-   )r   r	   r#   r$   r%   r   r
   r.   r+   r   r   r   test_mmio_writeH   s
   "zTestPaths.test_mmio_writec                 C   r   )Nzdata/example_1.nc)r   r   r   r	   r
   Znetcdf_filer   r   r   r   test_netcdf_fileN   r   zTestPaths.test_netcdf_filec                 C   s    t tjd }tjj| d S Nz"data/test-8000Hz-le-2ch-1byteu.wav)r   r   r   r	   r
   wavfilereadr   r   r   r   test_wavfile_readR   s   zTestPaths.test_wavfile_readc                 C   sp   t tjd }tjjt|\}}t }t ||j	 }tjj
||| W d    d S 1 s1w   Y  d S r2   )r   r   r   r	   r
   r3   r4   r   r   namewrite)r   Z
input_pathZrater   r   Zoutput_pathr   r   r   test_wavfile_writeV   s   "zTestPaths.test_wavfile_writeN)__name__
__module____qualname__npZarangeZastyper   r   r   r   r   r   r)   r,   r/   r0   r1   r5   r8   r   r   r   r   r      s    		

r   )__doc__Zpathlibr   Znumpyr<   Zscipy.ior	   Zscipy.io.wavfileZscipy._lib._tmpdirsr   Zscipy.sparser   r   r   r   r   <module>   s    