def test_get_ffmpeg_input_data(self):
path = os.path.join(DATA_DIR, 'ffmpeg-output.txt')
with open(path, 'r') as fp:
data = fp.read()
fp.close()
data = worker.get_ffmpeg_input_data(data)
lines = data.split('\n')
self.assertEqual(' Metadata:', lines[1])
def test_parse_ffmpeg_data(self):
path = os.path.join(DATA_DIR, 'ffmpeg-output.txt')
with open(path, 'r') as fp:
data = fp.read()
fp.close()
data = worker.get_ffmpeg_input_data(data)
data = worker.parse_ffmpeg_data(data)
self.assertEqual(1095, data['video_bitrate'])
self.assertEqual(126, data['audio_bitrate'])
self.assertEqual(29970, data['framerate'])
self.assertEqual(640, data['width'])
self.assertEqual(360, data['height'])
self.assertEqual(2, data['num_audio_channels'])
self.assertEqual(3180, data['duration']) # 31.8 * framerate
