Your IP : 216.73.216.213
�
g faq � @ sZ d Z d d l Z d d l m Z d d l Z d d l Z d d l Z Gd d � d � Z Gd d � d � Z Gd d � d � Z
Gd
d � d e e e
e j � Z Gd d
� d
� Z
Gd d � d e e e
e j � Z Gd d � d e
e j � Z Gd d � d e � Z Gd d � d e � Z Gd d � d e � Z d d � Z e d k rVe � n d S)zdUnit tests for memory-based file-like objects.
StringIO -- for unicode strings
BytesIO -- for bytes
� N)�supportc @ sL e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d S) �MemorySeekTestMixinc C s" | j d � } | j | � } d S)N�
1234567890)�buftype�ioclass)�self�buf�bytesIo� r
�7/opt/alt/python34/lib64/python3.4/test/test_memoryio.py�testInit s z MemorySeekTestMixin.testInitc C s� | j d � } | j | � } | j | d d � | j d � � | j | d d � | j d � � | j | d d � | j d � � | j | j | j � � d S)Nr � � � i� )r r � assertEqual�read�EOF)r r r r
r
r �testRead s ###z MemorySeekTestMixin.testReadc C sQ | j d � } | j | � } | j | | j � � | j | j | j � � d S)Nr )r r r r r )r r r r
r
r �testReadNoArgs s z"MemorySeekTestMixin.testReadNoArgsc C s� | j d � } | j | � } | j d � | j d � | j | | j � � | j d � | j | d d � | j � � | j t | j d � d S)Nr r r � g )r r r �seekr � assertRaises� TypeError)r r r r
r
r �testSeek# s
z MemorySeekTestMixin.testSeekc C s~ | j d � } | j | � } | j d | j � � | j d � | j d | j � � | j d � | j d | j � � d S)Nr r r i' )r r r �tellr )r r r r
r
r �testTell/ s
z MemorySeekTestMixin.testTellN)�__name__�
__module__� __qualname__r r r r r r
r
r
r r
s
r c @ s e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d � Z d! d" � Z d# d$ � Z d% d& � Z d' d( � Z d) S)*�MemoryTestMixinc C s& | j � } | j | j | j � d S)N)r r �UnsupportedOperation�detach)r r r
r
r � test_detach< s zMemoryTestMixin.test_detachc C s& | j | j | d � � d � | j | j d � d � | j | j | d � � d � | j | j � d � | j | j d � d � | j | j � d � | j | j | d � � d � | j | j d � d � | j | j | d � � d � | j | j d
� d
� | j | j � d � d S) Nzblah.r r zHello.� z world
� �hr
� )r �writer r �truncate)r �f�tr
r
r � write_ops@ s zMemoryTestMixin.write_opsc C s� | j d � } | j | � } | j | | j � | j | j � | � | j � } | j | | j � | j | j � | � | j t | j d � | j � | j t | j | j d � � d S)Nz hello world
� )
r r r+ r �getvaluer r r' �close�
ValueError)r r �memior
r
r �
test_writeM s
zMemoryTestMixin.test_writec C s� | j d � } | j � } | j | j | g d � d � | j | j � | d � | j g � | j | j � | d � | j � } | j t | j | g d g � | j | j � | � | j t | j d � | j � | j t | j g � d S)Nr �d r
) r r r �
writelinesr- r r r. r/ )r r r0 r
r
r �test_writelinesZ s
z MemoryTestMixin.test_writelinesc s; � j � } � f d d � } � j t | j | � � d S)Nc 3 s � j d � Vt � d S)N�spam)r �KeyboardInterruptr
)r r
r � error_genk s z8MemoryTestMixin.test_writelines_error.<locals>.error_gen)r r r6 r3 )r r0 r7 r
)r r �test_writelines_errori s z%MemoryTestMixin.test_writelines_errorc C s| | j d � } | j | � } | j t | j d � | j d � | j | j � d � | j | j � | d d � � | j | j d � d � | j | j � | d d � � | j | j � d � | j d d � | j | � | j | j � | d d � | � | j � } | j | j d � | � | j | j � | � | j t
| j d � | j � | j t | j d � d S) Nr r
r# r r � �0���) r r r r/ r( r r r- r r' r r. )r r r0 �posr
r
r �
test_truncateq s$
$
z MemoryTestMixin.test_truncatec C s� | j d � } | j | � } | j | j � | � | j d � } | j | j � | j � | j | d � | j | j � | d � | j | � | j | j � | � | j t | j g � d S)Nr r9 )r r r r- r �__init__r r )r r r0 r
r
r � test_init� s
zMemoryTestMixin.test_initc C s� | j d � } | j | � } | j | j d � | j � | j | j d � | d d � � | j | j d � | d d � � | j | j d � | d d � � | j | j � | j � | j d � | j | j � | � | j | j � | j � | j | j � d � | j d � | j | j d
� | � | j d � | j t | j � � t | � � | j d � | j t | j � � t | � � | j d � | j | j d � | � | j t | j d � | j
� | j t | j � d S) Nr r r
r r i� �
r2 r, r; ) r r r r r r r �typer r r. r/ )r r r0 r
r
r � test_read� s, ###
"
"
zMemoryTestMixin.test_readc C sU | j d � } | j | d � } | j | j d � | j � | j | j � | � | j | j � | � | j | j � | j � | j d � | j | j d � | d d � � | j | j d � | d d � � | j | j d � | d d � � | j d � | j | j d
� | � | j d � | j | j d � | j � | j d � } | j | d d d � � } | j | j � | � | j | j � | � | j | j � | d d � � | j | j � | j � | j d � | j t | j � � t | � � | j | j � | � | j t | j d � | j � | j t
| j � d S)
Nz 1234567890
r9 r r r@ � r
r r, r; r; r; ) r r r �readliner r rA r r r. r/ )r r r0 r
r
r �
test_readline� s4
###
"
z MemoryTestMixin.test_readlinec C s� | j d � } | j | d � } | j | j � | g d � | j d � | j | j � | d d � g | g d � | j d � | j | j d � | g d � | j d � | j | j d
� | g d � | j d � | j | j d � | g d � | j d � | j t | j � d � t | � � | j d � | j | j d � | g d � | j t | j d � | j � | j t | j � d S) Nz 1234567890
r@ r r$ r rC r9 r
r, r; )
r r r � readlinesr rA r r r. r/ )r r r0 r
r
r �test_readlines� s$
.
&
z MemoryTestMixin.test_readlinesc C s' | j d � } | j | d � } | j t | � | � | j t | d � � | j t | d � � d } x( | D] } | j | | � | d 7} qq W| j | d � | j d � d } x( | D] } | j | | � | d 7} q� W| j | d � | j | d � } | j � | j t | j
� d S)Nz 1234567890
r@ �__iter__�__next__r r
r9 ) r r r �iterZ
assertTrue�hasattrr r. r r/ rI )r r r0 �i�liner
r
r �
test_iterator� s&
z MemoryTestMixin.test_iteratorc C s� | j d � } | j | � } | j | j � | � | j � | j | j � | � | j t | j � � t | � � | j | d � } | j | j � d d � | j d � � | j | � } | j � | j t | j � d S)Nr i� r Z890���) r r r r- r rA r. r r/ )r r r0 r
r
r �
test_getvalue� s
")
z MemoryTestMixin.test_getvaluec C s� | j d � } | j | � } | j d � | j t | j d � | j t | j d d � | j t | j d d � | j | j d � d � | j | j d d � d � | j | j � | � | j | j d � d � | j | j d d � d � | j | j � | d d � � | j | j t | � � t | � � | j | j � | j � | j t | � d � | j | j � | j � | j | j d d � t | � � | j | j � | j � | j � | j t | j d � d S) Nr r r
r r r9 r; r; )
r r r r r/ r r �lenr r. )r r r0 r
r
r � test_seek s(
%"
zMemoryTestMixin.test_seekc C s� | j d � } | j | � } | j | j t | � d � d � | j | j � | j � | j | j � d � | j | j � | � | j | j � | j | j � | � | j | � | j | j � | | j d � | � d S)Nr r
� � )
r r r r rQ r r r r- r' )r r r0 r
r
r �
test_overseek s #
z MemoryTestMixin.test_overseekc C s� | j d � } | j | � } | j | j � d � | j d � | j | j � d � | j d � | j | j � d � | j � | j t | j � d S)Nr r r i' )r r r r r r. r r/ )r r r0 r
r
r � test_tell& s
zMemoryTestMixin.test_tellc C s8 | j d � } | j | � } | j | j � d � d S)Nr )r r r �flush)r r r0 r
r
r �
test_flush2 s zMemoryTestMixin.test_flushc C s� | j � } | j | j � d � | j | j � d � | j | j � d � | j | j � d � | j | j d � | j � | j t | j � | j t | j � | j t | j � | j t | j � | j | j d � d S)NTF)
r r �writable�readable�seekable�isatty�closedr. r r/ )r r0 r
r
r �
test_flags8 s
zMemoryTestMixin.test_flagsc sc � j d � � � � f d d � } � � f d d � } � j | � � � � j | � � � d S)Nr c s/ Gd d � d � j � } | � � } | j � S)Nc @ s e Z d Z d S)z>MemoryTestMixin.test_subclassing.<locals>.test1.<locals>.MemION)r r r r
r
r
r �MemIOJ s r_ )r r- )r_ �m)r r r
r �test1I s z/MemoryTestMixin.test_subclassing.<locals>.test1c s8 G� f d d � d � j � } | � d � } | j � S)Nc s" e Z d Z � f d d � Z d S)z>MemoryTestMixin.test_subclassing.<locals>.test2.<locals>.MemIOc s � j j | | � d S)N)r r> )�me�a�b)r r
r r> P s zGMemoryTestMixin.test_subclassing.<locals>.test2.<locals>.MemIO.__init__N)r r r r> r
)r r
r r_ O s r_ )r r- )r_ r` )r r r
r �test2N s z/MemoryTestMixin.test_subclassing.<locals>.test2)r r )r ra re r
)r r r �test_subclassingG s
z MemoryTestMixin.test_subclassingc C s0 x) t d � D] } | j � } d | _ q
Wd S)Nr2 r
)�ranger �foo)r �_r0 r
r
r �test_instance_dict_leakW s z'MemoryTestMixin.test_instance_dict_leakc s� � j d � } � j | � } d | _ | j d � G� f d d � d � j � } d d l } d | _ | j | _ | | _ | | d � } | j d � x� t d t
j d � D]� } x� | | f D]� } t
j t
j
| d
| �� } � j | j � | j � � � j | j | j � � j | j | j � � j | j � | j � � | j � � j t t
j
| | � q� Wq� W| ` d S) Nr �* r9 c s" e Z d Z � f d d � Z d S)z6MemoryTestMixin.test_pickling.<locals>.PickleTestMemIOc s � j j | | � | | _ d S)N)r r> rh )rb Z initvaluerh )r r
r r> e s z?MemoryTestMixin.test_pickling.<locals>.PickleTestMemIO.__init__N)r r r r> r
)r r
r �PickleTestMemIOd s rl r �__main__�P r
Zprotocol)r r rh r rm r r r rl rg �pickleZHIGHEST_PROTOCOL�loads�dumpsr r- � __class__r r. r r/ ) r r r0 rl rm Zsubmemio�proto�objZobj2r
)r r �
test_pickling^ s*
!z MemoryTestMixin.test_picklingN)r r r r" r+ r1 r4 r8 r= r? rB rE rG rN rP rR rU rV rX r^ rf rj ru r
r
r
r r : s(
r c @ s e Z d Z d d � Z d S)� BytesIOMixinc C s$ | j d � } | j � } | j t | � d � | j d � | j � } | j t | � d � | j t | j d d � | j t | j � | j t | j � | j
| j � d | d d � <| j t | � d � | j | j � d � ~ t
j � | j � | j � | j t | j � d S) Ns
1234567890r � xr2 s abcr r# s
123abc7890)r � getbufferr �bytesr r � BufferErrorr' r( r. � assertFalser] r- r Z
gc_collectr/ )r r0 r r
r
r �test_getbuffer� s$
zBytesIOMixin.test_getbufferN)r r r r| r
r
r
r rv � s rv c @ s� e Z d Z e j Z e d d � � Z e j Z d Z d d � Z
d d � Z d d � Z d
d � Z
d d
� Z d d � Z d S)�
PyBytesIOTestc C s
| j d � S)N�ascii)�encode)�sr
r
r r � s zPyBytesIOTest.buftype� c C sK | j d � } | j | � } | j t | j � | j | j � | � d S)Nr )r r r r �read1r r )r r r0 r
r
r �
test_read1� s zPyBytesIOTest.test_read1c C s� | j d � } | j | � } t d � } | j | j | � d � | j | d � | j | j | � d � | j | d � | j | j | � d � | j | d � t d � } | j d � | j | j | � d � | j | d � t d
� } | j d � | j | j | � d � | j | d
� | j t | j d � d d l } | j d d � } | j | � } | j | � | j | j � d � | j
� | j t | j | � | j d
� } t � } | j d � | j | � | j | d
� d S)Nr s hellor s 12345s 67890r s hello worldr@ s 1234567890dr� r, rd s 123rk ) r r � bytearrayr �readintor r r �array�tobytesr. r/ )r r r0 rd r� rc r
r
r �
test_readinto� s<
zPyBytesIOTest.test_readintoc C s7 | j d � } | j | � } | j | j d
d � d � | j | j d d � d � | j | j d d � d � | j | j d d � d � | j | j d d � d � | j | j d d � d � | j d
d � | j | j � | d d � � | j d � | j d d � | j | j � | d d � � d S)Nr r
r r r r9 r$ r@ rS r; ���r; rO rO )r r r r r )r r r0 r
r
r �test_relative_seek� s
z PyBytesIOTest.test_relative_seekc C sU | j � } | j t | j d � | j t | j d � | j t | j d g � d S)Nr )r r r r' r3 )r r0 r
r
r � test_unicode� s zPyBytesIOTest.test_unicodec C s� d } d d l } | j d t | � � } | j | � } | j | j � | � | j | j | � d � | j | j � | � d S)Ns
1234567890r rd r@ )r� �listr r r- r' )r r r� rc r0 r
r
r �test_bytes_array� s z PyBytesIOTest.test_bytes_arrayc C s? | j d � } | j d | � | j t | j | d d �d S)Nr Z
initial_bytesrh )r r r r )r r r
r
r �test_issue5449� s z PyBytesIOTest.test_issue5449N)r r r �pyior � staticmethodr �BytesIOr r r� r� r� r� r� r� r
r
r
r r} � s ! r} c @ s� e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d � Z d d � Z
d S)�TextIOTestMixinc s� | j d d � � � f d d � } | j � j d � � j d � | � | j � j d � � j d � | � | j � j d
� � j d � | � | j � j d � d S) N�newlinec s � j d � � j � d S)Nr )r r r
)r0 r
r � force_decode s
z<TextIOTestMixin.test_newlines_property.<locals>.force_decodeza
�
zb
�
zc
d�
)r� r� )r� r� r� )r r �newlinesr' )r r� r
)r0 r �test_newlines_property� s
z&TextIOTestMixin.test_newlines_propertyc C s� | j � } | j t | j d d � | j t | j d d � | j t | j d d � | j t | j d d � | j t | j d d � | j t | j d d � d S)Nr
r r9 r; rO r; )r r �OSErrorr )r r0 r
r
r r� s z"TextIOTestMixin.test_relative_seekc C s@ | j � } | j | j � | j | j � | j | j � d S)N)r Z assertIsNone�encoding�errorsr{ �line_buffering)r r0 r
r
r �test_textio_properties s z&TextIOTestMixin.test_textio_propertiesc C s� | j d � } | j t | � d d d g � | j | j � d � | j � } | j | j d � d � | j d � | j t | � d d d g � | j | j � d � d S)Nza
b
c
dza
zb
zc
d� r )r r r� r- r' r )r r0 r
r
r �test_newline_default! s
z$TextIOTestMixin.test_newline_defaultc C s� | j d d d �} | j t | � d d d d g � | j d � | j | j d � d � | j | j d
� d � | j | j d
� d � | j | j d � d
� | j | j � d � | j d d � } | j d
| j d � � | j d | j d � � | j d | j d � � | j d � | j | j � d � | j | j � d � | j d d d �} | j | j d � d � d S)Nza
b
c
dr� za
zb
zc
�dr r
rc r9 z
bz
cr� za
b
c
dr zb
zc
dza
bza
b)r r r� r r r- r' )r r0 r
r
r �test_newline_none, s" "
z!TextIOTestMixin.test_newline_nonec C sd | j d d d �} | j t | � d d d d g � | j d � | j | j d � d
� | j | j d � d � | j | j d
� d � | j | j � d � | j d d � } | j d | j d � � | j d | j d � � | j d | j d � � | j d | j d � � | j d � | j t | � d d d d g � | j | j � d � d S)Nza
b
c
dr� r, za
zb
zc
r� r r za
b
r9 z
cr
r� zb
z
d)r r r� r r r- r' )r r0 r
r
r �test_newline_emptyB s "
"z"TextIOTestMixin.test_newline_emptyc C s� | j d d d �} | j t | � d d d g � | j | j � d � | j d d � } | j | j d � d � | j d � | j t | � d d d g � | j | j � d � d S) Nza
b
c
dr� r� za
zb
zc
dr� r )r r r� r- r' r )r r0 r
r
r �test_newline_lfU s
z TextIOTestMixin.test_newline_lfc C s | j d d d �} | j | j � d � | j d � | j t | � d d d d d g � | j | j � d � | j d d � } | j | j d � d
� | j d � | j t | � d d d d d g � | j d � | j | j � d d d d d g � | j | j � d � d S) Nza
b
c
dr� r� za
b
c
dr za
zb
zc
r� r� )r r r r r� r- r' rF )r r0 r
r
r �test_newline_cra s
%
%
%z TextIOTestMixin.test_newline_crc C s
| j d d d �} | j | j � d � | j d � | j t | � d d d g � | j d � | j | j � d d d g � | j | j � d � | j d d � } | j | j d � d � | j d � | j t | � d d d g � | j | j � d � d S)
Nza
b
c
dr� z
z
a
b
c
dr za
zb
zc
dr� )r r r r r� rF r- r' )r r0 r
r
r �test_newline_crlfq s
z!TextIOTestMixin.test_newline_crlfc C sH | j d d d �} | j | j d � d � | j | j � d � d S)Nza
b
r� r za
b
)r r r r- )r r0 r
r
r �test_issue5265� s z TextIOTestMixin.test_issue5265c C sW | j t | j d d �| j t | j d d �x d D] } | j d | � q9 Wd S) Nr� s
�errorr, r� r� �
)Nr, r� r� r� )r r r r/ )r r� r
r
r �test_newline_argument� s
z%TextIOTestMixin.test_newline_argumentN)r r r r� r� r� r� r� r� r� r� r� r� r� r
r
r
r r� � s
r� c @ s: e Z d Z e Z e j Z e j Z d Z d d � Z
d S)�PyStringIOTestr, c C sX | j d � } | j | j � d � | j � } | j d � | j | j � d � d S)Nu �)r r r r' r- )r r0 r
r
r �test_lone_surrogates� s
z#PyStringIOTest.test_lone_surrogatesN) r r r �strr r� �StringIOr r r r� r
r
r
r r� � s
r� c @ sD e Z d Z d Z e Z e j Z d Z Gd d � d e j � Z
d S)�PyStringIOPickleTestzETest if pickle restores properly the internal state of StringIO.
r, c @ s( e Z d Z d d � Z d d � Z d S)z PyStringIOPickleTest.ioclassc O s" t j t j t j | | � � � S)N)ro rp rq r� r� )�cls�args�kwargsr
r
r �__new__� s z$PyStringIOPickleTest.ioclass.__new__c O s d S)Nr
)r r� r� r
r
r r> � s z%PyStringIOPickleTest.ioclass.__init__N)r r r r� r> r
r
r
r r � s r N) r r r �__doc__r� r r� r r r� r r
r
r
r r� � s
r� c @ sX e Z d Z e j Z e j Z d d � Z d d � Z e j
Z
e j d d � � Z d S)� CBytesIOTestc C s� | j � } | j � } | j t | � d � t | d � | j | d t � | d d k rw | j | d t � n | j � | j t
| j � d S)Nr r r
r9 ) r � __getstate__r rQ r� �assertIsInstance�int�dictr. r r/ )r r0 �stater
r
r �
test_getstate� s
zCBytesIOTest.test_getstatec C s
| j � } | j d � | j t d � d d f � | j d d i d d 6f � | j t | j d
� | j t | j d � | j t | j d � | j t | j d � | j t | j d � | j t | j � | j t | j d � | j � | j t | j d � d S)N� no errorr r r5 r� r
�unicode� � len-test� closed)r� r Nr; )r� r; N)r� r N)r� r� N)r� r r )r� r )r� r N)r � __setstate__r� r r/ r r. )r r0 r
r
r �
test_setstate� s
zCBytesIOTest.test_setstatec C s� t j d � } | j } | j t j t j � � | � | t j � | � | t j d � | d d � | t j d d � | d d � d S)NZP2nN2Pn� ar
i� )r Z calcobjsize� check_sizeofr �object�
__sizeof__�ior� )r ZbasesizeZcheckr
r
r � test_sizeof� s zCBytesIOTest.test_sizeofN)
r r r r� r� r r r� r� r r� Z cpython_onlyr� r
r
r
r r� � s r� c @ sF e Z d Z e j Z e j Z d d � Z d d � Z d d � Z d S)�
CStringIOTestc C s� | j d � } | j | � } | j | j � | � | j | j | � t | � � | j | j � t | � � | j | j � | � | j | j | � t | � � | j | j � t | � d � | j | j � | | � d S)Nu 𠌊𠍇r9 )r r r r- r' rQ r )r r r0 r
r
r �
test_widechar� s zCStringIOTest.test_widecharc C s� | j � } | j � } | j t | � d � | j | d t � | j | d t � | j | d t � | d d k r� | j | d t � n | j � | j t
| j � d S)Nr r r
r9 r ) r r� r rQ r� r� r� r� r. r r/ )r r0 r� r
r
r r� � s
zCStringIOTest.test_getstatec C s | j � } | j d
� | j d d d i d d 6f � | j t | j d � | j t | j d � | j t | j d � | j t | j d � | j t | j d � | j t | j d � | j t | j d � | j t | j � | j t | j d � | j � | j t | j d � d S)N�no errorr� r r, r r5 r) r
r� � �len-testr] )r� r� r N)r, r) r Nr; )r, r, r; N)r� r, r N)r, r� r N)r, r, r� N)r, r, r r )r� r )zclosedr, r N)r r� r r/ r r. )r r0 r
r
r r� � s
zCStringIOTest.test_setstateN)
r r r r� r� r r r� r� r� r
r
r
r r� � s
r� c @ s2 e Z d Z e j Z Gd d � d e j � Z d S)�CStringIOPickleTestc @ s( e Z d Z d d � Z d d � Z d S)zCStringIOPickleTest.ioclassc O s" t j t j t j | | � � � S)N)ro rp rq r� r� )r� r� r� r
r
r r� s z#CStringIOPickleTest.ioclass.__new__c O s d S)Nr
)r r� r� r
r
r r> s z$CStringIOPickleTest.ioclass.__init__N)r r r r� r> r
r
r
r r s r N)r r r r� r r� r r
r
r
r r� s r� c C s) t t t t t t g } t j | � d S)N)r} r� r� r� r� r� r Z run_unittest)Ztestsr
r
r � test_main s r� rm )r� ZunittestZtestr r� Z_pyior� ro r r rv ZTestCaser} r� r� r� r� r� r� r� r r
r
r
r �<module> s* -� K X� *0