|
Server : LiteSpeed System : Linux server51.dnsbootclub.com 4.18.0-553.62.1.lve.el8.x86_64 #1 SMP Mon Jul 21 17:50:35 UTC 2025 x86_64 User : nandedex ( 1060) PHP Version : 8.1.33 Disable Function : NONE Directory : /opt/cppython/lib/python3.8/test/__pycache__/ |
U
>��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 �ZG d
d � d ej�ZG d d
� d
ej�ZG dd� dej�ZG dd� de�ZG dd� dej�ZG dd� de�ZG dd� de�ZG dd� dej�Z G dd� de�Z G d d � d ej�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)d0k�r�e�*� dS )1� N)�StringIO)�
TemporaryFile)�support)� permutations)�dedent)� OrderedDictc @ s e Zd Zdd� ZdS )� BadIterablec C s t �d S �N��OSError��self� r �,/opt/cppython/lib/python3.8/test/test_csv.py�__iter__ s zBadIterable.__iter__N)�__name__�
__module__� __qualname__r r r r r r 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%d&� Zd'd(� Zd)d*� Zd+d,� Zejd-d.� �Zd/d0� Z d1d2� Z d3d4� Z d5d6� Z d7d8� Z d9d:� Z!d;d<� Z"d=d>� Z#d?d@� Z$dAdB� Z%dCS )D�Test_Csvc 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 �Zbad_attr�� delimiterZXXZfoo� )� quotechar)�lineterminator��quoting� )r r )� assertRaises� TypeError�csv�Error� QUOTE_ALL)r
�ctor�argr r r �_test_arg_valid s(
�
�zTest_Csv._test_arg_validc C s$ | � tjg � | �ttjt� � d S r )r% r �readerr r r r r r r �test_reader_arg_valid/ s z Test_Csv.test_reader_arg_validc C s4 | � tjt� � G dd� d�}| �ttj|� � d S )Nc @ s e Zd Zedd� �ZdS )z1Test_Csv.test_writer_arg_valid.<locals>.BadWriterc S s t �d S r r
r r r r �write6 s z7Test_Csv.test_writer_arg_valid.<locals>.BadWriter.writeN)r r r �propertyr( r r r r � BadWriter5 s r* )r% r �writerr r r )r
r* r r r �test_writer_arg_valid3 s z Test_Csv.test_writer_arg_validc G s� ||� }| � |jjd� | �|jjd� | � |jjd � | � |jjd� | � |jjd� | � |jjt j
� | �|jj d� | �|jj d� | �
tt|jd� | �
tt|jdd� | �
tt|jd� | �
tt|jdd � d S ) N�,T�
�"Fr �:r )� assertEqual�dialectr �assertIs� 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 r )r> r r& r r r r �test_reader_attrsM s zTest_Csv.test_reader_attrsc C s | � tjt� � d S r )r> r r+ r r r r r �test_writer_attrsP s zTest_Csv.test_writer_attrsc
G s� t dddddtjddd�}|||�}| �|jjd� | �|jjd� | �|jjd� | �|jj d� | �|jj
d� | �|jj tj� | �|jj d� | �|jj
d� d S )Nr0 F�\�
�*T)r r4 r5 r r r r7 r8 )�dictr �
QUOTE_NONEr1 r2 r r3 r4 r5 r r r r7 r8 )r
r# r< �kwargsr= r r r �_test_kw_attrsS s �
zTest_Csv._test_kw_attrsc C s | � tjg � d S r )rG r r& r r r r �test_reader_kw_attrsc s z Test_Csv.test_reader_kw_attrsc C s | � tjt� � d S r )rG r r+ r r r r r �test_writer_kw_attrsf s z Test_Csv.test_writer_kw_attrsc G s� G dd� d�}||f }||� }| � |jjd� | �|jjd� | � |jjd� | � |jjd� | � |jjd� | � |jjt j
� | �|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)
r r r r r4 r5 r r r r" r r7 r8 r r r r r2 k s r2 rJ FrK rL rM T)
r1 r2 r r3 r4 r5 r r r r r" r7 r8 )r
r# r< r2 r= r r r �_test_dialect_attrsi s
z Test_Csv._test_dialect_attrsc C s | � tjg � d S r )rN r r& r r r r �test_reader_dialect_attrs s z"Test_Csv.test_reader_dialect_attrsc C s | � tjt� � d S r )rN r r+ r r r r r �test_writer_dialect_attrs� s z"Test_Csv.test_writer_dialect_attrsc K sV t ddd��@}tj|f|�}|�|� |�d� | �|�� ||jj � W 5 Q R X d S �N�w+r ��newliner ) r r r+ �writerow�seekr1 �readr2 r )r
�fields�expectrF �fileobjr+ r r r � _write_test� s
�zTest_Csv._write_testc
K sd t ddd��N}tj|f|�}| �|�� |�|� W 5 Q R X |�d� | �|�� d� W 5 Q R X d S rQ )r r r+ r rU rV r1 rW )r
�excrX rF rZ r+ r r r �_write_error_test� s
zTest_Csv._write_error_testc C s� | � tjd � | �dd� | �d gd� | j tjd gtjd� | � tt� � G dd� d�}| � t|� � G dd� d�}| � t|� g� d S ) Nr r �""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� r
)r
�ir r r � __getitem__� s z:Test_Csv.test_write_arg_valid.<locals>.BadList.__getitem__N)r r r r` rc r r r r �BadList� s rd c @ s e Zd Zdd� ZdS )z.Test_Csv.test_write_arg_valid.<locals>.BadItemc S s t �d S r r
r r r r �__str__� s z6Test_Csv.test_write_arg_valid.<locals>.BadItem.__str__N)r r r re r r r r �BadItem� s rf )r] r r! r[ rE r r )r
rd rf r r r �test_write_arg_valid� s z Test_Csv.test_write_arg_validc C s$ dd }| � ||gd||f � d S )N�X�P� �%s,%s)r[ )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 �p,q� a,1,"p,q"r z "a",1,"p,q"z
"a","1","p,q"�a
bz "a
b","1")r[ r] r r! rE r6 �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 )Nrm r rn ro rA �r5 zp,"q"F)r5 r4 z
a,1,"p,\"q\""r/ �"""")r5 r z\")r5 r r4 za,1,p\,q)r[ r] r r! r6 rE r r r r �test_write_escape� s8 � � �
�
�
� �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 )
Nrm r rn ro za,1,r r^ r r- )r[ �iterr] r r! rE 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 5 Q R X d S ) Nc @ s e Zd Zdd� ZdS )z+Test_Csv.test_writerows.<locals>.BrokenFilec S s t �d S r r
)r
�bufr r r r( � s z1Test_Csv.test_writerows.<locals>.BrokenFile.writeN)r r r r( r r r r �
BrokenFile� s ry rm rR r rS �b�c�dr z
a,b
c,d
)
r r+ r r � writerowsr r rV r1 rW )r
ry r+ rZ r r r �test_writerows� s
zTest_Csv.test_writerowsc C s� t ddd��@}t�|�}|�dd gd dgg� |�d� | �|�� d� W 5 Q R X t ddd��<}t�|�}|�d gdgg� |�d� | �|�� d� W 5 Q R X t ddd��<}t�|�}|�dgd gg� |�d� | �|�� d � W 5 Q R X d S )
NrR r rS rm r| r za,
,d
z""
a
za
""
)r r r+ r} rV r1 rW �r
rZ r+ r r r �test_writerows_with_none� s
z!Test_Csv.test_writerows_with_nonec C sH t ddd��2}t�|�}| �t|jd � | �t|jt� � W 5 Q R X d S )NrR r rS )r r r+ r r r} r r r r r r �test_writerows_errors� s
z Test_Csv.test_writerows_errorsc C s` dd l }|�d�}tddd��8}t�|�}|�|gg� |�d� | �|�� d� W 5 Q R X d S )Nr rm rR r rS za
) � _testcapiZunicode_legacy_stringr r r+ r} rV r1 rW )r
r� r{ rZ r+ r r r � test_writerows_legacy_strings� s
z&Test_Csv.test_writerows_legacy_stringsc K s& t j|f|�}t|�}| �||� d S r )r r& �listr1 )r
�inputrY rF 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 �r8 zab c�abcr )r4 s ab c�r� 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,brm rz za,b
za,b
za,b
za,b
c,dza,b
c,dza,b
c,dr� 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,"rm r z"arK �
rs Tr� )r5 r8 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,crm rz r{ rA rs za,b\,czb,cza,"b\,c"za,"b,\c"z a,"b,c\""zb,c"za,"b,c"\zb,c\)r� 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/ �3rA )r r5 �r r5 z
,3,"5",7.3, 9r � g333333 @� r z"a
b", 7rp z 7zabc,3)r� r rE rq r �
ValueErrorr r r r �test_read_quoting4 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 5 t �|� X d S )Nri rh rj r )r Zfield_size_limitr� r1 r r! r )r
�limit�sizerk Zbigliner r r �test_read_bigfieldB 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 ra r� )r r& r1 Zline_num�nextr �
StopIteration)r
�rr r r �test_read_linenumU s zTest_Csv.test_read_linenumc C sr t ddd��\}t�|�}ddgddgg}|�|� |�d� tt�|��D ]\}}| �||| � qJW 5 Q R X d S ) NrR r rS rp rz r{ �x
dr )r r r+ r} rV � enumerater& r1 �r
rZ r+ �rowsrb �rowr r r � test_roundtrip_quoteed_newlinesa s
z(Test_Csv.test_roundtrip_quoteed_newlinesc C s� t ddd��l}tj|tjdd�}ddgdd gg}|�|� |�d
� ttj|tjdd��D ]\}}| �||| � qZW 5 Q R X d S ) NrR r rS rA r� rp rz r{ r� r ) r r r+ rE r} rV r� r& r1 r� r r r �(test_roundtrip_escaped_unquoted_newlinesj s
z1Test_Csv.test_roundtrip_escaped_unquoted_newlinesN)&r r r r% r' r, r> r? r@ rG rH rI rN rO rP r[ r] rg rl rr ru rw r~ r� r� r Z cpython_onlyr� r� r� r� r� r� r� r� r� r� r� r r r r r sD
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 badargumentr )r r r �
list_dialects� get_dialectr! �unregister_dialect�register_dialectr r r r �test_registry_badargst 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 �
myexceltsv� s r� r� ) r �excelr� �sortr� �
addCleanupr� r1 r� r �sorted)r
r� �nameZexpected_dialectsZ got_dialectsr r r �
test_registry� s z!TestDialectRegistry.test_registryc C sZ d}t j|dd� | �t j|� | �t �|�jd� | �dddggtt �dg|��� d S )NZfedcba�;r rh �Y�ZzX;Y;Z) r r� r� r� r1 r� r r� 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� Nr� 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 5 Q R X d S ) Nc @ s e Zd ZdZejZdZdS )z5TestDialectRegistry.test_space_dialect.<locals>.space� rA N)r r r r r rE r r5 r r r r �space� s r� rR zabc def
c1ccccc1 benzene
r �r2 r� �defZc1ccccc1Zbenzene)r r� r r( rV r& r1 r� )r
r� rZ r& r r r �test_space_dialect� s
z&TestDialectRegistry.test_space_dialectc O sZ t dddd��B}tj|f|�|�}|�dddg� |�d� | �|�� |� W 5 Q R X d S ) NrR r �utf-8�rT �encodingr ra r� r )r r r+ rU rV r1 rW )r
�expectedZ writeargsZ kwwriteargsrZ 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 5 t �d� X d S )Nc @ s e Zd ZdZdS )z5TestDialectRegistry.test_dialect_apply.<locals>.testAr� Nr� r r r r �testA� s r� c @ s e Zd ZdZdS )z5TestDialectRegistry.test_dialect_apply.<locals>.testBr0 Nr� r r r r �testB� s r� c @ s e Zd ZdZdS )z5TestDialectRegistry.test_dialect_apply.<locals>.testC�|Nr� r r r r �testC� s r� c @ s e Zd ZdZdS )z7TestDialectRegistry.test_dialect_apply.<locals>.testUniu ΛNr� r r r r �testUni� s r� z1,2,3
z1 2 3
z1:2:3
r� z1|2|3
z1;2;3
r� )r2 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 s, t �� D ] }t �|�}| �ttj|� qd S r )r r� r� r r �copy)r
r� r2 r r r � test_copy� s
z TestDialectRegistry.test_copyc C sB t �� D ]4}t �|�}ttjd �D ]}| �ttj||� q$qd S )Nr ) r r� r� �range�pickle�HIGHEST_PROTOCOLr r �dumps)r
r� r2 �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� s s
r� c @ s e Zd Zdd� Zdd� ZdS )� TestCsvBasec C sT t ddd��>}|�|� |�d� tj|| jd�}t|�}| �||� W 5 Q R X d S )NrR r rS r r� )r r( rV r r& r2 r� r1 )r
r� �expected_resultrZ r& rX r r r �readerAssertEqual� s
z TestCsvBase.readerAssertEqualc C sP t ddd��:}tj|| jd�}|�|� |�d� | �|�� |� W 5 Q R X d S )NrR r rS r� r )r r r+ r2 r} rV r1 rW )r
r� r� rZ 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 )Nr� �r� r r r r � test_single� s z TestDialectExcel.test_singlec C s | � ddddddgg� d S )Nz 1,2,3,4,5r� �2r� �4r� 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 )Nr^ 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 )Nrt r/ r� r r r r �test_single_quoted_quote s z)TestDialectExcel.test_single_quoted_quotec C s | � ddgg� d S )Nz""""""r^ r� r r r r �test_quoted_quotes s z#TestDialectExcel.test_quoted_quotesc C s | � ddgg� d S )Nza""br� r r r r �test_inline_quote s z"TestDialectExcel.test_inline_quotec C s | � ddgg� d S )Nza"b"cr� r r r r �test_inline_quotes s z#TestDialectExcel.test_inline_quotesc C s | � ddgg� d S )Nz"a"b�abr� r r r r �test_quotes_and_more s z%TestDialectExcel.test_quotes_and_morec C s | � ddgg� d S )Nza"br� 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,6r� r� r� zI think, therefore I amr� �6r� 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"r� r� r� z"I see," said the blind manz"as he picked up his hammer and sawr� 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,6r� r� r� z"I see,"
said the blind manz"as he picked up his
hammer and saw�9�8�7r� r� )r
r� 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 r� �r� r r r r � test_null= s zTestDialectExcel.test_nullc C s | � dggd� d S )Nr� zabc
r� r r r r �test_single_writer@ s z#TestDialectExcel.test_single_writerc C s | � dddddggd� d S )Nr ra r� r� � z
1,2,abc,3,4
r� r r r r �test_simple_writerC s z#TestDialectExcel.test_simple_writerc C s | � dddddggd� d S )Nr ra za"bc"r� r� z1,2,"a""bc""",3,4
r� r r r r � test_quotesF s z TestDialectExcel.test_quotesc C s | � dggd� d S �N�abc,defz "abc,def"
r� r r r r �test_quote_fieldsepI s z$TestDialectExcel.test_quote_fieldsepc C s | � dddddggd� d S )Nr ra za
bcr� r� z1,2,"a
bc",3,4
r� r r r r �
test_newlinesL s z TestDialectExcel.test_newlinesN) r r r r2 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 )� EscapedExcelrA N)r r r r rE r r5 r r r r r O s r c @ s" e Zd Ze� Zdd� Zdd� ZdS )�TestEscapedExcelc C s | � dggd� d S )Nr� �
abc\,def
r� r r r r �test_escape_fieldsepV s z%TestEscapedExcel.test_escape_fieldsepc C s | � ddgg� d S )Nr r� r� r r r r �test_read_escape_fieldsepY s z*TestEscapedExcel.test_read_escape_fieldsepN)r r r r r2 r r r r r r r S s r c @ s e Zd ZdZdd� Zdd� ZdS )�TestDialectUnixZunixc C s | � dddggd� d S )Nr �abc defr� �"1","abc def","abc"
r� r r r r r� _ s z"TestDialectUnix.test_simple_writerc C s | � ddddgg� d S )Nr r� r r� r� r r r r �test_simple_readerb s z"TestDialectUnix.test_simple_readerN)r r r r2 r� r r r r r r \ s r c @ s e Zd ZejZdZdS )�QuotedEscapedExcelrA N)r r r r rq r r5 r r r r r
e s r
c @ s" e Zd Ze� Zdd� Zdd� ZdS )�TestQuotedEscapedExcelc C s | � dggd� d S r� r� r r r r �test_write_escape_fieldsepl s z1TestQuotedEscapedExcel.test_write_escape_fieldsepc C s | � ddgg� d S )Nz "abc\,def"
r� r� r r r r r o s z0TestQuotedEscapedExcel.test_read_escape_fieldsepN)r r r r
r2 r r r r r r r i 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%d&� Zd'S )(�TestDictFieldsc C sD t ddd��.}tj|dddgd�}|�� }| �|d� W 5 Q R X d S ) NrR r rS �f1�f2�f3��
fieldnamesr_ )r r �
DictWriter� writeheaderr1 )r
rZ r+ Zwriteheader_return_valuer r r � test_writeheader_return_valueu s z,TestDictFields.test_writeheader_return_valuec C s� t ddd��n}tj|dddgd�}|�� |�d� | �|�� d � |�d
d d �� |�d� |�� | �|�� d
� W 5 Q R X d S )NrR r rS r r r r r �
f1,f2,f3
r_ r� )r r z 10,,abc
) r r r r rV r1 �readlinerU rW r r r r �test_write_simple_dict{ 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 r r r� �f�r r r ra � Zxyzz f1,f2,f3
1,abc,f
2,5,xyz
)r r r r r1 �getvaluer} r r r r � test_write_multiple_dict_rows� s
�
�z,TestDictFields.test_write_multiple_dict_rowsc C s t � }| �ttj|� d S r )r r r r r �r
rZ r r r �test_write_no_fields� s z#TestDictFields.test_write_no_fieldsc
C s� t ddd��|}tj|dddgd�}| �t��}|�dd d
d �� W 5 Q R X t|j�}| �d |� | �d
|� | � d|� | �d|� W 5 Q R X d S )NrR r rS r r r r r_ Zspamr� )Zf4r r r z'f4'z'f2'r� )
r r r r r� rU �str� exception�assertInZ assertNotIn)r
rZ 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�Z extrasaction)r r r� r r r 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 �raiser$ r r ra r� �Zf0r r r )r r r r r� rU �r
rZ r+ Zdictrowr 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 �ignorer$ r r ra r� r' z1,2
)r r r rU r1 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 5 Q R X d S ) NrR z 1,2,abc
r r r r r r� r� r� r �r r( rV r �
DictReaderr1 r� �r
rZ 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 5 Q R X d S ) NrR �f1,f2,f3
1,2,abc
r r� r� r� r r r r )r r( rV r r- r1 r� r 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 5 Q R X d S ) NrR r0 r r r r r r� r� r� r ) r r( rV r r- r� r& r1 r r. r r r �#test_read_dict_fieldnames_from_file� s
�z2TestDictFields.test_read_dict_fieldnames_from_filec C s� dd l }td��h}|�d� |�d� t�|�}t|�}|�|g|�D ],}| �|j dddg� | �|ddd d
�� qFW 5 Q R X d S ) Nr rR r0 r r r r� r� r� r )
� itertoolsr r( rV r r- r� �chainr1 r )r
r3 rZ 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 5 Q R X d S )NrR �1,2,abc,4,5,6
r r r r r� r� r� r� r� r� )r r Nr, 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 5 Q R X d S )NrR r7 r r r �_rest)r �restkeyr� r� r� r� r� r� �r r r9 r, 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 5 Q R X d S )NrR zf1,f2
1,2,abc,4,5,6
r r9 �r: r r r� r� r� r� r� r� r; )r r( rV r r- r1 r r� 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 5 Q R X d S )NrR z1,2,abc,4,5,6
1,2,abc
r � 1 2 3 4 5 6ZDEFAULT)r Zrestvalr� r� r� r� r� r� �r� r� r� r� r� r� )r r( rV r r- �splitr1 r� 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 s2r Z
2147483648z43.0e12Z17r� r� )Zi1�floatZi2�s1�s2�r r- rA r1 r� )r
�sampler& 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 ) Nr7 r. r? r r� r� r� r� r� r� r@ rF �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
r? r� )r r r� r� r� r� r� r� r@ rF rI 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/ r1 r2 r6 r8 r< r> rB rH rJ rK 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 5 Q R X d S )Nr c S s g | ] }d | �qS )� r ��.0rb r r r �
<listcomp>" s z2TestArrayWrites.test_int_write.<locals>.<listcomp>rM rb rR r rS r� r� r- c S s g | ] }t |��qS r �r rN r r r rP ( s r. �
�arrayr� r r r+ rU �joinrV r1 rW �r
rS �contentsrm rZ 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 5 Q R X d S )Nr c S s g | ]}d | d �qS �rM g�������?r rN r r r rP . s z5TestArrayWrites.test_double_write.<locals>.<listcomp>rM r| rR r rS r� r� r- c S s g | ] }t |��qS r rQ rN r r r rP 3 s r. rR rU 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 5 Q R X d S )Nr c S s g | ]}d | d �qS rX r rN r r r rP 9 s z4TestArrayWrites.test_float_write.<locals>.<listcomp>rM r rR r rS r� r� r- c S s g | ] }t |��qS r rQ rN r r r rP > s r. rR rU r r r �test_float_write7 s
z TestArrayWrites.test_float_writec C sz dd l }dd l}|� d|j�}tddd��F}tj|dd�}|�|� d�|�d }|�d� | � |�
� |� W 5 Q R X d S )
Nr �urR r rS r� r� r- r. ) rS �string�
ascii_lettersr r r+ rU rT rV r1 rW )r
rS r\ rm rZ r+ r� r r r �test_char_writeB s
z TestArrayWrites.test_char_writeN)r r r rW rY rZ r^ r r r r rL s rL 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 5 Q R X | �t |j �d� d|_ | �t j�� }|� W 5 Q R X | �t |j �d� d S ) Nc @ s& e Zd ZdZdZdZdZdZej Z
dS )z3TestDialectValidity.test_quoting.<locals>.mydialectr� rA FTr. N� r r r r r5 r4 r7 r r rE r r r r r � mydialectO s ra Tr/ z''z("quotechar" must be a 1-character stringr� z#"quotechar" must be string, not int)
r r� r1 r rE r r! r4 r" r �
assertTruer r! �r
ra r| �cmr r r � test_quotingN s0 � �z TestDialectValidity.test_quotingc C s G dd� dt j�}|� }| �|jd� d|_| �t j�� }|� W 5 Q R X | �t|j�d� d|_| �t j�� }|� W 5 Q R X | �t|j�d� d|_| �t j�� }|� W 5 Q R X | �t|j�d� d |_| �t j�� }|� W 5 Q R X | �t|j�d
� d S ) Nc @ s& e Zd ZdZdZdZdZdZej Z
dS )z5TestDialectValidity.test_delimiter.<locals>.mydialectr� rA FTr. Nr` r r r r ra q s ra r� �:::z("delimiter" must be a 1-character stringr � ,z%"delimiter" must be string, not bytesr� z#"delimiter" must be string, not int)r r� r1 r r r! r r! rc r r r �test_delimiterp s6 � � � �z"TestDialectValidity.test_delimiterc C sz G dd� dt j�}|� }| �|jd� d|_|� }| �|jd� d|_| �t j�� }|� W 5 Q R X | �t|j�d� d S )Nc @ s& e Zd ZdZdZdZdZdZej Z
dS )z:TestDialectValidity.test_lineterminator.<locals>.mydialectr� rA FTr. Nr` r r r r ra � s ra r. rf r� z!"lineterminator" must be a string)r r� r1 r r r! r r! rc r r r �test_lineterminator� s �z'TestDialectValidity.test_lineterminatorc
C sv dd� }dD ]d}| j |d��N | �tj||d� | �tj||d� | �tj||d� | �tj||d� W 5 Q R X q 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 ra � s ra )r r� r; )�
field_name�valuera r| r r r �create_invalid� s z>TestDialectValidity.test_invalid_chars.<locals>.create_invalid)r r5 r )rj r r� � xr )�subTestr r r! )r
rl rj r r r �test_invalid_chars� s z&TestDialectValidity.test_invalid_charsN)r r r re rh ri ro r r r r r_ M 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 �Snifferr3 �
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 rq )r rr r3 rs �sample8�header2rv r r r �'test_has_header_regex_special_delimiter� s z3TestSniffer.test_has_header_regex_special_delimiterc
C sp t �� }dD ]^}| �|��J |�|d�}| �|jd� | �|jd� | �|jd� | �|j d� W 5 Q R X q d S )N)z ;'123;4';z'123;4';z;'123;4'z'123;4'z,;r� �'F)
r rr rn �sniffr1 r r r3 r4 r7 )r
rw �headerr2 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/ Tr0 r| F)
r rr r} rt r1 r r r3 r7 �sample2�r
rw r2 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� r| �+)r rr r} �sample3r" r r1 �sample4�sample5�sample6�sample7r ry �sample9r� r r r �test_delimiters s* zTestSniffer.test_delimitersc C s� t �� }|�| j�}| �|j� |�| j�}| �|j� |�| j�}| �|j� |�| j �}| �|j� |�| j
�}| �|j� d S r ) r rr r} ru Z assertFalser4 rz r� rb ry r� r� r r r �test_doublequote# s z TestSniffer.test_doublequoteN)r r r rt r� ru r� r� r� r� r� rz ry r� rx r{ r r� r� r� r r r r rp � s"
rp c @ s e Zd Zdd� ZeZdS )�NULc G s d S r r )�sr< r r r r( 1 s z NUL.writeN)r r r r( �
writelinesr r r r r� 0 s r� �gettotalrefcountz requires sys.gettotalrefcount()c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� TestLeaksc C sx d}t �� }td�D ]R}t�� | �tjg � t �� }t�dg� t�dg� t�dg� || }|}q| � |d� d S )Nr rM �a,b,c
r� �
�sysr� r� �gc�collectr1 �garbager r& �
assertLess)r
�delta�lastrcrb �rcr r r �test_create_read8 s zTestLeaks.test_create_readc C sx d}t �� }t� }td�D ]L}t�� | �tjg � t �� }t� |� t� |� t� |� || }|}q| �
|d� d S )Nr rM r� ) r� r� r� r� r� r� r1 r� r r+ r� )r
r� r� r� rb r� r r r �test_create_writeG s
zTestLeaks.test_create_writec C sr d}dgd }t �� }td�D ]B}t�� | �tjg � t �� }t�|�}|D ]}qN|| }|}q | � |d� d S )Nr r� r rM r� )r
r� r� r� rb r� Zrdrr� r r r � test_readW s
zTestLeaks.test_readc C s� d}dddggd }t � }t�� }td�D ]L}t�� | �tjg � t�� }t� |�}|D ]}|�
|� qZ|| }|}q*| � |d� d S )Nr r ra r� r rM ) r� r� r� r� r� r� r1 r� r r+ rU r� ) r
r� r� r� r� rb r� r+ r� r r r �
test_writeg s
zTestLeaks.test_writeN)r r r r� r� r� r� r r r r r� 5 s r� 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 5 Q R X d S �NrR r r� r� r- r. r ) r r( rT �namesrV r r& r1 r� r. r r r �test_unicode_read s
z TestUnicode.test_unicode_readc C s^ t dddd��F}t�|�}|�| j� d�| j�d }|�d� | �|�� |� W 5 Q R X d S r� ) r r r+ rU r� rT rV r1 rW )r
rZ r+ r� r r r �test_unicode_write� s
z TestUnicode.test_unicode_writeN)r r r r� r� r� r r r r r� x s �r� c @ s e Zd Zdd� Zdd� ZdS )�KeyOrderingTestc
C s� t � }td�D ]f}tdddd��N}t�||�}|�� |�d� t�|�}t|j �}| �
||� |� |� W 5 Q R X q| �
t |�dd� d S ) NZabcderR r r� r� r �x z@Key ordering: some key permutations not collected (expected 120))
�setr r r r r rV r- �tupler r1 �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)r� �Graham)r� �ChapmanZOver1ZOver2)r� �Under1)r� N)r� �John)r� �CleeseZ OtherInfor= r �fname�lnamer )r� r� )r� r� )r� r� )r� r� )r� r� )r� N)r� r� )r� r� )r �
splitlinesr1 r� r r- r )r
�datar r r �test_ordered_dict_reader� s:
�
��
�
��
�
��z(KeyOrderingTest.test_ordered_dict_readerN)r r r r� r� r r r r r� � s r� c @ s e Zd Zdd� ZdS )� MiscTestCasec C s ddh}t j| td|d� d S )N�__doc__� __version__)r Z_csv)�extra)r Z check__all__r )r
r� r r r � test__all__� s zMiscTestCase.test__all__N)r r r r� r r r r r� � s r� �__main__)+r� r� Zunittest�ior �tempfiler r r� r� �testr r3 r �textwrapr � collectionsr r ZTestCaser r� r� r� r� r r r r
r r
rL r_ rp r� Z
skipUnless�hasattrr� r� r� r� r �mainr r r r �<module> sL ^ka ..jy �A4