U
1ea$ @ s^ d Z ddlmZmZ ddlmZmZmZ ddl Z ddl
m Z m Z ddl
Z
ddlZddlmZ ddlZz ddlZW n ek
r Y nX ddlmZ ddlmZ d d
Zd d Zd
d Zdd Zdd Z dd Z dd Z dd Z dd Z dd Z!d d Z"d d Z#d!d" Z$d#d$ Z%G d%d& d&e Z&d'd( Z'e' G d)d* d*e&Z(ed2d,d-Z)d.d/ Z*d0d1 Z+dS )3z Tests for IPython.lib.display.
)NamedTemporaryFilemkdtemp)splitjoindirnameN)TestCasemock)BytesIO)display)skipif_not_numpyc C s t d} t td} dS )z(FileLink: Test class can be instantiated example.txtN)r
FileLinkpathlibPurePathfl r B/tmp/pip-unpacked-wheel-3hxk2k58/IPython/lib/tests/test_display.pytest_instantiation_FileLink, s
r c C s t d} | ds tdS )zEFileLink: Calling _repr_html_ on non-existent files returns a warningr zPath (example.txt)N)r
r
_repr_html_
startswithAssertionErrorr r r r *test_warning_on_non_existent_path_FileLink2 s
r c C s: t } t| j}| }d| j| jf }||ks6tdS )zMFileLink: Calling _repr_html_ functions as expected on existing filepath
z'%s
N)r r
r
namer r tfr actualexpectedr r r test_existing_path_FileLink8 s
r c C s0 t } t| j}t|}| j}||ks,tdS )zHFileLink: Calling repr() functions as expected on existing filepath
N)r r
r
r reprr r r r r test_existing_path_FileLink_reprB s
r c C s t } tttj| dS )z1FileLink: Raises error when passed directory
N)r pytestraises
ValueErrorr
r
)tdr r r #test_error_on_directory_to_FileLinkL s r% c C s t d} dS )z.FileLinks: Test class can be instantiated
exampleN)r
FileLinksZflsr r r test_instantiation_FileLinksV s r) c C s t d} | ds tdS )zFFileLinks: Calling _repr_html_ on non-existent files returns a warningr& zPath (example)N)r
r' r r r r( r r r +test_warning_on_non_existent_path_FileLinks[ s
r* c C s t } t| d}t| d}t| }| }|d}| d| d|jddt|jd f d|jddt|jd f g}| ||kst dS ) zIFileLinks: Calling _repr_html_ functions as expected on existing dir
dir
z%s/
z3 %s
\/ N)
r r r
r' r r sortr replacer r$ tf1tf2r r r r r r test_existing_path_FileLinksa s"
r6 c C sj t } t| d}t| d}dd }tj| |d}| }|d}| ddg}| ||ksftdS ) zMFileLinks: Calling _repr_html_ functions as expected w/ an alt formatter
r+ c S s ddgS Nhelloworldr r fnamesZincluded_suffixesr r r fake_formatter} s zBtest_existing_path_FileLinks_alt_formatter..fake_formatter)Znotebook_display_formatterr- r8 r9 N)r r r
r' r r r1 r r$ r4 r5 r< r r r r r r *test_existing_path_FileLinks_alt_formatterw s
r> c C s t } t| d}t| d}t| }t|}|d}| d| dt|jd dt|jd g}| ||ks|tdS )zFFileLinks: Calling repr() functions as expected on existing directory r+ r- z%s/z %sr0 N) r r r
r' r r r1 r r r3 r r r !test_existing_path_FileLinks_repr s
*r? c C sj t } t| d}t| d}dd }tj| |d}t|}|d}| ddg}| ||ksftdS ) zEFileLinks: Calling repr() functions as expected w/ alt formatter
r+ c S s ddgS r7 r r: r r r r< s zGtest_existing_path_FileLinks_repr_alt_formatter..fake_formatter)Zterminal_display_formatterr- r8 r9 N)r r r
r' r r r1 r r= r r r /test_existing_path_FileLinks_repr_alt_formatter s
r@ c C s& t } t| d}tttj|j dS )z-FileLinks: Raises error when passed file
r+ N)r r r! r" r# r
r' r )r$ r4 r r r test_error_on_file_to_FileLinks s
rA c C s t } t| d}t | d}t|d}t| }t|}|d}t|dksTt|tj| dd}t|}|d}t|dkst|dS )z5FileLinks: Does not recurse when recursive=False
r+ r- F) recursive N)r r r
r' strr lenr )r$ r ZsubtdZsubtfr r r r r test_recursive_FileLinks s
rG c C s t ttd} tj| d d S )Nztest.wavfilename)pjoinr __file__r
Audio)pathr r r test_audio_from_file s rN c @ sP e Zd Zedd Zedd Zedd Zedd Zed d
Zd d Z d
S )TestAudioDataWithNumpyc C s2 t }tj|dd}tt|jt|ks.td S ND rate)
get_test_toner
rL rF read_wavdatar self test_toneaudior r r test_audio_from_numpy_array s z2TestAudioDataWithNumpy.test_audio_from_numpy_arrayc C s6 t }tjt|dd}tt|jt|ks2td S rP )rT r
rL listrF rU rV r rW r r r test_audio_from_list s z+TestAudioDataWithNumpy.test_audio_from_listc C s | ttjt d S N) assertRaisesr# r
rL rT rX r r r /test_audio_from_numpy_array_without_rate_raises s zFTestAudioDataWithNumpy.test_audio_from_numpy_array_without_rate_raisesc C sP t t jj}dD ]8}tjt|dd}t t t|j }||kst
qd S )N)r0 ?rD rQ rR ) numpyiinfoint16maxr
rL rT absrU rV r )rX expected_max_valuescalerZ actual_max_valuer r r test_audio_data_normalization s
z4TestAudioDataWithNumpy.test_audio_data_normalizationc C sr t t jj}dD ]Z}t|}t t |}t|| }tj|ddd}t t t |j
}||kst qd S )N)r0 rb g?rQ FrS normalize) rc rd re rf rT rg intr
rL rU rV r )rX Z max_int16ri rY Ztest_tone_max_absrh rZ rj r r r %test_audio_data_without_normalization s z zfTestAudioDataWithNumpy.test_audio_data_without_normalization_raises_for_invalid_data..c S s t jdgdddS )NgjtrQ Frl rp r r r r rq rr )r_ r# r` r r r =test_audio_data_without_normalization_raises_for_invalid_data s zTTestAudioDataWithNumpy.test_audio_data_without_normalization_raises_for_invalid_dataN)
__name__
__module__ __qualname__r r[ r] ra rk ro rs r r r r rO s
rO c C s@ z dd l } tdtjtdW S tk
r: dd Y S X d S )Nr z numpy.array)Z side_effectc S s | S r^ r )xr r r rq rr z.simulate_numpy_not_installed..)rc r patchZ MagicMock ImportErrorModuleNotFoundError)rc r r r simulate_numpy_not_installed s
r{ c @ s e Zd Zedd ZdS )TestAudioDataWithoutNumpyc s( t t gd | t fdd d S )NrD c s t j ddS rP rp r Z
stereo_signalr r rq rr zMTestAudioDataWithoutNumpy.test_audio_raises_for_nested_list..)r\ rT r_ TypeErrorr` r r} r !test_audio_raises_for_nested_list s z;TestAudioDataWithoutNumpy.test_audio_raises_for_nested_listN)rt ru rv r r r r r r r| s r| r0 c C s& t dt j d t ddd | S )NrD i r r0 rQ )rc sinpiZlinspace)ri r r r rT s rT c
C sT t t| <}|| }| | }td| |W 5 Q R S Q R X d S )Nz<%sh)waveopenr Z
readframesZ
getnframesZ getnchannelsstructunpack)rV Z wave_fileZ wave_dataZ num_samplesr r r rU s rU c C s" t jtd} | ds td S )NrH z