def test_get_media_type(self): # instantiate the class f = FFMpeg() self.assertEqual(f.get_media_type('test.mp4'), { 'audio': True, 'video': True }) self.assertEqual(f.get_media_type('test.aac'), {'audio': True}) self.assertEqual(f.get_media_type('noaudio.mp4'), {'video': True})