Your IP : 216.73.216.213
���]�0 � @ s� d d l m Z m Z m Z d d l Z d d l Z d d l Z d d l Z d d l Z Gd d � d e j � Z
Gd d � d � Z Gd d � d e � Z Gd d
� d
e � Z
d S) � )�findfile�TESTFN�unlinkNc @ s( e Z d Z d d � Z d d � Z d S)� UnseekableIOc C s
t j � d S)N)�io�UnsupportedOperation)�self� r �//opt/alt/python35/lib64/python3.5/audiotests.py�tell s zUnseekableIO.tellc O s
t j � d S)N)r r )r �args�kwargsr r r
�seek s zUnseekableIO.seekN)�__name__�
__module__� __qualname__r r r r r r
r s r c @ s: e Z d Z d Z d d � Z d d � Z d d � Z d S) �
AudioTestsFc C s d | _ | _ d S)N)�f�fout)r r r r
�setUp s zAudioTests.setUpc C sF | j d k r | j j � | j d k r8 | j j � t t � d S)N)r �closer r r )r r r r
�tearDown s
zAudioTests.tearDownc C sq | j | j � | � | j | j � | � | j | j � | � | j | j � | � | j | j � | � | j | j � | � | j � } | j | | | | | | | f � | j | j | � | j | j | � | j | j
| � | j | j | � | j | j | � | j | j
| � xF t t j d � D]1 } t j | | � }
| j t j |
� | � q8Wd S)N� )� assertEqual� getnchannels� getsampwidth� getframerate�
getnframesZ getcomptypeZ getcompname� getparams� nchannels� sampwidth� framerate�nframes�comptype�compname�range�pickle�HIGHEST_PROTOCOL�dumps�loads) r r r r r! r" r# r$ �params�proto�dumpr r r
� check_params s$ zAudioTests.check_paramsN)r r r �close_fdr 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# S)$�AudioWriteTestsc C sf | j j | d � } | _ | j | j � | j | j � | j | j � | j | j
| j � | S)N�wb) �module�openr � setnchannelsr � setsampwidthr � setframerater! � setcomptyper# r$ )r �testfiler r r r
� create_file7 s zAudioWriteTests.create_filec
C s� | j j | d � �� } | j | j � | j � | j | j � | j � | j | j � | j � | j | j � | � | j | j
| � | � Wd QRXd S)N�rb) r1 r2 r r r r r r r! r �
readframes)r r7 r" �framesr r r r
�
check_file? s zAudioWriteTests.check_filec C sn | j t � } | j | j � | j | j � | j | | j | j | j | j | j
| j � | j � d S)N)
r8 r �
setnframesr" � writeframesr; r- r r r! r# r$ r )r r r r r
�test_write_paramsG s z!AudioWriteTests.test_write_paramsc % C s� | j | j j � �" | j j t d � � Wd QRXWd QRX| j | j j � �8 t t d � �! } | j j | � � Wd QRXWd QRXWd QRXd S)Nr0 )� assertRaisesr1 �Errorr2 r )r r7 r r r
�&test_write_context_manager_calls_closeO s z6AudioWriteTests.test_write_context_manager_calls_closec C s` t t d � � } | j j | � �M } | j | j � | j | j � | j | j � | j | j
| j � Wd QRX| j | j
| j � Wd QRXt t d � �� } | j j | � �k } | j | j � j
� | j � } | j | j | j � | j | j | j � | j | j | j � Wd QRX| j s?| j | j � � | j | j
| j � Wd QRXd S)Nr0 r9 )r2 r r1 r3 r r4 r r5 r! r6 r# r$ r �closedr. � assertFalse�getfpr � assertIsNone)r r7 r r* r r r
�#test_context_manager_with_open_fileZ s" z3AudioWriteTests.test_context_manager_with_open_filec C s | j j t d � �M } | j | j � | j | j � | j | j � | j | j
| j � Wd QRX| j j t � �k } | j | j
� j � | j � } | j | j | j � | j | j | j � | j | j | j � Wd QRX| j s| j | j
� � d S)Nr0 )r1 r2 r r3 r r4 r r5 r! r6 r# r$ rD rE rC r r r. rF )r r r* r r r
�"test_context_manager_with_filenamem s z2AudioWriteTests.test_context_manager_with_filenamec C sV | j t � } | j | j � | j | j � | j � | j t | j | j � d S)N)r8 r r= r" r> r; r r< )r r r r r
�
test_write~ s
zAudioWriteTests.test_writec C s\ | j t � } | j | j � | j t | j � � | j � | j t | j | j � d S)N) r8 r r= r" r> � bytearrayr; r r< )r r r r r
�test_write_bytearray� s
z$AudioWriteTests.test_write_bytearrayc C sb | j t � } | j | j � | j t j d | j � � | j � | j t | j | j � d S)N�h) r8 r r= r" r> �arrayr; r r< )r r r r r
�test_write_array� s
z AudioWriteTests.test_write_arrayc C s\ | j t � } | j | j � | j t | j � � | j � | j t | j | j � d S)N) r8 r r= r" r> �
memoryviewr; r r< )r r r r r
�test_write_memoryview� s
z%AudioWriteTests.test_write_memoryviewc C s� t t d � �Q } | j d � | j | � } | j | j d � | j | j � | j � Wd QRXt t d � �9 } | j | j
d � d � | j | | j | j � Wd QRXd S)Nr0 s
ababagalamagar r9 �
) r2 r �writer8 r= r" r> r; r r �readr< )r r7 r r r r
�test_incompleted_write� s
z&AudioWriteTests.test_incompleted_writec C s� t t d � �� } | j d � | j | � } | j | j � | j | j } | j | j d | � � | j | j | d � � | j
� Wd QRXt t d � �9 } | j | j d � d � | j
| | j | j � Wd QRXd S)Nr0 s
ababagalamagar9 rQ )r2 r rR r8 r= r" r r r> r; r r rS r< )r r7 r � framesizer r r
�test_multiple_writes� s
z$AudioWriteTests.test_multiple_writesc C s� t t d � �Q } | j d � | j | � } | j | j d � | j | j � | j � Wd QRXt t d � �9 } | j | j
d � d � | j | | j | j � Wd QRXd S)Nr0 s
ababagalamagar r9 rQ ) r2 r rR r8 r= r" r> r; r r rS r< )r r7 r r r r
�test_overflowed_write� s
z%AudioWriteTests.test_overflowed_writec C so | j t � �' } | j | j � | j | j � Wd QRXt t d � � } | j | | j | j � Wd QRXd S)Nr9 )r8 r r= r" r> r; r r< )r r r7 r r r
�test_unseekable_read� s
z$AudioWriteTests.test_unseekable_readc C so t t d � �@ } | j | � �' } | j | j � | j | j � Wd QRXWd QRX| j t | j | j � d S)Nr0 )r r r8 r= r" r> r; r< )r r7 r r r r
�test_unseekable_write� s
z%AudioWriteTests.test_unseekable_writec " C s� t t d � �� } | j d � | j | � } | j | j d � y | j | j � Wn t k
rj Yn Xy | j � Wn t k
r� Yn XWd QRXt
t d � �= } | j | j d � d � | j
| | j d | j � Wd QRXd S)Nr0 s
ababagalamagar r9 rQ )r r rR r8 r= r" r> r; �OSErrorr r2 r rS r< )r r7 r r r r
�!test_unseekable_incompleted_write� s
z1AudioWriteTests.test_unseekable_incompleted_writec $ C s t t d � �� } | j d � | j | � } | j | j d � y | j | j � Wn t k
rj Yn Xy | j � Wn t k
r� Yn XWd QRXt
t d � �X } | j | j d � d � | j
| j } | j | | j d | j d | � � Wd QRXd S)Nr0 s
ababagalamagar r9 rQ )r r rR r8 r= r" r> r; rZ r r2 r rS r r r< )r r7 r rU r r r
� test_unseekable_overflowed_write� s
z0AudioWriteTests.test_unseekable_overflowed_writeN)r r r r8 r< r? rB rG rH rI rK rN rP rT rV rW rX rY r[ r\ r r r r
r/ 5 s" r/ c @ s^ e Z d Z e d d � � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
S)�AudioTestsWithSourceFilec C s t | j d d �| _ d S)NZsubdirZ audiodata)r Z sndfilename� sndfilepath)�clsr r r
�
setUpClass� s z#AudioTestsWithSourceFile.setUpClassc C sQ | j j | j � } | _ | j | | j | j | j | j | j | j
� d S)N) r1 r2 r^ r r- r r r! �sndfilenframesr# r$ )r r r r r
�test_read_params s z)AudioTestsWithSourceFile.test_read_paramsc C s� t | j d � �P } | j j | � } | _ | j | j � | j � | j | j | j � Wd QRXt t d � �z } | j j | d � } | _
| j | j � | j | j j � � | j � Wd QRX| j | j | j � | j � Wd QRXd S)Nr9 r0 )
r2 r^ r1 r rD rC r r r. r r r@ rA )r r7 r r r r r
�
test_close s
z#AudioTestsWithSourceFile.test_closec C s� | j | j } | j d d | � } | j d | d | � } | j j | j � } | _ | j | j d � d � | j | j � d � | j | j d � | � | j
� | j � } | j | d � | j | j d � | � | j � } | j | d � | j | j d � | � | j | � | j | j d � | � | j | � | j | j d � | � | j | j j
� � | j d � Wd QRX| j | j j
� � | j | j � d � Wd QRXd S)N� � r � r ���)r r r; r1 r2 r^ r r r: r �rewindZsetposr@ rA r )r rU Zchunk1Zchunk2r Zpos0Zpos2r r r
� test_read s,
z"AudioTestsWithSourceFile.test_readc C s | j j | j � } | _ | j j t d � } | _ | j | j � � d } | j � } x: | d k r� | d 7} | j | j
| � � | | 8} q` W| j � | j j t d � } | _ | j � | j
| j � | j � � | j
| j
| j � � | j
| j � � � d S)Nr0 r r r9 )r1 r2 r^ r r r Z setparamsr r r> r: r rh r )r r r �i�nr r r
� test_copy, s
z"AudioTestsWithSourceFile.test_copyc C s- t t d � �C } | j d � t | j d � � } | j | j � � Wd QRXWd QRXt t d � �� } | j | j d � d � | j j | d � �� } | j | j � | j � | j | j � | j
� | j | j � | j � | j | j
� | j � | j | j | j � | j � Wd QRXWd QRXd S)Nr0 s
ababagalamagar9 rQ )r2 r rR r^ rS r r1 r r r r r r! r ra r: r" r; )r r7 r r r r
�test_read_not_from_start= s
!z1AudioTestsWithSourceFile.test_read_not_from_startN)
r r r � classmethodr` rb rc ri rl rm r r r r
r] � s r] )Z test.supportr r r ZunittestrM r r& �sys�FileIOr r r/ r] r r r r
�<module> s %�