o
Zha$ @ s\ d Z ddlmZmZ ddlmZmZmZ ddl Z ddl
m Z m Z ddl
Z
ddlZddlmZ ddlZzddlZW n eyA Y nw 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 U/var/www/html/lang_env/lib/python3.10/site-packages/IPython/lib/tests/test_display.pytest_instantiation_FileLink, s
r c C t d} | dsJ dS )zEFileLink: Calling _repr_html_ on non-existent files returns a warningr zPath (example.txt)N)r
r
_repr_html_
startswithr r r r *test_warning_on_non_existent_path_FileLink2
r c C s: t } t| j}| }d| j| jf }||ksJ dS )zMFileLink: Calling _repr_html_ functions as expected on existing filepath
z'%s
N)r r
r
namer tfr actualexpectedr r r test_existing_path_FileLink8 s
r c C s0 t } t| j}t|}| j}||ksJ dS )zHFileLink: Calling repr() functions as expected on existing filepath
N)r r
r
r reprr 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 r )zFFileLinks: Calling _repr_html_ on non-existent files returns a warningr' zPath (example)N)r
r( r r r) r r r +test_warning_on_non_existent_path_FileLinks[ r r+ c C s t } t| d}t| d}t| }| }|d}| d| d|jddt|jd f d|jddt|jd f g}| ||ksLJ 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% tf1tf2r r r r r r test_existing_path_FileLinksa s"
r7 c C sj t } t| d}t| d}dd }tj| |d}| }|d}| ddg}| ||ks3J dS ) zMFileLinks: Calling _repr_html_ functions as expected w/ an alt formatter
r, c S ddgS Nhelloworldr r fnamesZincluded_suffixesr r r fake_formatter} zBtest_existing_path_FileLinks_alt_formatter..fake_formatter)Znotebook_display_formatterr. r: r; N)r r r
r( r r r2 r% r5 r6 r> r r r r r r *test_existing_path_FileLinks_alt_formatterw
rA c C s t } t| d}t| d}t| }t|}|d}| d| dt|jd dt|jd g}| ||ks>J dS )zFFileLinks: Calling repr() functions as expected on existing directory r, r. z%s/z %sr1 N)r r r
r( r r r2 r r4 r r r !test_existing_path_FileLinks_repr s
*rC c C sj t } t| d}t| d}dd }tj| |d}t|}|d}| ddg}| ||ks3J dS ) zEFileLinks: Calling repr() functions as expected w/ alt formatter
r, c S r8 r9 r r< r r r r> r? zGtest_existing_path_FileLinks_repr_alt_formatter..fake_formatter)Zterminal_display_formatterr. r: r; N)r r r
r( r r r2 r@ r r r /test_existing_path_FileLinks_repr_alt_formatter rB rD 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% r5 r r r test_error_on_file_to_FileLinks s
rE c C s t } t| d}t | d}t|d}t| }t|}|d}t|dks*J |tj| dd}t|}|d}t|dksDJ |dS )z5FileLinks: Does not recurse when recursive=False
r, r. F) recursive N)r r r
r( strr len)r% r ZsubtdZsubtfr r r r r test_recursive_FileLinks s
rK c C s t ttd} tj| d d S )Nztest.wavfilename)pjoinr __file__r
Audio)pathr r r test_audio_from_file s rR c @ sP e Zd Zedd Zedd Zedd Zedd Zed d
Zd d Z d
S )TestAudioDataWithNumpyc C s2 t }tj|dd}tt|jt|ksJ d S ND rate)
get_test_toner
rP rJ read_wavdataself test_toneaudior r r test_audio_from_numpy_array s z2TestAudioDataWithNumpy.test_audio_from_numpy_arrayc C s6 t }tjt|dd}tt|jt|ksJ d S rT )rX r
rP listrJ rY rZ r[ r r r test_audio_from_list s z+TestAudioDataWithNumpy.test_audio_from_listc C s | ttjt d S N) assertRaisesr$ r
rP rX r\ 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 ] }tjt|dd}t t t|j }||ks%J q d S )N)r1 ?rH rU rV )
numpyiinfoint16maxr
rP rX absrY rZ )r\ expected_max_valuescaler^ actual_max_valuer r r test_audio_data_normalization s z4TestAudioDataWithNumpy.test_audio_data_normalizationc C sr t t jj}dD ]-}t|}t t |}t|| }tj|ddd}t t t |j
}||ks6J q d S )N)r1 rf g?rU FrW normalize) rg rh ri rj rX rk intr
rP rY rZ )r\ Z max_int16rm r] Ztest_tone_max_absrl r^ rn r r r %test_audio_data_without_normalization s z zfTestAudioDataWithNumpy.test_audio_data_without_normalization_raises_for_invalid_data..c S rt )NgjtrU Frp ru r r r r rv rw )rc r$ rd 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_ ra re ro rs rx r r r r rS s
rS c C s: zdd l } tdtjtdW S ty dd Y S w )Nr z numpy.array)Z side_effectc S s | S rb r )xr r r rv s z.simulate_numpy_not_installed..)rg r patchZ MagicMock ImportErrorModuleNotFoundError)rg r r r simulate_numpy_not_installed s r c @ s e Zd Zedd ZdS )TestAudioDataWithoutNumpyc s( t t gd | t fdd d S )NrH c s t j ddS rT ru r Z
stereo_signalr r rv s zMTestAudioDataWithoutNumpy.test_audio_raises_for_nested_list..)r` rX rc TypeErrorrd r r r !test_audio_raises_for_nested_list s z;TestAudioDataWithoutNumpy.test_audio_raises_for_nested_listN)ry rz r{ r r r r r r r s r r1 c C s& t dt j d t ddd | S )NrH i r r1 rU )rg sinpiZlinspace)rm r r r rX s &rX c C s` t t| }|| }| | }td| |W d S 1 s)w Y d S )Nz<%sh)waveopenr Z
readframesZ
getnframesZ getnchannelsstructunpack)rZ Z wave_fileZ wave_dataZ num_samplesr r r rY s
$rY c C s" t jtd} | dsJ d S )NrL z