From 0df5e4505b6a388bf611767f39d2d4f5a65475eb Mon Sep 17 00:00:00 2001 From: Chidi Williams Date: Wed, 1 Feb 2023 08:59:46 +0000 Subject: [PATCH] Update torch --- tests/transcriber_test.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/tests/transcriber_test.py b/tests/transcriber_test.py index 21f29b3..8a5446b 100644 --- a/tests/transcriber_test.py +++ b/tests/transcriber_test.py @@ -108,8 +108,9 @@ class TestWhisperFileTranscriber: @pytest.mark.parametrize( 'word_level_timings,expected_segments,model,check_progress', [ - (False, [Segment(0, 6560, - ' Bienvenue dans Passe-Relle. Un podcast pensé pour évêiller la curiosité des apprenances')], + (False, [Segment(0, 8400, + ' Bienvenue dans Passe-Relle. Un podcast pensé pour évêiller la curiosité des apprenances ' + 'et des apprenances de français.')], TranscriptionModel(model_type=ModelType.WHISPER, whisper_model_size=WhisperModelSize.TINY), True), (True, [Segment(40, 299, ' Bien'), Segment(299, 329, 'venue dans')], TranscriptionModel(model_type=ModelType.WHISPER, whisper_model_size=WhisperModelSize.TINY), True), @@ -146,9 +147,7 @@ class TestWhisperFileTranscriber: [call_args.args[0] == (0, 100) for call_args in mock_progress.call_args_list]) assert any( [call_args.args[0] == (100, 100) for call_args in mock_progress.call_args_list]) - assert any( - [(0 < call_args.args[0][0] < 100) and (call_args.args[0][1] == 100) for call_args in - mock_progress.call_args_list]) + assert len(mock_progress.call_args_list) > 2 mock_completed.assert_called() segments = mock_completed.call_args[0][0]