o
    ;Di                     @   s   d dl mZmZm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mZmZ d dlmZmZmZmZmZmZ eeZ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S )    )absolute_importprint_functiondivision)NamedTemporaryFileN)PY2)ieqeq_)fromcsvfromtsvtocsv	appendcsvtotsv	appendtsvc                  C   s^   g d} t ddd}|d|  |  d}t|jdd}t| t|| t|| d S )	N   foo,bar   a,1   b,2   c,2wbFmodedelete   
foobar)a1)b2)cr    asciiencodingr   writejoincloser	   namedebugr   datafexpectactual r0   N/var/www/Datamplify/venv/lib/python3.10/site-packages/petl/test/io/test_csv.pytest_fromcsv   s   
r2   c               	   C   s   g d} d}dD ]H}t t| tddd}|||  |  t|jd}t t|  W d    n1 s;w   Y  t	|jdd	}t | t
|| qd S )
Nr   r      r      
r   Fr   rbr"   r#   )r*   reprr   r&   r'   r(   openr)   readr	   r   )r,   r.   ltr-   gr/   r0   r0   r1   test_fromcsv_lineterminators(   s   r<   c                  C   sh   dd l } g d}tddd}|d| |  d}t|j| jd}t| t	|| t	|| d S )	Nr   )s   "foo","bar"s   "a",1s   "b",2s   "c",2r   Fr   r   r   )r      )r      )r!   r?   )quoting)
csvr   r&   r'   r(   r	   r)   QUOTE_NONNUMERICr*   r   )rA   r,   r-   r.   r/   r0   r0   r1   test_fromcsv_quoted@   s   
rC   c                  C   sV   g d} t ddd}|d|  |  d}t|jdd}t|| t|| d S )	N   foo	bar   a	1   b	2   c	2r   Fr   r   r   r"   r#   )r   r&   r'   r(   r
   r)   r   r+   r0   r0   r1   test_fromtsvT   s   
rI   c                  C   s   g d} d}t rd}nd}|D ]9}tdd}|  |jd }t|j| t|d}||	|  |  t
|d	d
}t|| t|| qd S )Nr   r   )r   r5   r3   Fr   .gzr   r"   r#   )r   r   r(   r)   osrenamegzipr8   r&   r'   r	   r   )r,   r.   ltsr:   r-   fnfzr/   r0   r0   r1   test_fromcsv_gzg   s"   


rR   c                  C      d} t dd}|  t| |jddd t|jd}g d}d	|d	 }| }t|| W d    n1 s9w   Y  d
}t||jddd t|jd}g d}d	|d	 }| }t|| W d    d S 1 sow   Y  d S )Nr=   FrJ   r"   
r$   lineterminatorr6   r   r   r   )d   )e	   )r-   r>   r   r   r   r   s   d,7s   e,9s   f,1)	r   r(   r   r)   r8   r'   r9   r   r   tabler-   or,   r.   r/   table2r0   r0   r1   test_tocsv_appendcsv   $   
"ra   c                  C   s|   d} t dd}t| |jdddd t|jd}g d}d	|d	 }| }t|| W d    d S 1 s7w   Y  d S )
Nr=   FrJ   r"   rT   )r$   rV   write_headerr6   r   r   r   r   )r   r   r)   r8   r'   r9   r   )r^   r-   r_   r,   r.   r/   r0   r0   r1   test_tocsv_noheader   s   
"re   c                  C   rS   )Nr=   FrJ   r"   rT   rU   r6   rD   r   rW   )rE   rF   rG   rH   s   d	7s   e	9s   f	1)	r   r(   r   r)   r8   r'   r9   r   r   r]   r0   r0   r1   test_totsv_appendtsv   rb   rf   c                  C   s   d} t dd}|jd }|  t| |ddd t|d}zg d	}d
|d
 }| }t|| W |  n|  w d}t	||ddd t|d}zg d}d
|d
 }| }t|| W |  d S |  w )Nr=   FrJ   rK   r"   rT   rU   r6   r   r   rW   r\   )
r   r)   r(   r   rN   r8   r'   r9   r   r   )r^   r-   rP   r_   r,   r.   r/   r`   r0   r0   r1   test_tocsv_appendcsv_gz   s*   

rg   c                  C   sh   ddg} g d}t ddd}|d| |  d}t|jd	| d
}t| t|| t|| d S )Nr   r   rd   r   Fr   r   r   r"   )r$   headerr%   )rh   r,   r-   r.   r/   r0   r0   r1   test_fromcsv_header&  s   
ri   )#
__future__r   r   r   tempfiler   rN   rL   loggingpetl.compatr   petl.test.helpersr   r   petl.io.csvr	   r
   r   r   r   r   	getLogger__name__loggerr*   r2   r<   rC   rI   rR   ra   re   rf   rg   ri   r0   r0   r0   r1   <module>   s(    
,,2