def test_build_with_no_playlist(self):
function = 'build'
no_playlist_file = None
with self.assertRaises(ValueError):
run_dataset(function, self.valid_output, no_playlist_file,
self.url, self.valid_audio_file,
self.valid_transcript_file, self.videoid)
def test_align_with_invalid_audio(self):
function = 'align'
invalid_audio_file = 'mock_audio'
with self.assertRaises(ValueError):
run_dataset(function, self.valid_output, self.valid_playlist_file,
self.url, invalid_audio_file,
self.valid_transcript_file, self.videoid)
def test_download_with_no_url(self):
function = 'download'
url = None
with self.assertRaises(ValueError):
run_dataset(function, self.valid_output, self.valid_playlist_file,
url, self.valid_audio_file, self.valid_transcript_file,
self.videoid)
def test_build_with_invalid_output(self):
function = 'build'
invalid_output = 'mock_output'
with self.assertRaises(ValueError):
run_dataset(function, invalid_output, self.valid_playlist_file,
self.url, self.valid_audio_file,
self.valid_transcript_file, self.videoid)
def test_crop_with_no_videoid(self):
function = 'crop'
no_videoid = None
with self.assertRaises(ValueError):
run_dataset(function, self.valid_output, self.valid_playlist_file,
self.url, self.valid_audio_file,
self.valid_transcript_file, no_videoid)
def test_download_with_url(self, mock_download):
function = 'download'
run_dataset(function, self.valid_output, self.valid_playlist_file,
self.url, self.valid_audio_file,
self.valid_transcript_file, self.videoid)
self.assertTrue(mock_download.called)
def test_build_calls_build_with_valid_input(self, mock_build):
function = 'build'
run_dataset(function, self.valid_output, self.valid_playlist_file,
self.url, self.valid_audio_file,
self.valid_transcript_file, self.videoid)
self.assertTrue(mock_build.called)
def test_crop_calls_crop_video_with_valid_input(self, mock_crop_video):
function = 'crop'
run_dataset(function, self.valid_output, self.valid_playlist_file,
self.url, self.valid_audio_file,
self.valid_transcript_file, self.videoid)
self.assertTrue(mock_crop_video.called)
def test_align_calls_get_align_with_valid_input(self, mock_get_align):
function = 'align'
run_dataset(function, self.valid_output, self.valid_playlist_file,
self.url, self.valid_audio_file,
self.valid_transcript_file, self.videoid)
self.assertTrue(mock_get_align.called)
