*j= d Z ddlmZ ddlZddlmc mZ ddl m
Z
ddl m Z ddl
mZmZmZmZ e
G d d Z G d d
ZddZd Zd Zd Zd Zd Zd ZdS )uN Tests for compiler/stages/transcribe.py — offline via fake downloader/model. ) annotationsN) dataclass)Path)_resolve_audio_urlis_audio_sourcetranscribe_alltranscribe_sourcec < e Zd ZU ded< ded< ded< dZded< dS ) _Segmentfloatstartendstrtext
SPEAKER_01speakerN)__name__
__module__ __qualname____annotations__r G/Users/jameslopez/projects/MentorCore/tests/compiler/test_transcribe.pyr r s> LLLJJJ
III G r r c e Zd ZddZd dZdS )
_FakeModelreturnNonec d| _ d S )Nr )calls)selfs r __init__z_FakeModel.__init__ s
r audior c | xj dz
c_ |j }d} || }|sddt j v st j | rt j | ndt j | t j | t j | dz }t t j | d x}x}}|d }d}||k }|slt j d|fd ||f t j | t j | d
z } d d | iz }t t j | d x}x}}t d
dd t ddd gddifS )N .m4azJassert %(py6)s
{%(py6)s = %(py2)s
{%(py2)s = %(py0)s.endswith
}(%(py4)s)
}r" )py0py2py4py6languageen==)z%(py1)s == %(py4)s)py1r( zassert %(py6)sr) g g ?zHello from the archive.g ?g @zThis is the second turn.) r endswith @py_builtinslocals
@pytest_ar_should_repr_global_name _safereprAssertionError_format_explanation_call_reprcomparer )
r r" kwargs @py_assert1 @py_assert3 @py_assert5 @py_format7 @py_assert0 @py_assert2 @py_format5s
r
transcribez_FakeModel.transcribe s
a
~%f%~~f%%%%%%%%%%%%%%%u%%%%%u%%%%%%~%%%%f%%%%%%%%%%%%%%%%%%%%%%%%%%j!)T)!T))))))))))!T)))))!))))T))))))))))))))))))))))))))) d$=>>c#=>>
r N)r r )r" r )r r r r! r@ r r r r r s<
r r srcdict media_dirr r c z | dd || d dz }| d |S )NT)parentsexist_okidr% s
fake audio)mkdir write_bytes)rA rC paths r _fake_downloaderrK ) sH
OOD4O000 #d)))) )D]### Kr c ddd} t | }|sddt j v st j t rt j t ndt j | t j | dz }t
t j | d x} }ddd} t | }|sddt j v st j t rt j t ndt j | t j | dz }t
t j | d x} }ddd } t | }|sddt j v st j t rt j t ndt j | t j | dz }t
t j | d x} }dd
d d } t | }| }|sd
dt j v st j t rt j t ndt j | t j | dz }t
t j | d x} x}}d S )Nxyoutube)rG
fetch_typez,assert %(py4)s
{%(py4)s = %(py0)s(%(py2)s)
}r )r& r' r( rss interview)rG typeurlarticle)rG rO rR z0assert not %(py4)s
{%(py4)s = %(py0)s(%(py2)s)
})r r0 r1 r2 r3 r4 r5 r6 )r9 r: r? r; @py_format6s r ?test_is_audio_source_matches_audio_fetch_types_and_source_typesrV 0 s "%Y??@?? @ @@ @@@@@@@@@@@@?@@@@@?@@@@@@?@@@@ @@@@@@@@@@@@@@@@@@@@"%U;; 00:00:01.250] SPEAKER_01: Hello from the archive.)in)z%(py1)s in %(py3)s
transcript)r. py3zassert %(py5)spy5zD[00:00:01.500 --> 00:00:03.000] SPEAKER_01: This is the second turn.r$ z-%(py2)s
{%(py2)s = %(py0)s.calls
} == %(py5)sr\ r& r' rf assert %(py7)sra )
r r rK r2 r7 r0 r1 r3 r4 r5 r6 read_textr )r_ r\ rA r^ r: r; @py_assert6 @py_assert8r9 @py_format9 @py_format11rd r= r> @py_format4rU @py_assert4 @py_format8s r Btest_transcribe_source_downloads_and_writes_timestamped_transcriptrr 7 s LLE . C 5# F !&9X%9(99%(99969 99999999996999999999999969999969999999999999X99999X9999999999(99999999999999999999999999999999!!##J P^ PT^ ^^^^^^^^^^ PT^^^^^^ P^^^^^^^^^^^T^^^^^^T^^^^^^^^^^^^^^^^^^^^^^^^^^^^ Q_ QU_ __________ QU______ Q___________U______U____________________________ ;!;! ;!55;!r c t }| dz }| |dz }| d dddd}t ||t | }||k }|st j d |fd
||f d t j v st j | rt j
| nd d t j v st j | rt j
| nd d
z }dd|iz }t t j | d }|j
} | } d}
| |
k } | st j d | fd| |
f d t j v st j | rt j
| nd t j
| t j
| t j
|
dz } dd| iz }
t t j |
d x}x} x} }
|j }d}||k } | st j d | fd||f dt j v st j | rt j
| ndt j
| t j
| dz }dd|iz } t t j | d x}x} }d S )NrY r] zcached transcript
rX rQ rN rG rR rO rZ r, )z%(py0)s == %(py2)sr^ cached)r& r' zassert %(py4)sr( )zH%(py4)s
{%(py4)s = %(py2)s
{%(py2)s = %(py0)s.read_text
}()
} == %(py7)sr` assert %(py9)spy9r rg r\ rh ri ra )r rH
write_textr rK r2 r7 r0 r1 r3 r4 r5 r6 rj r )r_ r\ raw_dirru rA r^ r9 @py_format3r? r: rk r; rq @py_format10rp rU s r -test_transcribe_source_uses_cached_transcriptr| O s8 LLE G MMOOO
(
(F
+,,, 9
M
MC
sG8HPU
V
V
VF V 6V66VV 6 6!66 !6 6666666666 !66666666666666666666666666 6666 6666!666666666666666666666666666666 ;!;! ;!55;!r c t }ddddddddd g}t || d
z t | }t | }dg}||k }|st j d |fd
||f dt
j v st j t rt j t nddt
j v st j | rt j | ndt j | t j | dz }dd|iz }t t j | d x}x}}|d } | j } | }|s`dt j | t j | t j | dz }
t t j |
d x} x}}|j
} d}| |k } | st j d | fd| |f dt
j v st j | rt j | ndt j | t j | dz }
dd|
iz }
t t j |
d x} x} }d S )Nz article-1rT rS rt podcast-1podcastrP zhttps://x.test/rss)rG rR rO rss_urlrY rZ r, )z0%(py3)s
{%(py3)s = %(py0)s(%(py1)s)
} == %(py6)slistresults)r& r. re r) zassert %(py8)spy8zAassert %(py5)s
{%(py5)s = %(py3)s
{%(py3)s = %(py1)s.exists
}()
})r. re rf r$ rg r\ rh ri ra )r r rK r r2 r7 r0 r1 r3 r4 r5 r6 existsr )r_ r\ sourcesr r> r; rp r< rm r= rU r9 r: rq s r &test_transcribe_all_skips_text_sourcesr ^ s LLEIUCCIUOcddG
5# G ==)[M)=M ))))))))))=M))))))))))))4)))))4))))))))))))))))))))))))=))))M))))))))))))))))))))))))))) ; ( &( & & ( (( ((((( (((( &(((( (((((((((((((((((((((( ;!;! ;!55;!r c ddddd} t | }d}||k }|st j d|fd||f dt j v st j t rt j t ndt j | t j | t j | d z }d
d |iz }t t j | d x} x}x}}d S ) Nr~ rP z https://example.com/feed.xmlz#https://cdn.example.com/episode.mp3)rG rO r episode_urlr, z0%(py4)s
{%(py4)s = %(py0)s(%(py2)s)
} == %(py7)sr r` rv rw ) r r2 r7 r0 r1 r3 r4 r5 r6 )r9 r: rk r; rq r{ s r *test_resolve_audio_url_prefers_episode_urlr q s " #9 D
1
1 1
1
1
r c 0 | dz }| d ddt | d}t | }d}||k }|st j d|fd||f d t j v st j t rt j t nd t j | t j | t j | d
z }d d |iz }t t j
| d x}x}x}}d S )
Nzfeed.xmlz
-
Episode 1
r~ rP )rG rO r z%https://cdn.example.com/episode-1.mp3r, r r r` rv rw ) rx r r r2 r7 r0 r1 r3 r4 r5 r6 )r_ feedr9 r: rk r; rq r{ s r )test_resolve_audio_url_from_rss_enclosurer s j DOO " # t99
3
3 3
3
3 r )rA rB rC r r r )__doc__
__future__r builtinsr0 _pytest.assertion.rewrite assertionrewriter2 dataclassesr pathlibr compiler.stages.transcriber r r r r r rK rV rr r| r r r r r r r so T T " " " " " " ! ! ! ! ! !
" T T T 0 & r