def test_throw_no_frame_rate_exception_on_audio(self):
try:
Undertest.get_frame_rate(self.__test_audio_path)
except Exception as e:
self.assertTrue(isinstance(e, NoFrameRateException))
else:
self.fail("Should have thrown exception")
def test_throw_terminal_exception_on_bad_video(self):
try:
Undertest.extract_audio("bad_video_file_path", True, 16000)
except Exception as e:
self.assertTrue(isinstance(e, TerminalException))
self.assertFalse(os.path.exists("bad_video_file_path.mp4.wav"))
else:
self.fail("Should have thrown exception")
def test_extract_audio_wav_from_start_to_end(self):
self.__audio_file_path = Undertest.extract_audio(
self.__video_file_path, True, 16000)
segment_path, duration = Undertest.extract_audio_from_start_to_end(
self.__audio_file_path, "00:00:13,750", "00:00:16,150")
self.assertTrue(os.path.isfile(segment_path))
self.__segment_paths.append(segment_path)
self.assertEqual(2.4, duration)
def test_throw_exception_on_extract_audio_timeout(self, mock_communicate):
try:
Undertest.extract_audio(self.__video_file_path)
except Exception as e:
self.assertTrue(mock_communicate.called)
self.assertTrue(isinstance(e, TerminalException))
self.assertTrue("interrupted" in str(e))
else:
self.fail("Should have thrown exception")
def test_throw_exception_on_vtt2srt_exception(self, mock_communicate):
try:
Undertest.extract_audio(self.__video_file_path)
except Exception as e:
self.assertTrue(mock_communicate.called)
self.assertTrue(isinstance(e, TerminalException))
self.assertTrue("Cannot extract audio from video:" in str(e))
else:
self.fail("Should have thrown exception")
def test_throw_exception_on_get_frame_rate_interrupted(
self, mock_communicate):
try:
Undertest.get_frame_rate(self.__video_file_path)
except Exception as e:
self.assertTrue(mock_communicate.called)
self.assertTrue(isinstance(e, TerminalException))
self.assertTrue("interrupted" in str(e))
else:
self.fail("Should have thrown exception")
def test_throw_exception_on_get_frame_rate_timeout(self, mock_communicate):
try:
Undertest.get_frame_rate(self.__video_file_path)
except Exception as e:
self.assertTrue(mock_communicate.called)
self.assertTrue(isinstance(e, NoFrameRateException))
self.assertTrue(
"Timeout on extracting the frame rate from video:" in str(e))
else:
self.fail("Should have thrown exception")
def test_throw_exception_on_extract_partial_audio_interrupted(
self, mock_communicate):
try:
Undertest.extract_audio_from_start_to_end(self.__test_audio_path,
"00:00:13,750",
"00:00:16,150")
except Exception as e:
self.assertTrue(mock_communicate.called)
self.assertTrue(isinstance(e, TerminalException))
self.assertTrue("interrupted" in str(e))
else:
self.fail("Should have thrown exception")
def test_throw_exception_on_extract_audio_with_error_code(
self, mock_popen):
mock_popen.returncode.return_value = 1
mock_popen.communicate = Mock()
mock_popen.communicate.return_value = 1
try:
Undertest.extract_audio(self.__video_file_path)
except Exception as e:
self.assertTrue(mock_popen.communicate.called_with(180))
self.assertTrue(isinstance(e, TerminalException))
self.assertTrue("Cannot extract audio from video:" in str(e))
else:
self.fail("Should have thrown exception")
def test_get_audio_segment_starts_and_ends(self):
subs = pysrt.open(self.__subtitle_file_path, encoding="utf-8")
segment_starts, segment_ends, new_subs = Undertest.get_audio_segment_starts_and_ends(
subs)
self.assertEqual(len(segment_starts), len(segment_ends))
self.assertEqual(len(segment_starts), len(new_subs))
for sub in new_subs:
self.assertIsInstance(sub, pysrt.SubRipFile)
def test_get_log_loss_on_speech_shorter_than_subtitle(self):
undertest_obj = Undertest(n_mfcc=20)
shorter_audio_file_path, _ = MediaHelper.extract_audio_from_start_to_end(self.__audio_file_path, "00:00:00,000", "00:00:32,797")
self.__audio_file_paths.append(shorter_audio_file_path)
subs, audio_file_path, voice_probabilities, frame_rate = undertest_obj.predict_single_pass(
shorter_audio_file_path, self.__srt_file_path, self.__weights_dir
)
log_loss = undertest_obj.get_log_loss(voice_probabilities, subs)
self.assertGreater(log_loss, 0)
self.assertIsNone(frame_rate)
def test_extract_audio_aac(self):
self.__audio_file_path = Undertest.extract_audio(
self.__video_file_path)
self.assertTrue(os.path.isfile(self.__audio_file_path))
def test_extract_audio_wav(self):
self.__audio_file_path = Undertest.extract_audio(
self.__video_file_path, True, 16000)
self.assertTrue(os.path.isfile(self.__audio_file_path))
def test_refragment_with_min_duration(self):
subs = pysrt.open(self.__subtitle_file_path, encoding="utf-8")
new_subs = Undertest.refragment_with_min_duration(subs, 20)
self.assertTrue(len(new_subs) < len(subs))
self.assertEqual(new_subs[0].start, subs[0].start)
self.assertTrue(new_subs[-1].end, subs[-1].end)
def test_get_frame_rate(self):
self.assertEqual(24.0,
Undertest.get_frame_rate(self.__video_file_path))
def test_get_duration_in_seconds(self):
duration = Undertest.get_duration_in_seconds(start="02:10:12,222",
end="03:12:24,328")
self.assertEqual(3732.106, duration)
def test_get_duration_in_seconds_without_start(self):
duration = Undertest.get_duration_in_seconds(start=None,
end="01:01:01,100")
self.assertEqual(3661.100, duration)