File: //usr/local/lib/python3.7/test/__pycache__/test_csv.cpython-37.pyc
B
��g=� � @ s� d dl Z d dlZd dlZd dlmZ d dlmZ d dlZd dlZd dl Z d dl
m Z d dl m
Z
d dlmZ d dlmZ G dd � d ej�ZG d
d � d ej�ZG d d
� d
ej�ZG dd� de�ZG dd� dej�ZG dd� de�ZG dd� de�ZG dd� dej�ZG dd� de�Z G dd� dej�Z G d d � d ej�Z G d d � d ej�Z G d d!� d!ej�Z G d"d#� d#�Z!e�"e#ed$�d%�G d&d'� d'ej��Z$G d(d)� d)ej�Z%G d*d+� d+ej�Z&G d,d-� d-ej�Z'e(d.k�r�e�)� dS )/� N)�StringIO)�
TemporaryFile)�support)� permutations)�dedent)� OrderedDictc @ s e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zejd,d-� �Zd.d/� Z d0d1� Z d2d3� Z d4d5� Z d6d7� Z d8d9� Z!d:d;� Z"d<d=� Z#d>d?� Z$d@dA� Z%dBS )C�Test_Csvz�
Test the underlying C csv parser in ways that are not appropriate
from the high level interface. Further tests of this nature are done
in TestDialectRegistry.
c C s� | � t|� | � t|d � | j t||dd� | j t||dd� | j t||dd� | � tj||d� | j t||d d� | j t||dd� | j t||dd� | j t||d d� | j t||dd� | j t||d d � | j t||tjd
d � | j t||tjd d � d S ) Nr )�bad_attr)� delimiterZXXZfoo� )� quotechar)�lineterminator)�quoting� )r r )� assertRaises� TypeError�csv�Error� QUOTE_ALL)�self�ctor�arg� r �)/usr/local/lib/python3.7/test/test_csv.py�_test_arg_valid s
zTest_Csv._test_arg_validc C s | � tjg � d S )N)r r �reader)r r r r �test_reader_arg_valid) s z Test_Csv.test_reader_arg_validc C s | � tjt� � d S )N)r r �writerr )r r r r �test_writer_arg_valid, s z Test_Csv.test_writer_arg_validc G s� ||� }| � |jjd� | � |jjd� | � |jjd � | � |jjd� | � |jjd� | � |jjtj � | � |jj
d� | � |jj d� | � t
t|jd� | � t
t|jdd� | � t
t|jd� | � t
t|jdd � d S ) N�,Tz
�"Fr
�:r )� assertEqual�dialectr
� doublequote�
escapecharr
r r r �
QUOTE_MINIMAL�skipinitialspace�strictr �AttributeError�delattr�setattr)r r �args�objr r r �_test_default_attrs/ s z Test_Csv._test_default_attrsc C s | � tjg � d S )N)r. r r )r r r r �test_reader_attrsA s zTest_Csv.test_reader_attrsc C s | � tjt� � d S )N)r. r r r )r r r r �test_writer_attrsD s zTest_Csv.test_writer_attrsc
G s� t dddddtjddd�}|||�}| �|jjd� | �|jjd� | �|jjd� | �|jjd� | �|jj d� | �|jj
tj� | �|jj d� | �|jj d� d S )Nr! F�\�
�*T)r
r$ r% r
r r r' r( )
�dictr �
QUOTE_NONEr" r# r
r$ r% r
r r r' r( )r r r, �kwargsr- r r r �_test_kw_attrsG s
zTest_Csv._test_kw_attrsc C s | � tjg � d S )N)r7 r r )r r r r �test_reader_kw_attrsW s z Test_Csv.test_reader_kw_attrsc C s | � tjt� � d S )N)r7 r r r )r r r r �test_writer_kw_attrsZ s z Test_Csv.test_writer_kw_attrsc G s� G dd� d�}||f }||� }| � |jjd� | � |jjd� | � |jjd� | � |jjd� | � |jjd� | � |jjtj � | � |jj
d� | � |jj d� d S ) Nc @ s. e Zd ZdZdZdZdZdZej Z
dZ dZ dS )z-Test_Csv._test_dialect_attrs.<locals>.dialect�-F�^�$�#TN)
�__name__�
__module__� __qualname__r
r$ r% r
r r r r r' r( r r r r r# _ s r# r: Fr; r< r= T) r" r# r
r$ r% r
r r r r r' r( )r r r, r# r- r r r �_test_dialect_attrs] s
z Test_Csv._test_dialect_attrsc C s | � tjg � d S )N)rA r r )r r r r �test_reader_dialect_attrss s z"Test_Csv.test_reader_dialect_attrsc C s | � tjt� � d S )N)rA r r r )r r r r �test_writer_dialect_attrsv s z"Test_Csv.test_writer_dialect_attrsc K sV t ddd��@}tj|f|�}|�|� |�d� | �|�� ||jj � W d Q R X d S )Nzw+r )�newliner ) r r r �writerow�seekr" �readr# r
)r �fields�expectr6 �fileobjr r r r � _write_testz s
zTest_Csv._write_testc
K sd t ddd��N}tj|f|�}| �|�� |�|� W d Q R X |�d� | �|�� d� W d Q R X d S )Nzw+r )rD r )r r r r rE rF r" rG )r �excrH r6 rJ r r r r �_write_error_test� s
zTest_Csv._write_error_testc C s| | � tjd � | �dd� | �d gd� | j tjd gtjd� G dd� d�}| � t|� � G dd� d�}| � t|� g� d S ) Nr r z"")r c @ s e Zd Zdd� Zdd� ZdS )z.Test_Csv.test_write_arg_valid.<locals>.BadListc S s dS )N�
r )r r r r �__len__� s z6Test_Csv.test_write_arg_valid.<locals>.BadList.__len__c S s |dkr t �d S )N� )�OSError)r �ir r r � __getitem__� s z:Test_Csv.test_write_arg_valid.<locals>.BadList.__getitem__N)r> r? r@ rO rS r r r r �BadList� s rT c @ s e Zd Zdd� ZdS )z.Test_Csv.test_write_arg_valid.<locals>.BadItemc S s t �d S )N)rQ )r r r r �__str__� s z6Test_Csv.test_write_arg_valid.<locals>.BadItem.__str__N)r> r? r@ rU r r r r �BadItem� s rV )rM r r rK r5 rQ )r rT rV r r r �test_write_arg_valid� s z Test_Csv.test_write_arg_validc C s$ dd }| � ||gd||f � d S )N�XiP� z%s,%s)rK )r � bigstringr r r �test_write_bigfield� s z Test_Csv.test_write_bigfieldc C s� | � dddgd� | jtjdddgtjd� | j dddgdtjd� | j dddgdtjd� | j dddgdtjd� | j ddgd tjd� d S )
N�ar zp,qz a,1,"p,q")r z "a",1,"p,q"z
"a","1","p,q"za
bz "a
b","1")rK rM r r r5 r& �QUOTE_NONNUMERICr )r r r r �test_write_quoting� s
zTest_Csv.test_write_quotingc C s� | j dddgddd� | jtjdddgd dd � | j dddgd
ddd � | j d gd dtjd
� | j d gddtjdd� | j d gddtjd
� | j dddgddtjd
� d S )Nr[ r zp,qz a,1,"p,q"r1 )r% zp,"q"F)r% r$ z
a,1,"p,\"q\""r z"""")r% r z\")r% r r$ za,1,p\,q)rK rM r r r& r5 )r r r r �test_write_escape� s
zTest_Csv.test_write_escapec C s� | � tdddg�d� | � tddd g�d� | � tg �d� | � td g�d� | jtjtd g�tjd� | � td d g�d � d S )
Nr[ r zp,qz a,1,"p,q"za,1,r z"")r r )rK �iterrM r r r5 )r r r r �test_write_iterable� s z Test_Csv.test_write_iterablec C s� G dd� d�}t �|� �}| �t|jdgg� tddd��P}t �|�}| �t|jd � |�ddgdd gg� |�d
� | �|� � d � W d Q R X d S ) Nc @ s e Zd Zdd� ZdS )z+Test_Csv.test_writerows.<locals>.BrokenFilec S s t �d S )N)rQ )r �bufr r r �write� s z1Test_Csv.test_writerows.<locals>.BrokenFile.writeN)r> r? r@ rb r r r r �
BrokenFile� s rc r[ zw+r )rD �b�c�dr z
a,b
c,d
)
r r r rQ � writerowsr r rF r" rG )r rc r rJ r r r �test_writerows� s
zTest_Csv.test_writerowsc C s� t ddd��@}t�|�}|�dd gd dgg� |�d� | �|�� d� W d Q R X t ddd��<}t�|�}|�d gdgg� |�d� | �|�� d� W d Q R X t ddd��<}t�|�}|�dgd gg� |�d� | �|�� d � W d Q R X d S )
Nzw+r )rD r[ rf r za,
,d
z""
a
za
""
)r r r rg rF r" rG )r rJ r r r r �test_writerows_with_none� s
z!Test_Csv.test_writerows_with_nonec C s` dd l }|�d�}tddd��8}t�|�}|�|gg� |�d� | �|�� d� W d Q R X d S )Nr r[ zw+r )rD za
) � _testcapiZunicode_legacy_stringr r r rg rF r" rG )r rj re rJ r r r r � test_writerows_legacy_strings� s
z&Test_Csv.test_writerows_legacy_stringsc K s& t j|f|�}t|�}| �||� d S )N)r r �listr" )r �inputrI r6 r �resultr r r �
_read_test� s zTest_Csv._read_testc C s� | � g g � | � dgg g� | jtj| j dgd dd� | jtj| j dgd dd� | j dgdggdd� | �tj| j d gd � d S )
Nr z"ab"cr )r( zab c�abcr )r$ s ab c)ro r r r )r r r r �test_read_oddinputs� s z Test_Csv.test_read_oddinputsc C s� | � dgddgg� | � dgddgg� | � dgddgg� | � dgddgg� | �tj| j dgg � | �tj| j dgg � | �tj| j d gg � d S )
Nza,br[ rd za,b
za,b
za,b
za,b
c,dza,b
c,dza,b
c,d)ro r r r )r r r r �
test_read_eol s zTest_Csv.test_read_eolc C s� | � dgddgg� | � dgdgg� | j dgdggdd� | jtj| j dgg dd � | jtj| j dgg dd � | jtj| j dgg ddd
� d S ) Nza,"r[ r z"ar; �
)r% T)r( )r% r( )ro r r r )r r r r �
test_read_eof s zTest_Csv.test_read_eofc C s� | j dgdddggdd� | j dgddggdd� | j d gddggdd� | j d
gddggdd� | j d gdd ggdd� | j d
gddggdd� d S )Nza,\b,cr[ rd re r1 )r% za,b\,czb,cza,"b\,c"za,"b,\c"z a,"b,c\""zb,c"za,"b,c"\zb,c\)ro )r r r r �test_read_escape s zTest_Csv.test_read_escapec C s� | � dgdddgg� | j dgdddddggd dd� | j dgdddddggtjdd � | j d
gd d dd
dggtjd� | � dgddgg� | jt| j dgg gtjd� d S )Nz 1,",3,",5�1z,3,�5r �3r1 )r r% )r r% z
,3,"5",7.3, 9r � g333333 @� )r z"a
b", 7za
bz 7zabc,3)ro r r5 r\ r �
ValueError)r r r r �test_read_quoting! s
zTest_Csv.test_read_quotingc C s� t �� }z�d}d| }d||f }| �|g||gg� t �|� | �|g||gg� | �t �� |� t �|d � | �t j| j|gg � | �tt jd � | �tt jdd � W d t �|� X d S )NiP� rX z%s,%sr )r Zfield_size_limitro r" r r r )r �limit�sizerY Zbigliner r r �test_read_bigfield/ s
zTest_Csv.test_read_bigfieldc C s� t �dddg�}| �|jd� t|� | �|jd� t|� | �|jd� t|� | �|jd� | �tt|� | �|jd� d S )Nzline,1zline,2zline,3r r rP ry )r r r" Zline_num�nextr �
StopIteration)r �rr r r �test_read_linenumB s zTest_Csv.test_read_linenumc C s� t ddd��p}t�|�}| �t|jd � ddgddgg}|�|� |�d� x*tt�|��D ]\}}| � ||| � q\W W d Q R X d S ) Nzw+r )rD za
brd re zx
dr )
r r r r r rg rF � enumerater r" )r rJ r �rowsrR �rowr r r � test_roundtrip_quoteed_newlinesN s
z(Test_Csv.test_roundtrip_quoteed_newlinesc C s� t ddd��p}tj|tjdd�}ddgdd gg}|�|� |�d
� x2ttj|tjdd��D ]\}}| �||| � q\W W d Q R X d S ) Nzw+r )rD r1 )r r% za
brd re zx
dr ) r r r r5 rg rF r� r r" )r rJ r r� rR r� r r r �(test_roundtrip_escaped_unquoted_newlinesX s
z1Test_Csv.test_roundtrip_escaped_unquoted_newlinesN)&r> r? r@ �__doc__r r r r. r/ r0 r7 r8 r9 rA rB rC rK rM rW rZ r] r^ r` rh ri r Z cpython_onlyrk ro rq rr rt ru r| r r� r� r� r r r r r sB
r c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dS )�TestDialectRegistryc C s� | � ttjd � | � ttj� | � tjtjd � | � tjtjd� | � ttj� | � tjtjd � | � tjtjd� | � ttjd � | � ttjd d � | � ttjddd� | j ttjdd d� | j ttjdd d� | � ttjg � d S )NZnonesuchr )Z badargument)r )r r r �
list_dialects� get_dialectr �unregister_dialect�register_dialect)r r r r �test_registry_badargsb s z)TestDialectRegistry.test_registry_badargsc C sv G dd� dt j�}d}t �� |g }|�� t �||� | �t j|� | �t �|�j d� t
t �� �}| �||� d S )Nc @ s e Zd ZdZdS )z5TestDialectRegistry.test_registry.<locals>.myexceltsv� N)r> r? r@ r
r r r r �
myexceltsvt s r� r� ) r �excelr� �sortr� �
addCleanupr� r" r� r
�sorted)r r� �nameZexpected_dialectsZ got_dialectsr r r �
test_registrys s z!TestDialectRegistry.test_registryc C sZ d}t j|dd� | �t j|� | �t �|�jd� | �dddggtt �dg|��� d S )NZfedcba�;)r
rX �Y�ZzX;Y;Z) r r� r� r� r" r� r
rl r )r r� r r r �test_register_kwargs s
z(TestDialectRegistry.test_register_kwargsc C s$ G dd� dt j�}| �t j|� d S )Nc @ s e Zd ZdZdS )z?TestDialectRegistry.test_incomplete_dialect.<locals>.myexceltsvr� N)r> r? r@ r
r r r r r� � s r� )r �Dialectr r )r r� r r r �test_incomplete_dialect� s z+TestDialectRegistry.test_incomplete_dialectc C sv G dd� dt j�}td��R}|�d� |�d� t j||� d�}| �t|�ddg� | �t|�d d
g� W d Q R X d S ) Nc @ s e Zd ZdZejZdZdS )z5TestDialectRegistry.test_space_dialect.<locals>.space� r1 N)r> r? r@ r
r r5 r r% r r r r �space� s r� zw+zabc def
c1ccccc1 benzene
r )r# rp �defZc1ccccc1Zbenzene)r r� r rb rF r r" r� )r r� rJ r r r r �test_space_dialect� s
z&TestDialectRegistry.test_space_dialectc O sZ t dddd��B}tj|f|�|�}|�dddg� |�d� | �|�� |� W d Q R X d S ) Nzw+r zutf-8)rD �encodingr rP ry r )r r r rE rF r" rG )r �expectedZ writeargsZ kwwriteargsrJ r r r r �compare_dialect_123� s
z'TestDialectRegistry.compare_dialect_123c C s� G dd� dt j�}G dd� dt j�}G dd� dt j�}G dd� dt j�}t �d|� zV| �d � | �d
|� | jd |� d � | jd
dd � | jd|dd� | jd|d � W d t �d� X d S )Nc @ s e Zd ZdZdS )z5TestDialectRegistry.test_dialect_apply.<locals>.testAr� N)r> r? r@ r
r r r r �testA� s r� c @ s e Zd ZdZdS )z5TestDialectRegistry.test_dialect_apply.<locals>.testBr! N)r> r? r@ r
r r r r �testB� s r� c @ s e Zd ZdZdS )z5TestDialectRegistry.test_dialect_apply.<locals>.testC�|N)r> r? r@ r
r r r r �testC� s r� c @ s e Zd ZdZdS )z7TestDialectRegistry.test_dialect_apply.<locals>.testUniu ΛN)r> r? r@ r
r r r r �testUni� s r� z1,2,3
z1 2 3
z1:2:3
)r# z1|2|3
z1;2;3
r� )r# r
u 1Λ2Λ3
)r r� r� r� r� )r r� r� r� r� r r r �test_dialect_apply� s
z&TestDialectRegistry.test_dialect_applyc C sT | j ttjg dd� | j ttjg d d� | j ttjg dd� | j ttjg dd� d S )Nr )r )r
���)r �d )r r r r )r r r r �test_bad_dialect� s z$TestDialectRegistry.test_bad_dialectc C s0 x*t �� D ] }t �|�}| �ttj|� q
W d S )N)r r� r� r r �copy)r r� r# r r r � test_copy� s
z TestDialectRegistry.test_copyc C sJ xDt �� D ]8}t �|�}x(ttjd �D ]}| �ttj||� q(W q
W d S )Nr ) r r� r� �range�pickleZHIGHEST_PROTOCOLr r �dumps)r r� r# �protor r r � test_pickle� s
z TestDialectRegistry.test_pickleN)
r> r? r@ r� r� r� r� r� r� r� r� r� r� r r r r r� a s
r� c @ s e Zd Zdd� Zdd� ZdS )� TestCsvBasec C sT t ddd��>}|�|� |�d� tj|| jd�}t|�}| �||� W d Q R X d S )Nzw+r )rD r )r# )r rb rF r r r# rl r" )r rm �expected_resultrJ r rH r r r �readerAssertEqual� s
z TestCsvBase.readerAssertEqualc C sP t ddd��:}tj|| jd�}|�|� |�d� | �|�� |� W d Q R X d S )Nzw+r )rD )r# r )r r r r# rg rF r" rG )r rm r� rJ r r r r �writerAssertEqual� s
z TestCsvBase.writerAssertEqualN)r> r? r@ r� r� r r r r r� � s r� c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Z d4S )5�TestDialectExcelr� c C s | � ddgg� d S )Nrp )r� )r r r r � test_single� s z TestDialectExcel.test_singlec C s | � ddddddgg� d S )Nz 1,2,3,4,5rv �2rx �4rw )r� )r r r r � test_simple� s z TestDialectExcel.test_simplec C s | � dg � d S )Nr )r� )r r r r �test_blankline� s z TestDialectExcel.test_blanklinec C s | � dddgg� d S )Nr r )r� )r r r r �test_empty_fields� s z"TestDialectExcel.test_empty_fieldsc C s | � ddgg� d S )Nz""r )r� )r r r r �test_singlequoted� s z"TestDialectExcel.test_singlequotedc C s | � dddgg� d S )Nz"",r )r� )r r r r � test_singlequoted_left_empty� s z-TestDialectExcel.test_singlequoted_left_emptyc C s | � dddgg� d S )Nz,""r )r� )r r r r � test_singlequoted_right_empty� s z.TestDialectExcel.test_singlequoted_right_emptyc C s | � ddgg� d S )Nz""""r )r� )r r r r �test_single_quoted_quote� s z)TestDialectExcel.test_single_quoted_quotec C s | � ddgg� d S )Nz""""""z"")r� )r r r r �test_quoted_quotes� s z#TestDialectExcel.test_quoted_quotesc C s | � ddgg� d S )Nza""b)r� )r r r r �test_inline_quote� s z"TestDialectExcel.test_inline_quotec C s | � ddgg� d S )Nza"b"c)r� )r r r r �test_inline_quotes� s z#TestDialectExcel.test_inline_quotesc C s | � ddgg� d S )Nz"a"bZab)r� )r r r r �test_quotes_and_more s z%TestDialectExcel.test_quotes_and_morec C s | � ddgg� d S )Nza"b)r� )r r r r �test_lone_quote s z TestDialectExcel.test_lone_quotec C s | � ddgg� d S )Nz"a" "b"za "b")r� )r r r r �test_quote_and_quote s z%TestDialectExcel.test_quote_and_quotec C s | � ddgg� d S )Nz "a")r� )r r r r �test_space_and_quote
s z%TestDialectExcel.test_space_and_quotec C s | � dddddddgg� d S )Nz#1,2,3,"I think, therefore I am",5,6rv r� rx zI think, therefore I amrw �6)r� )r r r r � test_quoted s z TestDialectExcel.test_quotedc C s | � ddddddgg� d S )NzJ1,2,3,"""I see,"" said the blind man","as he picked up his hammer and saw"rv r� rx z"I see," said the blind manz"as he picked up his hammer and saw)r� )r r r r �test_quoted_quote s z"TestDialectExcel.test_quoted_quotec C s* d}| � |dddddgddd d
gg� d S ) NzR1,2,3,"""I see,""
said the blind man","as he picked up his
hammer and saw"
9,8,7,6rv r� rx z"I see,"
said the blind manz"as he picked up his
hammer and saw�9�8�7r� )r� )r rm r r r �test_quoted_nl s z TestDialectExcel.test_quoted_nlc C s | � dddddgg� d S )Nz 12,12,1",Z12z1"r )r� )r r r r �test_dubious_quote( s z#TestDialectExcel.test_dubious_quotec C s | � g d� d S )Nr )r� )r r r r � test_null+ s zTestDialectExcel.test_nullc C s | � dggd� d S )Nrp zabc
)r� )r r r r �test_single_writer. s z#TestDialectExcel.test_single_writerc C s | � dddddggd� d S )Nr rP rp ry � z
1,2,abc,3,4
)r� )r r r r �test_simple_writer1 s z#TestDialectExcel.test_simple_writerc C s | � dddddggd� d S )Nr rP za"bc"ry r� z1,2,"a""bc""",3,4
)r� )r r r r � test_quotes4 s z TestDialectExcel.test_quotesc C s | � dggd� d S )Nzabc,defz "abc,def"
)r� )r r r r �test_quote_fieldsep7 s z$TestDialectExcel.test_quote_fieldsepc C s | � dddddggd� d S )Nr rP za
bcry r� z1,2,"a
bc",3,4
)r� )r r r r �
test_newlines: s z TestDialectExcel.test_newlinesN) r> r? r@ r# r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� � s4 r� c @ s e Zd ZejZdZdS )� EscapedExcelr1 N)r> r? r@ r r5 r r% r r r r r� = s r� c @ s" e Zd Ze� Zdd� Zdd� ZdS )�TestEscapedExcelc C s | � dggd� d S )Nzabc,defz
abc\,def
)r� )r r r r �test_escape_fieldsepD s z%TestEscapedExcel.test_escape_fieldsepc C s | � ddgg� d S )Nz
abc\,def
zabc,def)r� )r r r r �test_read_escape_fieldsepG s z*TestEscapedExcel.test_read_escape_fieldsepN)r> r? r@ r� r# r� r� r r r r r� A s r� c @ s e Zd ZdZdd� Zdd� ZdS )�TestDialectUnixZunixc C s | � dddggd� d S )Nr zabc defrp z"1","abc def","abc"
)r� )r r r r r� M s z"TestDialectUnix.test_simple_writerc C s | � ddddgg� d S )Nz"1","abc def","abc"
rv zabc defrp )r� )r r r r �test_simple_readerP s z"TestDialectUnix.test_simple_readerN)r> r? r@ r# r� r� r r r r r� J s r� c @ s e Zd ZejZdZdS )�QuotedEscapedExcelr1 N)r> r? r@ r r\ r r% r r r r r� S s r� c @ s" e Zd Ze� Zdd� Zdd� ZdS )�TestQuotedEscapedExcelc C s | � dggd� d S )Nzabc,defz "abc,def"
)r� )r r r r �test_write_escape_fieldsepZ s z1TestQuotedEscapedExcel.test_write_escape_fieldsepc C s | � ddgg� d S )Nz "abc\,def"
zabc,def)r� )r r r r r� ] s z0TestQuotedEscapedExcel.test_read_escape_fieldsepN)r> r? r@ r� r# r� r� r r r r r� W s r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%S )&�TestDictFieldsc C s� t ddd��n}tj|dddgd�}|�� |�d� | �|�� d � |�d
d d �� |�d� |�� | �|�� d
� W d Q R X d S )Nzw+r )rD �f1�f2�f3)�
fieldnamesr z
f1,f2,f3
rN rp )r� r� z 10,,abc
) r r �
DictWriter� writeheaderrF r" �readlinerE rG )r rJ r r r r �test_write_simple_dictc s
z%TestDictFields.test_write_simple_dictc C sd t � }tj|dddgd�}|�� | �|�� d� |�dddd �d
d d d �g� | �|�� d
� d S )Nr� r� r� )r� z
f1,f2,f3
r rp �f)r� r� r� rP � Zxyzz f1,f2,f3
1,abc,f
2,5,xyz
)r r r� r� r" �getvaluerg )r rJ r r r r � test_write_multiple_dict_rowsn s
z,TestDictFields.test_write_multiple_dict_rowsc C s t � }| �ttj|� d S )N)r r r r r� )r rJ r r r �test_write_no_fieldsx s z#TestDictFields.test_write_no_fieldsc
C s� t ddd��|}tj|dddgd�}| �t��}|�dd d
d �� W d Q R X t|j�}| �d |� | �d
|� | � d|� | �d|� W d Q R X d S )Nzw+r )rD r� r� r� )r� rN Zspamrp )Zf4r� r r� z'f4'z'f2'rv )
r r r� r r{ rE �str� exception�assertInZ assertNotIn)r rJ r Zcxr� r r r �#test_write_fields_not_in_fieldnames| s
z2TestDictFields.test_write_fields_not_in_fieldnamesc C s$ t � }| jttj|ddgdd� d S )Nr� r� Zraised)� extrasaction)r r r{ r r� )r rJ r r r �&test_typo_in_extrasaction_raises_error� s z5TestDictFields.test_typo_in_extrasaction_raises_errorc C s@ t � }tj|ddgdd�}ddddd �}| �ttjj||� d S )
Nr� r� �raise)r� r r rP ry )�f0r� r� r� )r r r� r r{ rE )r rJ r �dictrowr r r �)test_write_field_not_in_field_names_raise� s z8TestDictFields.test_write_field_not_in_field_names_raisec C sJ t � }tj|ddgdd�}ddddd �}tj�||� | �|�� d
� d S ) Nr� r� �ignore)r� r r rP ry )r� r� r� r� z1,2
)r r r� rE r" r� )r rJ r r� r r r �*test_write_field_not_in_field_names_ignore� s
z9TestDictFields.test_write_field_not_in_field_names_ignorec C sX t d��F}|�d� |�d� tj|dddgd�}| �t|�dd d
d �� W d Q R X d S ) Nzw+z 1,2,abc
r r� r� r� )r� rv r� rp )r� r� r� )r rb rF r �
DictReaderr" r� )r rJ r r r r �test_read_dict_fields� s
z$TestDictFields.test_read_dict_fieldsc C sb t d��P}|�d� |�d� t�|�}| �t|�dddd�� | �|jdd d
g� W d Q R X d S ) Nzw+zf1,f2,f3
1,2,abc
r rv r� rp )r� r� r� r� r� r� )r rb rF r r� r" r� r� )r rJ r r r r � test_read_dict_no_fieldnames� s
z+TestDictFields.test_read_dict_no_fieldnamesc C sp t d��^}|�d� |�d� tj|tt�|��d�}| �|jdddg� | �t|�dd d
d �� W d Q R X d S ) Nzw+zf1,f2,f3
1,2,abc
r )r� r� r� r� rv r� rp )r� r� r� ) r rb rF r r� r� r r" r� )r rJ r r r r �#test_read_dict_fieldnames_from_file� s
z2TestDictFields.test_read_dict_fieldnames_from_filec C s� dd l }td��l}|�d� |�d� t�|�}t|�}x>|�|g|�D ],}| �|j dddg� | �|ddd d
�� qHW W d Q R X d S ) Nr zw+zf1,f2,f3
1,2,abc
r� r� r� rv r� rp )r� r� r� )
� itertoolsr rb rF r r� r� �chainr" r� )r r rJ r �firstr� r r r � test_read_dict_fieldnames_chain� s
z.TestDictFields.test_read_dict_fieldnames_chainc
C s^ t d��L}|�d� |�d� tj|ddgd�}| �t|�ddd d
d d gd
�� W d Q R X d S )Nzw+z1,2,abc,4,5,6
r r� r� )r� rv r� rp r� rw r� )r� r� N)r rb rF r r� r" r� )r rJ r r r r �test_read_long� s
z TestDictFields.test_read_longc
C s` t d��N}|�d� |�d� tj|ddgdd�}| �t|�dd d
d d d
gd�� W d Q R X d S )Nzw+z1,2,abc,4,5,6
r r� r� �_rest)r� �restkeyrv r� rp r� rw r� )r� r� r )r rb rF r r� r" r� )r rJ r r r r �test_read_long_with_rest� s
z'TestDictFields.test_read_long_with_restc
C sl t d��Z}|�d� |�d� tj|dd�}| �|jddg� | �t|�dd d
d d d
gd�� W d Q R X d S )Nzw+zf1,f2
1,2,abc,4,5,6
r r )r r� r� rv r� rp r� rw r� )r� r� r )r rb rF r r� r" r� r� )r rJ r r r r �&test_read_long_with_rest_no_fieldnames� s
z5TestDictFields.test_read_long_with_rest_no_fieldnamesc C s| t d��j}|�d� |�d� tj|d�� dd�}| �t|�ddd d
d d d
�� | �t|�ddd dddd
�� W d Q R X d S )Nzw+z1,2,abc,4,5,6
1,2,abc
r z 1 2 3 4 5 6ZDEFAULT)r� Zrestvalrv r� rp r� rw r� )rv r� rx r� rw r� )r rb rF r r� �splitr" r� )r rJ r r r r �test_read_short� s
z TestDictFields.test_read_shortc C s< dddg}t j|d�� d�}| �t|�dddd d
d �� d S ) Nz 2147483648,43.0e12,17,abc,def
z 147483648,43.0e2,17,abc,def
z47483648,43.0,170,abc,def
zi1 float i2 s1 s2)r� Z
2147483648z43.0e12Z17rp r� )Zi1�floatZi2�s1�s2)r r� r
r" r� )r Zsampler r r r �test_read_multi� s z TestDictFields.test_read_multic
C sX t jdddgd�� d�}| �t|�ddddd d
d �� | �t|�ddddd d
d �� d S ) Nz1,2,abc,4,5,6
z
z 1 2 3 4 5 6)r� rv r� rp r� rw r� )rv r� rx r� rw r� )r r� r
r" r� )r r r r r �test_read_with_blanks� s z$TestDictFields.test_read_with_blanksc
C s8 t jdgd�� dd�}| �t|�ddddd d
d �� d S ) Nz1;2;abc;4;5;6
z 1 2 3 4 5 6r� )r� r
rv r� rp r� rw r� )rv r� rx r� rw r� )r r� r
r" r� )r r r r r �test_read_semi_sep s
z!TestDictFields.test_read_semi_sepN)r> r? r@ r� r� r� r� r� r� r� r� r� r r r r r r r r r r r r r r� ` s$
r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�TestArrayWritesc C s� dd l }dd� td�D �}|� d|�}tddd��P}tj|d d
�}|�|� d �d d� |D ��d
}|�d� | �|� � |� W d Q R X d S )Nr c S s g | ] }d | �qS )� r )�.0rR r r r �
<listcomp>
s z2TestArrayWrites.test_int_write.<locals>.<listcomp>r rR zw+r )rD r� )r# r c S s g | ] }t |��qS r )r� )r rR r r r r s z
)
�arrayr� r r r rE �joinrF r" rG )r r �contentsr[ rJ r r� r r r �test_int_write s
z TestArrayWrites.test_int_writec C s� dd l }dd� td�D �}|� d|�}tddd��P}tj|d d
�}|�|� d �d d� |D ��d
}|�d� | �|� � |� W d Q R X d S )Nr c S s g | ]}d | d �qS )r g�������?r )r rR r r r r s z5TestArrayWrites.test_double_write.<locals>.<listcomp>r rf zw+r )rD r� )r# r c S s g | ] }t |��qS r )r� )r rR r r r r s z
)
r r� r r r rE r rF r" rG )r r r r[ rJ r r� r r r �test_double_write s
z!TestArrayWrites.test_double_writec C s� dd l }dd� td�D �}|� d|�}tddd��P}tj|d d
�}|�|� d �d d� |D ��d
}|�d� | �|� � |� W d Q R X d S )Nr c S s g | ]}d | d �qS )r g�������?r )r rR r r r r ! s z4TestArrayWrites.test_float_write.<locals>.<listcomp>r r� zw+r )rD r� )r# r c S s g | ] }t |��qS r )r� )r rR r r r r & s z
)
r r� r r r rE r rF r" rG )r r r r[ rJ r r� r r r �test_float_write s
z TestArrayWrites.test_float_writec C sz dd l }dd l}|� d|j�}tddd��F}tj|dd�}|�|� d�|�d }|�d� | � |�
� |� W d Q R X d S )
Nr �uzw+r )rD r� )r# r z
) r �stringZ
ascii_lettersr r r rE r rF r" rG )r r r r[ rJ r r� r r r �test_char_write* s
z TestArrayWrites.test_char_writeN)r> r? r@ r r r r r r r r r s r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�TestDialectValidityc C s� G dd� dt j�}|� }| �|jt j� d |_| �t j|� d|_t j|_d|_ |� }| �|jt j� | �|j d� | �
|j� d|_ | �t j�� }|� W d Q R X | �t |j �d� d|_ | �t j�� }|� W d Q R X | �t |j �d� d S ) Nc @ s& e Zd ZdZdZdZdZdZej Z
dS )z3TestDialectValidity.test_quoting.<locals>.mydialectr� r1 FTz
N) r> r? r@ r
r% r$ r' r
r r5 r r r r r � mydialect7 s r Tr z''z("quotechar" must be a 1-character stringr� z#"quotechar" must be string, not int)
r r� r" r r5 r r r$ r r �
assertTruer� r� )r r rf �cmr r r � test_quoting6 s, z TestDialectValidity.test_quotingc C s G dd� dt j�}|� }| �|jd� d|_| �t j�� }|� W d Q R X | �t|j�d� d|_| �t j�� }|� W d Q R X | �t|j�d� d|_| �t j�� }|� W d Q R X | �t|j�d� d |_| �t j�� }|� W d Q R X | �t|j�d
� d S ) Nc @ s& e Zd ZdZdZdZdZdZej Z
dS )z5TestDialectValidity.test_delimiter.<locals>.mydialectr� r1 FTz
N) r> r? r@ r
r% r$ r' r
r r5 r r r r r r Y s r r� z:::z("delimiter" must be a 1-character stringr � ,z%"delimiter" must be string, not bytesr� z#"delimiter" must be string, not int)r r� r" r
r r r� r� )r r rf r" r r r �test_delimiterX s. z"TestDialectValidity.test_delimiterc C sz G dd� dt j�}|� }| �|jd� d|_|� }| �|jd� d|_| �t j�� }|� W d Q R X | �t|j�d� d S )Nc @ s& e Zd ZdZdZdZdZdZej Z
dS )z:TestDialectValidity.test_lineterminator.<locals>.mydialectr� r1 FTz
N) r> r? r@ r
r% r$ r' r
r r5 r r r r r r | s r z
z:::r� z!"lineterminator" must be a string)r r� r" r
r r r� r� )r r rf r" r r r �test_lineterminator{ s z'TestDialectValidity.test_lineterminatorc
C sz dd� }xldD ]d}| j |d��N | �tj||d� | �tj||d� | �tj||d� | �tj||d� W d Q R X qW d S ) Nc S s( G dd� dt j�}t|| |� |� }d S )Nc @ s e Zd ZdS )zQTestDialectValidity.test_invalid_chars.<locals>.create_invalid.<locals>.mydialectN)r> r? r@ r r r r r � s r )r r� r+ )�
field_name�valuer rf r r r �create_invalid� s z>TestDialectValidity.test_invalid_chars.<locals>.create_invalid)r
r% r )r' r rp � xr� )�subTestr r r )r r) r' r r r �test_invalid_chars� s
z&TestDialectValidity.test_invalid_charsN)r> r? r@ r# r% r& r, r r r r r 5 s "#r c @ sh e Zd ZdZdZdZdZdZdZdZ dZ
d Z d
Z d Z
d d
� Zdd� Zdd� Zdd� Zdd� Zdd� ZdS )� TestSnifferz�Harry's, Arlington Heights, IL, 2/1/03, Kimi Hayes
Shark City, Glendale Heights, IL, 12/28/02, Prezence
Tommy's Place, Blue Island, IL, 12/28/02, Blue Sunday/White Crow
Stonecutters Seafood and Chop House, Lemont, IL, 12/19/02, Week Back
a 'Harry''s':'Arlington Heights':'IL':'2/1/03':'Kimi Hayes'
'Shark City':'Glendale Heights':'IL':'12/28/02':'Prezence'
'Tommy''s Place':'Blue Island':'IL':'12/28/02':'Blue Sunday/White Crow'
'Stonecutters ''Seafood'' and Chop House':'Lemont':'IL':'12/19/02':'Week Back'
z+"venue","city","state","date","performers"
z�05/05/03?05/05/03?05/05/03?05/05/03?05/05/03?05/05/03
05/05/03?05/05/03?05/05/03?05/05/03?05/05/03?05/05/03
05/05/03?05/05/03?05/05/03?05/05/03?05/05/03?05/05/03
zT2147483648;43.0e12;17;abc;def
147483648;43.0e2;17;abc;def
47483648;43.0;170;abc;def
zaaa bbb
AAA
BBB
za|b|c
d|e|f
z'a'|'b'|'c'
'd'|e|f
z+"venue"+"city"+"state"+"date"+"performers"
z�Harry's+ Arlington Heights+ IL+ 2/1/03+ Kimi Hayes
Shark City+ Glendale Heights+ IL+ 12/28/02+ Prezence
Tommy's Place+ Blue Island+ IL+ 12/28/02+ Blue Sunday/White Crow
Stonecutters Seafood and Chop House+ Lemont+ IL+ 12/19/02+ Week Back
a 'Harry''s'+ Arlington Heights'+ 'IL'+ '2/1/03'+ 'Kimi Hayes'
'Shark City'+ Glendale Heights'+' IL'+ '12/28/02'+ 'Prezence'
'Tommy''s Place'+ Blue Island'+ 'IL'+ '12/28/02'+ 'Blue Sunday/White Crow'
'Stonecutters ''Seafood'' and Chop House'+ 'Lemont'+ 'IL'+ '12/19/02'+ 'Week Back'
c C s: t �� }| �|�| j�d� | �|�| j| j �d� d S )NFT)r �Snifferr" �
has_header�sample1�header1)r �snifferr r r �test_has_header� s zTestSniffer.test_has_headerc C s: t �� }| �|�| j�d� | �|�| j| j �d� d S )NFT)r r. r" r/ �sample8�header2)r r2 r r r �'test_has_header_regex_special_delimiter� s z3TestSniffer.test_has_header_regex_special_delimiterc
C st t �� }xfdD ]^}| �|��J |�|d�}| �|jd� | �|jd� | �|jd� | �|j d� W d Q R X qW d S )N)z ;'123;4';z'123;4';z;'123;4'z'123;4'z,;r� �'F)
r r. r+ �sniffr" r
r ZassertIsr$ r' )r r2 �headerr# r r r � test_guess_quote_and_delimiter� s
z*TestSniffer.test_guess_quote_and_delimiterc C sx t �� }|�| j�}| �|jd� | �|jd� | �|jd� |�| j�}| �|jd� | �|jd� | �|jd� d S )Nr r Tr! r7 F) r r. r8 r0 r" r
r r' �sample2)r r2 r# r r r �
test_sniff� s zTestSniffer.test_sniffc C s t �� }|�| j�}| �|j| j� |j| jdd�}| �|jd� |j| jdd�}| �|jd� |�| j�}| �|jd� |�| j�}| �|jd� |�| j �}| �|jd� |�| j
�}| �|jd� | �|j d � |�| j �}| �|jd
� |�| j
�}| �|jd
� | �|j d � d S ) Nz?,)Z
delimiters�?z/,�/r� r� r� r7 �+)r r. r8 �sample3r� r
r" �sample4�sample5�sample6�sample7r r4 �sample9)r r2 r# r r r �test_delimiters� s* zTestSniffer.test_delimitersc C s� t �� }|�| j�}| �|j� |�| j�}| �|j� |�| j�}| �|j� |�| j �}| �|j� |�| j
�}| �|j� d S )N) r r. r8 r1 Z assertFalser$ r5 r; r! r4 rE )r r2 r# r r r �test_doublequote
s z TestSniffer.test_doublequoteN)r> r? r@ r0 r; r1 r@ rA rB rC rD r5 r4 rE r3 r6 r: r< rF rG r r r r r- � s"
r- c @ s e Zd Zdd� ZeZdS )�NULc G s d S )Nr )�sr, r r r rb s z NUL.writeN)r> r? r@ rb �
writelinesr r r r rH s rH �gettotalrefcountz requires sys.gettotalrefcount()c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� TestLeaksc C s� d}t �� }x^td�D ]R}t�� | �tjg � t �� }t�dg� t�dg� t�dg� || }|}qW | �|dk d� d S )Nr r za,b,c
ry T) �sysrK r� �gc�collectr" �garbager r )r �delta�lastrcrR �rcr r r �test_create_read" s zTestLeaks.test_create_readc C s� d}t �� }t� }xXtd�D ]L}t�� | �tjg � t �� }t� |� t� |� t� |� || }|}q W | �|dk d� d S )Nr r ry T)
rM rK rH r� rN rO r" rP r r )r rQ rR rI rR rS r r r �test_create_write1 s
zTestLeaks.test_create_writec C s~ d}dgd }t �� }xRtd�D ]F}t�� | �tjg � t �� }t�|�}x |D ]}qRW || }|}q W | �|dk d� d S )Nr za,b,c
r� r T) rM rK r� rN rO r" rP r r )r rQ r� rR rR rS Zrdrr� r r r � test_readA s
zTestLeaks.test_readc C s� d}dddggd }t � }t�� }x\td�D ]P}t�� | �tjg � t�� }t� |�}x|D ]}|�
|� q^W || }|}q,W | �|dk d� d S )Nr r rP ry r� r T) rH rM rK r� rN rO r" rP r r rE ) r rQ r� rI rR rR rS r r� r r r �
test_writeQ s
zTestLeaks.test_writeN)r> r? r@ rT rU rV rW r r r r rL s rL c @ s( e Zd ZddddgZdd� Zdd� Zd S )
� TestUnicodeu Martin von Löwisu Marc André LemburgzGuido van Rossumu François Pinardc C s\ t dddd��D}|�d�| j�d � |�d� t�|�}| �t|�| jg� W d Q R X d S )Nzw+r zutf-8)rD r� r z
r ) r rb r �namesrF r r r" rl )r rJ r r r r �test_unicode_readi s
z TestUnicode.test_unicode_readc C s^ t dddd��F}t�|�}|�| j� d�| j�d }|�d� | �|�� |� W d Q R X d S )Nzw+r zutf-8)rD r� r z
r ) r r r rE rY r rF r" rG )r rJ r r� r r r �test_unicode_writeq s
z TestUnicode.test_unicode_writeN)r> r? r@ rY rZ r[ r r r r rX b s rX c @ s e Zd Zdd� Zdd� ZdS )�KeyOrderingTestc
C s� t � }xrtd�D ]f}tdddd��N}t�||�}|�� |�d� t�|�}t|j �}| �
||� |� |� W d Q R X qW | �
t |�dd� d S ) NZabcdezw+r zutf-8)rD r� r �x z@Key ordering: some key permutations not collected (expected 120))
�setr r r r� r� rF r� �tupler� r" �add�len)r Z resultset�keysZ
fileobjectZdwZdrZktr r r �,test_ordering_for_the_dict_reader_and_writer{ s
z<KeyOrderingTest.test_ordering_for_the_dict_reader_and_writerc
C s� t d��� }| �tt�|��tddg�tddd ddgfg�tdd g�td
d g�g� | �ttj|d d
��tddg�tddd ddgfg�tdd g�td
d g�g� |d= | �ttj|ddgd��tddg�tddd ddgfg�tddg�tddg�g� d S )Nz� FirstName,LastName
Eric,Idle
Graham,Chapman,Over1,Over2
Under1
John,Cleese
)� FirstName�Eric)�LastName�Idle)rd �Graham)rf �ChapmanZOver1ZOver2)rd �Under1)rf N)rd �John)rf �CleeseZ OtherInfo)r r �fname�lname)r� )rm re )rn rg )rm rh )rn ri )rm rj )rn N)rm rk )rn rl )r �
splitlinesr" rl r r� r )r �datar r r �test_ordered_dict_reader� s*
z(KeyOrderingTest.test_ordered_dict_readerN)r> r? r@ rc rq r r r r r\ y s r\ c @ s e Zd Zdd� ZdS )� MiscTestCasec C s ddh}t j| td|d� d S )Nr� � __version__)r Z_csv)�extra)r Z check__all__r )r rt r r r � test__all__� s zMiscTestCase.test__all__N)r> r? r@ ru r r r r rr � s rr �__main__)*r� rM Zunittest�ior Ztempfiler r rN r� Ztestr r r �textwrapr � collectionsr ZTestCaser r� r� r� r� r� r� r� r� r� r� r r r- rH Z
skipUnless�hasattrrL rX r\ rr r> �mainr r r r �<module> sH Rka (.j{
A4