o
Ďia$ @ 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 Z/home/jeff/fluffinator/venv/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
FileLinksflsr 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"
r8 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 fnamesincluded_suffixesr r r fake_formatter} zBtest_existing_path_FileLinks_alt_formatter..fake_formatter)notebook_display_formatterr/ r; r< N)r r r
r( r r r3 r% r6 r7 r@ r r r r r r *test_existing_path_FileLinks_alt_formatterw
rD 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 %sr2 N)r r r
r( r r r3 r r5 r r r !test_existing_path_FileLinks_repr s
*rF 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 r9 r: r r= r r r r@ rA zGtest_existing_path_FileLinks_repr_alt_formatter..fake_formatter)terminal_display_formatterr/ r; r< N)r r r
r( r r r3 rC r r r /test_existing_path_FileLinks_repr_alt_formatter rE rH 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% r6 r r r test_error_on_file_to_FileLinks s
rI 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 subtdsubtfr r r r r test_recursive_FileLinks s
rQ c C s t ttd} tj| d d S )Nztest.wavfilename)pjoinr __file__r
Audio)pathr r r test_audio_from_file s rX 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
rV rN 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 rZ )r^ r
rV listrN r_ r` ra r r r test_audio_from_list s z+TestAudioDataWithNumpy.test_audio_from_listc C s | ttjt d S N) assertRaisesr$ r
rV r^ rb 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)r2 ?rL r[ r\ )
numpyiinfoint16maxr
rV r^ absr_ r` )rb expected_max_valuescalerd 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)r2 rl g?r[ Fr] normalize) rm rn ro rp r^ rq intr
rV r_ r` )rb max_int16rs rc test_tone_max_absrr rd rt r r r %test_audio_data_without_normalization s z zfTestAudioDataWithNumpy.test_audio_data_without_normalization_raises_for_invalid_data..c S r| )Ngjtr[ Frv r} r r r r r~ r )ri r$ rj 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 re rg rk ru r{ r r r r r rY s
rY c C s: zdd l } tdtjtdW S ty dd Y S w )Nr z numpy.array) side_effectc S s | S rh r )xr r r r~ s z.simulate_numpy_not_installed..)rm r patch MagicMock ImportErrorModuleNotFoundError)rm r r r simulate_numpy_not_installed s r c @ s e Zd Zedd ZdS )TestAudioDataWithoutNumpyc s( t t gd | t fdd d S )NrL c s t j ddS rZ r} r
stereo_signalr r r~ s zMTestAudioDataWithoutNumpy.test_audio_raises_for_nested_list..)rf r^ ri TypeErrorrj r r r !test_audio_raises_for_nested_list s z;TestAudioDataWithoutNumpy.test_audio_raises_for_nested_listN)r r r r r r r r r r s r r2 c C s& t dt j d t ddd | S )NrL i r r2 r[ )rm sinpilinspace)rs r r r r^ s &r^ c C s` t t| }|| }| | }td| |W d S 1 s)w Y d S )Nz<%sh)waveopenr
readframes
getnframes getnchannelsstructunpack)r` wave_file wave_data num_samplesr r r r_ s
$r_ c C s" t jtd} | dsJ d S )NrR z