DSj!2dZddlmZddlZddlmcmZddl Z ddl Z ddl Z ddl m Z ddlZde j d<ddlmZddlmZmZddlZddlmZd Zd Zd Zd Zd ZdZGddZGddZ GddZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(y)uMeeting-mode tests — WAV recorder, multilingual STT selection/fallback, transcript ordering, and the background transcription worker + backlog/finalize logic. No real models, no audio devices, no Claude session (fakes throughout). Run: .venv/bin/python -m pytest test_meeting.py -q ) annotationsN)Path0RESUME)MeetingSession WavRecorder)VoiceAppc |dz }t|}tjtjdddtjdz}tj ddtj}|j ||j ||j}d }||k(}|stjd |fd ||fd tjvstj|rtj|nd tj|tj|d z}dd|iz} ttj| dx}x}}|j } d }d} || z } | | z } t#| }d}||k}|s2tjd|fd||fdtjvstjt"rtjt"ndd tjvstj|rtj|nd tj| tj|tj| tj|tj|dz}dd|iz}ttj|dx} x}x} x} x} x}x}}|j%t'j(t+|d5}|j,}|}d} || k(}|stjd |fd|| fdtjvstj|rtj|ndtj|tj|tj| dz} dd| iz}ttj|dx}x}x}} |j.}|}d} || k(}|stjd |fd || fdtjvstj|rtj|ndtj|tj|tj| dz} dd| iz}ttj|dx}x}x}} |j0}|}d} || k(}|stjd |fd!|| fdtjvstj|rtj|ndtj|tj|tj| dz} dd| iz}ttj|dx}x}x}} |j2}|}d } || k(}|stjd |fd"|| fdtjvstj|rtj|ndtj|tj|tj| dz} dd| iz}ttj|dx}x}x}} tj4|j7d d#j9tjd$z }dddtj:}dd}d%} |||| &} | sd'd(tjvstjtrtjtnd(tj|tj|d)tjvstj|rtj|nd)tj| tj| d*z}ttj|dx}x}x} } tj:}|dd}d%} |||| &} | sd'd(tjvstjtrtjtnd(tj|tj|d+tjvstj|rtj|nd+tj| tj| d*z}ttj|dx}x}x} } y#1swY_xYw),Nzclip.wavri@dtypeg?ig?i==)z/%(py2)s {%(py2)s = %(py0)s.samples } == %(py5)srecpy0py2py5assert %(py7)spy7i>gư>)<)zg%(py11)s {%(py11)s = %(py0)s((%(py3)s {%(py3)s = %(py1)s.seconds } - (%(py5)s / %(py7)s))) } < %(py14)sabs)rpy1py3rrpy11py14zassert %(py16)spy16rb)zK%(py4)s {%(py4)s = %(py2)s {%(py2)s = %(py0)s.getnchannels }() } == %(py7)sw)rrpy4rassert %(py9)spy9)zK%(py4)s {%(py4)s = %(py2)s {%(py2)s = %(py0)s.getsampwidth }() } == %(py7)s)zK%(py4)s {%(py4)s = %(py2)s {%(py2)s = %(py0)s.getframerate }() } == %(py7)s)zI%(py4)s {%(py4)s = %(py2)s {%(py2)s = %(py0)s.getnframes }() } == %(py7)sOd66ec|dz }t|}|jtjgdtj|j t jt|d5}tj|jdd}dddg}d}d}||k(}|} |r|d } d} | } | | k(} | } | stjd |fd ||ftj|tj|d z}d d|iz}|j||r_tjd  fd  ftj| tj dz}dd|iz}|j|tj|diz}dd|iz}t!tj"|dx} x}x}x}x}x} x} x} } |d} d}| |k(}|sltjd |fd| |ftj| tj|dz}dd|iz}t!tj"|dx} x}}y#1swYxYw)Nzhot.wav)g@ggg?r rr%rirr)z%(py3)s == %(py6)s)rpy6z%(py8)spy8)z%(py11)s == -%(py14)s)rrz%(py17)spy17zassert %(py20)spy20r$z%(py1)s == %(py4)srr!assert %(py6)sr\)rr.r'arrayr,r9r:r;r<rArBr0r1r5append_format_boolopr6r7)rErFrr rWrGrLrUrH @py_assert0rPrQ @py_assert15rR @py_format7 @py_format9 @py_format16 @py_format18 @py_format19 @py_format21rI @py_format5s rXtest_wav_recorder_clipsro0sv9A a.CIIbhh,BJJ?@IIK 3q64 AmmALLO59 !/3q6/U/6U?/s1v/%/%/v////6U///6///U///////v///v///%/////////////// q6Q6Q;6Q6Q ! s 6'II%ctj}d}||k(}|stjd|fd||fdt j vstj trtjtndtj|tj|dz}dd|iz}ttj|dx}x}}tj}d}||u}|stjd|fd ||fdt j vstj trtjtndtj|tj|dz}dd|iz}ttj|dx}x}}tjd }tj}||k(}|stjd|fd ||ftj|dt j vstj trtjtndtj|d z}dd|iz}ttj|dx}x}}d tjD}t|}|sddt j vstj trtjtndtj|tj|dz}ttj|dx}}dtjD}t|}|sddt j vstj trtjtndtj|tj|dz}ttj|dx}}y)N"mlx-community/whisper-large-v3-mlxr)z9%(py2)s {%(py2)s = %(py0)s.MEETING_STT_MODEL } == %(py5)svmrrris)z8%(py2)s {%(py2)s = %(py0)s.MEETING_STT_LANG } is %(py5)sr)z9%(py1)s == %(py5)s {%(py5)s = %(py3)s.MEETING_STT_MODEL })rrrc3$K|]}d|v yw)turboN.0rs rX z,test_meeting_stt_defaults..B7%6w!|%6,assert %(py4)s {%(py4)s = %(py0)s(%(py2)s) }anyrrr!c3$K|]}d|v yw)smallNrwrxs rXr{z,test_meeting_stt_defaults..Cr|r}) rrMEETING_STT_MODELr0r1r2r3r4r5r6r7MEETING_STT_LANG_STT_FALLBACKSr)rGrHrIrJrKrfrLrns rXtest_meeting_stt_defaultsr=s4  G#GG #G GGGG #GGGGGGG2GGG2GGG GGG#GGGGGGGG  &$& $ &&&& $&&&&&&2&&&2&&& &&&$&&&&&&&  Q 72#7#77 #7 7777 #7777 77777727772777#777777777R%6%67737 77 777777377737777777 77777777R%6%67737 77 777777377737777777 7777777c g G fdd}|jtd|tj\}}d}||v}|stjd|fd||ftj |dt jvstj|rtj |nddz}d d |iz}ttj|d x}}gd } |k(}|stjd |fd |fdt jvstj rtj ndtj |dz}d d |iz}ttj|d x}}y )zDlarge-v3 + turbo fail to load -> loader degrades to the small model.ceZdZfdZy)6test_load_meeting_stt_fallback..FakeMeetingSTTcZj|d|vrt|d||_y)Nrz not in cache)rd RuntimeErrorrepo)selfrattemptss rX__init__z?test_load_meeting_stt_fallback..FakeMeetingSTT.__init__Ks0 OOD !d""dV=#9::DIrN__name__ __module__ __qualname__r)rsrXFakeMeetingSTTrJs rr MeetingSTTrinz%(py1)s in %(py3)snamerrassert %(py5)srN)rqz$mlx-community/whisper-large-v3-turbozmlx-community/whisper-small-mlxr)z%(py0)s == %(py3)sr)rr) setattrrrload_meeting_sttr0r1r5r2r3r4r6r7) monkeypatchrsttrrfrL @py_format4rJrGrs @rXtest_load_meeting_stt_fallbackrFs$HL.9##%IC 7d?7d7dd8  8     rc6Gdd}|jtd| tjd}|sNtjddzdtj |iz}t tj|d}y#t$rYywxYw)NceZdZdZy),test_load_meeting_stt_all_fail..Boomctd)Nzno model)r)rrs rXrz5test_load_meeting_stt_all_fail..Boom.__init__]s z* *rNrrwrrXBoomr\s +rrrFz#should raise when every model failsz >assert %(py1)sr) rrrrr0_format_assertmsgr5r6r7r)rrrf @py_format2s rXtest_load_meeting_stt_all_failr[sq++L$/  ;u;;;;;;u;;;;;;    sA(B BBc& tdd}|jddd|jdd d |jjd d}|d }|j}d }||}|stdt j |t j |t j |t j |dz}tt j|dx}x}x}}|d}|j}d}||}|stdt j |t j |t j |t j |dz}tt j|dx}x}x}}|j}|}|j} |jd} | | k(} | st jd| fd| | fdtjvst j|rt j |ndt j |t j |t j | t j | dz} dd| iz} tt j| dx}x}x} x} } |j }|}|j} |jd} | | k(} | st jd| fd| | fdtjvst j|rt j |ndt j |t j |t j | t j | dz} dd| iz} tt j| dx}x}x} x} } |j}|j}d} || } | sddtjvst j|rt j |ndt j |t j |t j | t j | dz}tt j|dx}x}x} } y)NonlinezQ3 syncmodetitlethemsecondY@tsAhmedfirstgI@r$rz Ahmed: firstzLassert %(py7)s {%(py7)s = %(py3)s {%(py3)s = %(py1)s.startswith }(%(py5)s) }rrrrrz them: secondz _room.wavr)zf%(py6)s {%(py6)s = %(py4)s {%(py4)s = %(py2)s {%(py2)s = %(py0)s.room_wav_path }() }.name } == %(py9)smrrr!r\r#assert %(py11)srz _them.wav)zf%(py6)s {%(py6)s = %(py4)s {%(py4)s = %(py2)s {%(py2)s = %(py0)s.them_wav_path }() }.name } == %(py9)smeeting_zfassert %(py8)s {%(py8)s = %(py4)s {%(py4)s = %(py2)s {%(py2)s = %(py0)s.stem }.startswith }(%(py6)s) })rrr!r\r])radd full_text splitlines startswithr0r5r6r7 room_wav_pathrstemr1r2r3r4 them_wav_path)rlinesrfrLrHrMrKrGrIrUrN @py_assert7rV @py_format12ris rX'test_meeting_session_ordering_and_pathsrisHI6AEE&(uE%EE'7tE$ KKM $ $ &qr *E 8.8  .~. ~ .. ...8... ...~... ....... 8.8  .~. ~ .. ...8... ...~... ....... ??9? 9  ! !9xy%99 !%9 9999 !%999999919991999?999 999 !999%99999999 ??9? 9  ! !9xy%99 !%9 9999 !%999999919991999?999 999 !999%99999999 66(6  (Z( Z (( ((((((1(((1(((6((( (((Z((( (((((((rceZdZdZdZy)rz~Transcribe returns a label derived from the audio's constant fill so a test can prove WHICH utterance produced WHICH line.c Ndttt|ddzS)Nzsaid rd)introundfloat)raudios rX transcribezFakeMeetingSTT.transcribe{s's5uQx3!6789::rN)rrr__doc__rrwrrXrrxs ;;rrceZdZdZdZy)FakeTTScg|_yN)saidrs rXrzFakeTTS.__init__s  rc|jj|tjdtjS)Nr )rrdr'zerosint16)rtexts rXsynthz FakeTTS.synths) xx**rN)rrrrrrwrrXrrs +rrceZdZdZdZy) FakeSpeakerc d|_d|_y)NrF)playedspeakingrs rXrzFakeSpeaker.__init__s  rc.|xjdz c_y)Nr)r)rrWs rXplayzFakeSpeaker.plays q rN)rrrrrrwrrXrrs rrcVt}t|_t|_|Sr)r rttsrspeaker)apps rX_apprs *CiCG-CK Jrcn|jtddd}tj|y)NrctdfS)Nfake)rrwrrXz;test_worker_transcribes_labels_and_orders..s 9I68RrcKt}td}tjd}||_tj |j ||}|jdtjddtjd |jd tjdd tjd |jdtj|d d{|j}|}|sddtjvst!j"|rt!j$|ndt!j$|t!j$|dz}t't!j(|dx}}|j+j-dd}|d}d}||k(} | slt!j.d| fd||ft!j$|t!j$|dz}dd|iz} t't!j(| dx}x} }|d}d}||k(} | slt!j.d| fd||ft!j$|t!j$|dz}dd|iz} t't!j(| dx}x} }y7ͭw)Nrr maxsizerg?gi@rrg?r)timeoutz?assert %(py4)s {%(py4)s = %(py2)s {%(py2)s = %(py0)s.done }() }workerrr$rzAhmed: said 10rr`rarbr\rz them: said 90)rrasyncioQueue_meeting_stt_queue create_task_meeting_transcribe_worker_enqueue_meeting_audior'r-r, put_nowaitwait_fordoner2r3r0r4r5r6r7rrr1) rsessionqrrGrIrnbodyrfrLrhs rXgoz5test_worker_transcribes_labels_and_orders..gosf h/ MM" %!"$$S%C%CGQ%OP ""62773bjj+Ie"T ""7BGGCrzz,Ju"U Tvq111{{{}}vv{}  "--/3Aw***w*****w****w***********Aw)/)w/))))w/)))w)))/))))))) 2sC2K4K5G K)rrrrrunrrs rX)test_worker_transcribes_labels_and_ordersrs+.0RS*  KKrc>d}tj|y)Nc Kt}tjd|_t dD]9}|j dt jddt jd;|j}|j}|}d}||k(}|stjd |fd ||fd tjvstj|rtj|nd tj|tj|tj|tj|d z}d d|iz}t!tj"|dx}x}x}x}}yw)Nrrrrg??rr)zl%(py6)s {%(py6)s = %(py4)s {%(py4)s = %(py2)s {%(py2)s = %(py0)s._meeting_stt_queue }.qsize }() } == %(py9)srrrr)rrrrrangerr'r-r,qsizer0r1r2r3r4r5r6r7) r_rGrIrUrNrrVrs rXrz=test_enqueue_backlog_never_raises_and_is_wav_safe..gosf!(q!9rA  & &vrwwsC/LQT & U%%2%++2+-22-2222-222222s222s222%222+222-22222222222sE5E7rrrs rX1test_enqueue_backlog_never_raises_and_is_wav_safers3  KKrct}td|_d|_d|_d|_d|_d|_ifd}||_|fS)N in_personrcK|d<yw)N was_backlogrw)rrrcaptureds rX fake_finalizez)_prep_stopping_app..fake_finalizes"- ) rrmeeting _room_wav _them_wav_remote_capture_task _sysaudio_meeting_worker_task_finalize_meeting)rrrs @rX_prep_stopping_apprsW &C k2CKCMCM#CCM#CH.)C =rc>d}tj|y)NcKt\}}tjd|_|j d{tj dd{|d}d}||u}|slt jd|fd||ft j|t j|dz}d d |iz}tt j|dx}x}}d |jjD}t|}|sd d tjvst j trt jtnd t j|t j|dz}tt j|dx}}d|jjD}t|}| }|sdd tjvst j trt jtnd t j|t j|dz} tt j| dx}x}}y7-7w)Nxr{Gz?rFrsz%(py1)s is %(py4)srarbr\c3$K|]}d|v yw)StoppedNrwryss rXr{z@test_stop_no_backlog_says_stopped..go..s8.go..sG,Q|qwwy0,z0assert not %(py4)s {%(py4)s = %(py0)s(%(py2)s) })rrrr _stop_meetingsleepr0r1r5r6r7rrrr2r3r4) rrrfrIrLrnrhrGrUrJs rXrz-test_stop_no_backlog_says_stopped..gosy*, X!(s!;!!!mmD!!! &/%/&%////&%///&///%///////8377<<88s888888888s888s88888888888888G#'',,GG3GGGGGGGGGGGG3GGG3GGGGGGGGGGGGGG "!s"d}tj|y)NcKt\}}tjd}tdD]8}|j dt j dt jdf:||_|jd{tjdd{|d}d }||u}|sltjd |fd ||ftj|tj|d z}d d|iz}ttj|dx}x}}d|j j"D} t%| }|sddt'j(vstj*t$rtjt$ndtj| tj|dz}ttj|dx} }t|j-Dcgc]}|j/} }d}|| v}|stjd|fd|| ftj|dt'j(vstj*| rtj| nddz} dd| iz} ttj| dx}}y7>7$cc}ww)Nr rr[rrrr!rTrsr"rarbr\c3@K|]}d|jvywr(r*r%s rXr{zKtest_stop_with_backlog_says_still_processing..go..sCl<1779,lr,r~rrrrdrainedrrr)rrrr rr'rr,rr-r.r0r1r5r6r7rrrr2r3r4r  get_nowait) rrrr rfrIrLrnrhrGr3rrJs rXrz8test_stop_with_backlog_says_still_processing..gos*, X MM# &qA LL&"((3 ";SA B!"!!!mmD!!! &.$.&$....&$...&...$.......CcggllCCsCCCCCCCCCsCCCsCCCCCCCCCCCCCC+0+;<+;a1<<>+;<twtwtww "!=s8BKKK#K$E!KK B)KK Kr r s rX,test_stop_with_backlog_says_still_processingr5s  KKrcDfd}tj|y)Nc|Kt}tdd}|jdddij|dfd fd }||_|j |dd d{j }d }||}d }||u}|stjd|fd||fdtjvstjrtjndtj|tj|tj|tj|dz}dd|iz} ttj| dx}x}x}x}}|j}|j }|}| }|sddtjvstj|rtj|ndtj|tj|tj|dz} ttj| dx}x}x}}|jj#} d} | j$} | }| |v}|stjd|fd| |ftj| dtjvstj| rtj| ndtj| tj|dz} dd| iz}ttj|dx} x}x} }y7έw)Nrz finalize testrrhellorrsavec0jddxsdS)NrFzx.md) setdefault)savedsrXrzDtest_finalize_saves_and_notifies_inbox..go..su/?/?V/L/VPV/VrcKdd<yw)NT summarizedrw)mmr<s rX no_summaryzFtest_finalize_saves_and_notifies_inbox..go..no_summarys"&E, rT)rr>rs)zI%(py6)s {%(py6)s = %(py2)s {%(py2)s = %(py0)s.get }(%(py4)s) } is %(py9)sr<rrrz_assert not %(py6)s {%(py6)s = %(py4)s {%(py4)s = %(py2)s {%(py2)s = %(py0)s.inbox }.empty }() }r)rrr!r\readyr)zD%(py1)s in %(py7)s {%(py7)s = %(py5)s {%(py5)s = %(py3)s.lower }() }msgrr"r#)rrrr_summarize_meetingrgetr0r1r2r3r4r5r6r7inboxemptyr4r+)rrr@rGrIrUrNrrVrrKrBrfrHrMrLr<rs @rXrz2test_finalize_saves_and_notifies_inbox..gosf  ? C fg#&Av'VW '!+##At#>>>yy..y&.$.&$....&$......u...u...y......&...$.......99$9??$?$$$$$$$$$$$3$$$3$$$9$$$?$$$$$$$$$$ii""$%#))%)+%w+%%%%w+%%%w%%%%%%#%%%#%%%)%%%+%%%%%%% ?sA'L<+L9,KL<r rs` rX&test_finalize_saves_and_notifies_inboxrGs& KKr))r __future__rbuiltinsr2_pytest.assertion.rewrite assertionrewriter0rosr:pathlibrnumpyr'environ voice.meetingrrrrrmainappmodr rYrorrrrrrrrrrrr/r5rGrwrrXrTs #  85 1*8*   );;++,  "r