o
    qi	                     @   sn   d Z ddlmZ ddlZddlmZmZmZmZ ddl	m
Z
 e
d Zedd Zd	d
 Zdd Zdd ZdS )zTest IO Module    )PathN)remove_filedownload_from_urlpickle_loadpickle_dump)
MODELS_URLSENTIMENT_V1_WEIGHTS/c                 C   sB   d}|  }| d}|d}t|| t|}||ksJ dS )	Test CaseO   ترقی رکنے سے آہستہ آہستہ پاکستان نیچے چلاگیsub_dirz	hello.pklN)splitmkdirjoinr   r   )tmpdirtextdatatmp_dirtmp_fileloaded r   P/home/ubuntu/.local/lib/python3.10/site-packages/urduhack/utils/tests/test_io.pytest_pickle_dump   s   


r   c                 C   s   t jtdd tddddd W d   n1 sw   Y  t t tdtdd	d W d   n1 s7w   Y  | d
}t|d t }tttsPJ t|tsWJ ttts^J tttdt|d t	|
 spJ dS )r   strmatchtest_abc{   i  
cache_test)	file_nameurldownload_dir	cache_dirNi  subz/test_sentiment/test_sentiment)pytestraises	TypeErrorr   SENTIMENT_WEIGHTS_URLr   r   SENTIMENT_FILE_NAME
isinstancer   exists)r   temp_model_dir
model_pathr   r   r   test_download_from_url   s   
r/   c                 C   sB  t jtdd tdd W d   n1 sw   Y  t t tdd W d   n1 s1w   Y  | d}|d}t|d	d
d}|d W d   n1 sWw   Y  t	|}t
| shJ t
| spJ t|dd
d}| }|dksJ W d   n1 sw   Y  t| t
| du sJ dS )r   z[str]r   i@ )r    Nno_filer   z	hello.txtwzutf-8)encodingr   rF)r&   r'   r(   r   FileNotFoundErrorr   r   openwriter   r   r,   read)r   r   r   	temp_filer    	read_fileliner   r   r   test_remove_file+   s(   

r;   )__doc__pathlibr   r&   urduhack.utils.ior   r   r   r   urduhack.configr   r)   r   r*   r   r/   r;   r   r   r   r   <module>   s   