| Server IP : 118.27.122.248 / Your IP : 216.73.216.153 Web Server : Apache System : Linux web0264.sh.tyo1 4.18.0-553.79.1.lve.el7h.x86_64 #1 SMP Wed Oct 15 16:34:46 UTC 2025 x86_64 User : c9415830 ( 11735) PHP Version : 8.4.17 Disable Function : NONE MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : ON | Sudo : OFF | Pkexec : OFF Directory : /opt/alt/python36/lib64/python3.6/test/__pycache__/ |
Upload File : |
3
�w2_=� � @ 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!ej"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�ej)� 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� | j t|� | j t|d � | j t||dd� | j t||dd� | j t||dd� | j 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 �2/opt/alt/python36/lib64/python3.6/test/test_csv.py�_test_arg_valid s
zTest_Csv._test_arg_validc C s | j 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 | j 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� ||� }| j |jjd� | j |jjd� | j |jjd � | j |jjd� | j |jjd� | j |jjtj � | j |jj
d� | j |jj d� | j t
t|jd� | j t
t|jdd� | j t
t|jd� | j 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 | j tjg � d S )N)r. r r )r r r r �test_reader_attrsA s zTest_Csv.test_reader_attrsc C s | j 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�}|||�}| j|jjd� | j|jjd� | j|jjd� | j|jjd� | j|jj d� | j|jj
tj� | j|jj d� | j|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 | j 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 | j 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 }||� }| j |jjd� | j |jjd� | j |jjd� | j |jjd� | j |jjd� | j |jjtj � | j |jj
d� | j |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 | j 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 | j 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|�}|j|� |jd� | j|j� ||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|�}| j|�� |j|� W d Q R X |jd� | j|j� 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| | j tjd � | jf d� | jd gd� | j tjd gtjd� G dd� d�}| j t|� � G dd� d�}| j t|� g� d S )Nr 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 }| j ||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� | j 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� | j tdddg�d� | j tddd g�d� | j tg �d� | j td g�d� | jtjtd g�tjd� | j 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 j|� �}| jt|jdgg� tddd��P}t j|�}| jt|jd � |jddgdd gg� |jd
� | j|j � 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 Zbufr r r �write� s z1Test_Csv.test_writerows.<locals>.BrokenFile.writeN)r> r? r@ ra r r r r �
BrokenFile� s rb r[ zw+r )rD �b�c�dr z
a,b
c,d
)
r r r rQ � writerowsr r rF r" rG )r rb r rJ r r r �test_writerows� s
zTest_Csv.test_writerowsc C s� t ddd��@}tj|�}|jdd gd dgg� |jd� | j|j� d� W d Q R X t ddd��<}tj|�}|jd gdgg� |jd� | j|j� d� W d Q R X t ddd��<}tj|�}|jdgd gg� |jd� | j|j� d � W d Q R X d S )
Nzw+r )rD r[ re r za,
,d
z""
a
za
""
)r r r rf 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 }|jd�}tddd��8}tj|�}|j|gg� |jd� | j|j� d� W d Q R X d S )Nr r[ zw+r )rD za
) � _testcapiZunicode_legacy_stringr r r rf rF r" rG )r ri rd rJ r r r r � test_writerows_legacy_strings� s
z&Test_Csv.test_writerows_legacy_stringsc K s& t j|f|�}t|�}| j||� d S )N)r r �listr" )r �inputrI r6 r �resultr r r �
_read_test� s zTest_Csv._read_testc C s� | j g g � | j dgg g� | jtj| j dgd dd� | jtj| j dgd dd� | j dgdggdd� | jtj| j d gd � d S )
Nr z"ab"cr )r( zab c�abcr )r$ s ab c)rn r r r )r r r r �test_read_oddinputs� s z Test_Csv.test_read_oddinputsc C s� | j dgddgg� | j dgddgg� | j dgddgg� | j dgddgg� | jtj| j dgg � | jtj| j dgg � | jtj| j d gg � d S )
Nza,br[ rc za,b
za,b
za,b
za,b
c,dza,b
c,dza,b
c,d)rn r r r )r r r r �
test_read_eol s zTest_Csv.test_read_eolc C s� | j dgddgg� | j 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( )rn 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[ rc rd r1 )r% za,b\,czb,cza,"b\,c"za,"b,\c"z a,"b,c\""zb,c"za,"b,c"\zb,c\)rn )r r r r �test_read_escape s zTest_Csv.test_read_escapec C s� | j dgdddgg� | j dgdddddggd dd� | j dgdddddggtjdd � | j d
gd d dd
dggtjd� | j 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)rn r r5 r\ r �
ValueError)r r r r �test_read_quoting! s
zTest_Csv.test_read_quotingc C s� t j� }z�d}d| }d||f }| j|g||gg� t j|� | j|g||gg� | jt j� |� t j|d � | jt j| j|gg � | jtt jd � | jtt jdd � W d t j|� X d S )NiP� rX z%s,%sr )r Zfield_size_limitrn r" r r r )r �limit�sizerY Zbigliner r r �test_read_bigfield/ s
zTest_Csv.test_read_bigfieldc C s� t jdddg�}| j|jd� t|� | j|jd� t|� | j|jd� t|� | j|jd� | jtt|� | j|jd� d S )Nzline,1zline,2zline,3r r rP rx )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}tj|�}| jt|jd � ddgddgg}|j|� |jd� x*ttj|��D ]\}}| j ||| � q\W W d Q R X d S ) Nzw+r )rD za
brc rd zx
dr )
r r r r r rf 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}|j|� |jd
� x2ttj|tjdd��D ]\}}| j||| � q\W W d Q R X d S ) Nzw+r )rD r1 )r r% za
brc rd zx
dr ) r r r r5 rf 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` rg rh r Z cpython_onlyrj rn rp rq rs rt 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� | j ttjd � | j ttj� | j tjtjd � | j tjtjd� | j ttj� | j tjtjd � | j tjtjd� | j ttjd � | j ttjd d � | j ttjddd� | j ttjdd d� | j ttjdd d� | j 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 j� |g }|j� t j||� | jt j|� | jt j|�j d� t
t j� �}| j||� 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� | jt j|� | jt j|�jd� | jdddggtt jdg|��� d S )NZfedcba�;)r
rX �Y�ZzX;Y;Z) r r� r� r� r" r� r
rk r )r r� r r r �test_register_kwargs s
z(TestDialectRegistry.test_register_kwargsc C s$ G dd� dt j�}| jt 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}|jd� |jd� t j||� d�}| jt|�ddg� | jt|�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# ro �defZc1ccccc1Zbenzene)r r� r ra 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|�|�}|jdddg� |jd� | j|j� |� W d Q R X d S ) Nzw+r zutf-8)rD �encodingr rP rx 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 jd|� zV| jd � | jd
|� | jd |� d � | jd
dd � | jd|dd� | jd|d � W d t jd� 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 )r �d ���)r r r r )r r r r �test_bad_dialect� s z$TestDialectRegistry.test_bad_dialectc C s0 x*t j� D ] }t j|�}| jttj|� 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 j� D ]8}t j|�}x(ttjd �D ]}| jttj||� 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��>}|j|� |jd� tj|| jd�}t|�}| j||� W d Q R X d S )Nzw+r )rD r )r# )r ra rF r r r# rk r" )r rl �expected_resultrJ r rH r r r �readerAssertEqual� s
z TestCsvBase.readerAssertEqualc C sP t ddd��:}tj|| jd�}|j|� |jd� | j|j� |� W d Q R X d S )Nzw+r )rD )r# r )r r r r# rf rF r" rG )r rl 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 | j ddgg� d S )Nro )r� )r r r r � test_single� s z TestDialectExcel.test_singlec C s | j ddddddgg� d S )Nz 1,2,3,4,5ru �2rw �4rv )r� )r r r r � test_simple� s z TestDialectExcel.test_simplec C s | j dg � d S )Nr )r� )r r r r �test_blankline� s z TestDialectExcel.test_blanklinec C s | j dddgg� d S )Nr r )r� )r r r r �test_empty_fields� s z"TestDialectExcel.test_empty_fieldsc C s | j ddgg� d S )Nz""r )r� )r r r r �test_singlequoted� s z"TestDialectExcel.test_singlequotedc C s | j dddgg� d S )Nz"",r )r� )r r r r � test_singlequoted_left_empty� s z-TestDialectExcel.test_singlequoted_left_emptyc C s | j dddgg� d S )Nz,""r )r� )r r r r � test_singlequoted_right_empty� s z.TestDialectExcel.test_singlequoted_right_emptyc C s | j ddgg� d S )Nz""""r )r� )r r r r �test_single_quoted_quote� s z)TestDialectExcel.test_single_quoted_quotec C s | j ddgg� d S )Nz""""""z"")r� )r r r r �test_quoted_quotes� s z#TestDialectExcel.test_quoted_quotesc C s | j ddgg� d S )Nza""b)r� )r r r r �test_inline_quote� s z"TestDialectExcel.test_inline_quotec C s | j ddgg� d S )Nza"b"c)r� )r r r r �test_inline_quotes� s z#TestDialectExcel.test_inline_quotesc C s | j ddgg� d S )Nz"a"bZab)r� )r r r r �test_quotes_and_more s z%TestDialectExcel.test_quotes_and_morec C s | j ddgg� d S )Nza"b)r� )r r r r �test_lone_quote s z TestDialectExcel.test_lone_quotec C s | j 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 | j ddgg� d S )Nz "a")r� )r r r r �test_space_and_quote
s z%TestDialectExcel.test_space_and_quotec C s | j dddddddgg� d S )Nz#1,2,3,"I think, therefore I am",5,6ru r� rw zI think, therefore I amrv �6)r� )r r r r � test_quoted s z TestDialectExcel.test_quotedc C s | j ddddddgg� d S )NzJ1,2,3,"""I see,"" said the blind man","as he picked up his hammer and saw"ru r� rw 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}| j |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,6ru r� rw z"I see,"
said the blind manz"as he picked up his
hammer and saw�9�8�7r� )r� )r rl r r r �test_quoted_nl s z TestDialectExcel.test_quoted_nlc C s | j 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 | j g d� d S )Nr )r� )r r r r � test_null+ s zTestDialectExcel.test_nullc C s | j dggd� d S )Nro zabc
)r� )r r r r �test_single_writer. s z#TestDialectExcel.test_single_writerc C s | j dddddggd� d S )Nr rP ro rx � z
1,2,abc,3,4
)r� )r r r r �test_simple_writer1 s z#TestDialectExcel.test_simple_writerc C s | j dddddggd� d S )Nr rP za"bc"rx r� z1,2,"a""bc""",3,4
)r� )r r r r � test_quotes4 s z TestDialectExcel.test_quotesc C s | j dggd� d S )Nzabc,defz "abc,def"
)r� )r r r r �test_quote_fieldsep7 s z$TestDialectExcel.test_quote_fieldsepc C s | j dddddggd� d S )Nr rP za
bcrx 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 | j dggd� d S )Nzabc,defz
abc\,def
)r� )r r r r �test_escape_fieldsepD s z%TestEscapedExcel.test_escape_fieldsepc C s | j 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 | j dddggd� d S )Nr zabc defro z"1","abc def","abc"
)r� )r r r r r� M s z"TestDialectUnix.test_simple_writerc C s | j ddddgg� d S )Nz"1","abc def","abc"
ru zabc defro )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 | j dggd� d S )Nzabc,defz "abc,def"
)r� )r r r r �test_write_escape_fieldsepZ s z1TestQuotedEscapedExcel.test_write_escape_fieldsepc C s | j 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�}|j� |jd� | j|j� d � |jd
d d �� |jd� |j� | j|j� d
� W d Q R X d S )Nzw+r )rD �f1�f2�f3)�
fieldnamesr z
f1,f2,f3
rN ro )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�}|j� | j|j� d� |jdddd �d
d d d �g� | j|j� d
� d S )Nr� r� r� )r� z
f1,f2,f3
r ro �f)r� r� r� rP � Zxyzz f1,f2,f3
1,abc,f
2,5,xyz
)r r r� r� r" �getvaluerf )r rJ r r r r � test_write_multiple_dict_rowsn s
z,TestDictFields.test_write_multiple_dict_rowsc C s t � }| jttj|� 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�}| jt��}|jdd d
d �� W d Q R X t|j�}| jd |� | jd
|� | j d|� | jd|� W d Q R X d S )Nzw+r )rD r� r� r� )r� rN Zspamro )Zf4r� r r� z'f4'z'f2'ru )
r r r� r rz 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 rz 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 �}| jttjj||� d S )
Nr� r� �raise)r� r r rP rx )�f0r� r� r� )r r r� r rz 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 �}tjj||� | j|j� d
� d S ) Nr� r� �ignore)r� r r rP rx )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}|jd� |jd� tj|dddgd�}| jt|�dd d
d �� W d Q R X d S ) Nzw+z 1,2,abc
r r� r� r� )r� ru r� ro )r� r� r� )r ra 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}|jd� |jd� tj|�}| jt|�dddd�� | j|jdd d
g� W d Q R X d S ) Nzw+zf1,f2,f3
1,2,abc
r ru r� ro )r� r� r� r� r� r� )r ra 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��^}|jd� |jd� tj|ttj|��d�}| j|jdddg� | jt|�dd d
d �� W d Q R X d S ) Nzw+zf1,f2,f3
1,2,abc
r )r� r� r� r� ru r� ro )r� r� r� ) r ra 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}|jd� |jd� tj|�}t|�}x>|j|g|�D ],}| j|j dddg� | j|ddd d
�� qHW W d Q R X d S ) Nr zw+zf1,f2,f3
1,2,abc
r� r� r� ru r� ro )r� r� r� )
� itertoolsr ra 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}|jd� |jd� tj|ddgd�}| jt|�ddd d
d d gd
�� W d Q R X d S )Nzw+z1,2,abc,4,5,6
r r� r� )r� ru r� ro r� rv r� )r� r� N)r ra 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}|jd� |jd� tj|ddgdd�}| jt|�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� �restkeyru r� ro r� rv r� )r� r� r )r ra 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}|jd� |jd� tj|dd�}| j|jddg� | jt|�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� ru r� ro r� rv r� )r� r� r )r ra 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}|jd� |jd� tj|dj� dd�}| jt|�ddd d
d d d
�� | jt|�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� Zrestvalru r� ro r� rv r� )ru r� rw r� rv r� )r ra 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|dj� d�}| jt|�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.0e12Z17ro 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 jdddgdj� d�}| jt|�ddddd d
d �� | jt|�ddddd d
d �� d S ) Nz1,2,abc,4,5,6
z
z 1 2 3 4 5 6)r� ru r� ro r� rv r� )ru r� rw r� rv 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 jdgdj� dd�}| jt|�ddddd d
d �� d S ) Nz1;2;abc;4;5;6
z 1 2 3 4 5 6r� )r� r
ru r� ro r� rv r� )ru r� rw r� rv 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 �}|j d|�}tddd��P}tj|d d
�}|j|� d jd d� |D ��d
}|jd� | j|j � |� 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 �}|j d|�}tddd��P}tj|d d
�}|j|� d jd d� |D ��d
}|jd� | j|j � |� 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 re 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 �}|j d|�}tddd��P}tj|d d
�}|j|� d jd d� |D ��d
}|jd� | j|j � |� 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}|j d|j�}tddd��F}tj|dd�}|j|� dj|�d }|jd� | j |j
� |� 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�}|� }| j|jt j� d |_| jt j|� d|_t j|_d|_ |� }| j|jt j� | j|j d� | j
|j� d|_ | jt j�� }|� W d Q R X | jt |j �d� d|_ | jt j�� }|� W d Q R X | jt |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 re �cmr r r � test_quoting6 s, z TestDialectValidity.test_quotingc C s G dd� dt j�}|� }| j|jd� d|_| jt j�� }|� W d Q R X | jt|j�d� d|_| jt j�� }|� W d Q R X | jt|j�d� d|_| jt j�� }|� W d Q R X | jt|j�d� d |_| jt j�� }|� W d Q R X | jt|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 re r! r r r �test_delimiterX s. z"TestDialectValidity.test_delimiterc
C sz G dd� dt j�}|� }| j|jd� d|_|� }| j|jd� d|_| jt j�� }|� W d Q R X | jt|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 re r! r r r �test_lineterminator{ s z'TestDialectValidity.test_lineterminatorc
C sz dd� }xld D ]d}| j |d��N | jtj||d� | jtj||d� | jtj||d � | jtj||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 re r r r �create_invalid� s z>TestDialectValidity.test_invalid_chars.<locals>.create_invalidr
r% r )r& r ro � xr� )r
r% r )�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� }| j|j| j�d� | j|j| 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� }| j|j| j�d� | j|j| j| j �d� d S )NFT)r r- r" r. �sample8�header2)r r1 r r r �'test_has_header_regex_special_delimiter� s z3TestSniffer.test_has_header_regex_special_delimiterc C st t j� }xfd D ]^}| j|��J |j|d�}| j|jd� | j|jd� | j|jd� | j|j d� W d Q R X qW d S )
N� ;'123;4';�'123;4';�;'123;4'�'123;4'z,;r� �'F)r6 r7 r8 r9 )
r r- r* �sniffr" r
r ZassertIsr$ r' )r r1 �headerr# r r r � test_guess_quote_and_delimiter� s
z*TestSniffer.test_guess_quote_and_delimiterc C sx t j� }|j| j�}| j|jd� | j|jd� | j|jd� |j| j�}| j|jd� | j|jd� | j|jd� d S )Nr r Tr! r: F) r r- r; r/ r" r
r r' �sample2)r r1 r# r r r �
test_sniff� s zTestSniffer.test_sniffc C s t j� }|j| j�}| j|j| j� |j| jdd�}| j|jd� |j| jdd�}| j|jd� |j| j�}| j|jd� |j| j�}| j|jd� |j| j �}| j|jd� |j| j
�}| j|jd� | j|j d � |j| j �}| j|jd
� |j| j
�}| j|jd
� | j|j d � d S ) Nz?,)Z
delimiters�?z/,�/r� r� r� r: �+)r r- r; �sample3r� r
r" �sample4�sample5�sample6�sample7r r3 �sample9)r r1 r# r r r �test_delimiters� s* zTestSniffer.test_delimitersc C s� t j� }|j| j�}| j|j� |j| j�}| j|j� |j| j�}| j|j� |j| j �}| j|j� |j| j
�}| j|j� d S )N) r r- r; r0 Z assertFalser$ r4 r> r r3 rH )r r1 r# r r r �test_doublequote
s z TestSniffer.test_doublequoteN)r> r? r@ r/ r> r0 rC rD rE rF rG r4 r3 rH r2 r5 r= r? rI rJ r r r r r, � s"
r, c @ s e Zd Zdd� ZeZdS )�NULc G s d S )Nr )�sr, r r r ra s z NUL.writeN)r> r? r@ ra �
writelinesr r r r rK s rK �gettotalrefcountz requires sys.gettotalrefcount()c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� TestLeaksc C s� d}t j� }x^td�D ]R}tj� | jtjg � t j� }tjdg� tjdg� tjdg� || }|}qW | j|dk d� d S )Nr r za,b,c
rx T) �sysrN r� �gc�collectr" �garbager r )r �delta�lastrcrR �rcr r r �test_create_read" s zTestLeaks.test_create_readc C s� d}t j� }t� }xXtd�D ]L}tj� | jtjg � t j� }tj |� tj |� tj |� || }|}q W | j|dk d� d S )Nr r rx T)
rP rN rK r� rQ rR r" rS r r )r rT rU rL rR rV r r r �test_create_write1 s
zTestLeaks.test_create_writec C s~ d}dgd }t j� }xRtd�D ]F}tj� | jtjg � t j� }tj|�}x |D ]}qRW || }|}q W | j|dk d� d S )Nr za,b,c
r� r T) rP rN r� rQ rR r" rS r r )r rT r� rU rR rV Zrdrr� r r r � test_readA s
zTestLeaks.test_readc C s� d}dddggd }t � }tj� }x\td�D ]P}tj� | jtjg � tj� }tj |�}x|D ]}|j
|� q^W || }|}q,W | j|dk d� d S )Nr r rP rx r� r T) rK rP rN r� rQ rR r" rS r r rE ) r rT r� rL rU rR rV r r� r r r �
test_writeQ s
zTestLeaks.test_writeN)r> r? r@ rW rX rY rZ r r r r rO s rO 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}|jdj| j�d � |jd� tj|�}| jt|�| jg� W d Q R X d S )Nzw+r zutf-8)rD r� r z
r ) r ra r �namesrF r r r" rk )r rJ r r r r �test_unicode_readi s
z TestUnicode.test_unicode_readc
C s^ t dddd��F}tj|�}|j| j� dj| j�d }|jd� | j|j� |� W d Q R X d S )Nzw+r zutf-8)rD r� r z
r ) r r r rE r\ r rF r" rG )r rJ r r� r r r �test_unicode_writeq s
z TestUnicode.test_unicode_writeN)r> r? r@ r\ r] r^ r r r r r[ b s r[ c @ s e Zd Zdd� Zdd� ZdS )�KeyOrderingTestc C s� t � }xrtd�D ]f}tdddd��N}tj||�}|j� |jd� tj|�}t|j �}| j
||� |j |� W d Q R X qW | j
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�j� }| jttj|��tddg�tddd dd gfg�tddg�tddg�g� | jttj|d
d��tdd g�td d d
dd gfg�td d g�td!d"g�g� |d= | jttj|ddgd��td#d$g�td%d&d dd gfg�td'd(g�td)d*g�g� d S )+Nz� FirstName,LastName
Eric,Idle
Graham,Chapman,Over1,Over2
Under1
John,Cleese
� FirstName�Eric�LastName�Idle�Graham�ChapmanZOver1ZOver2�Under1�John�CleeseZ OtherInfo)r r �fname�lname)r� )rg rh )ri rj )rg rk )ri rl )rg rm )ri N)rg rn )ri ro )rg rh )ri rj )rg rk )ri rl )rg rm )ri N)rg rn )ri ro )rp rh )rq rj )rp rk )rq rl )rp rm )rq N)rp rn )rq ro )r �
splitlinesr" rk r r� r )r �datar r r �test_ordered_dict_reader� s*
z(KeyOrderingTest.test_ordered_dict_readerN)r> r? r@ rf rt 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 �_csv)�extra)r rw )r Z check__all__r )r rx r r r � test__all__� s zMiscTestCase.test__all__N)r> r? r@ ry r r r r ru � s ru �__main__)*r� rP Zunittest�ior Ztempfiler r rQ r� Ztestr r r �textwrapr � collectionsr ZTestCaser r� r� r� r� r� r� r� r� r� r� r r r, rK Z
skipUnless�hasattrrO r[ r_ ru r> �mainr r r r �<module> sH Rka (.j{
A4