def test_clean_up(self, ecmd): processor = FileProcessor('another-input.mkv', 'output.mkv', 'roku') # Clean up, make sure all files are removed inputs = ['audio-2.mp3', 'audio-4.mp3', 'subtitle-5.srt'] processor.clean_up(inputs) cmd = ['rm', 'audio-2.mp3', 'audio-4.mp3', 'subtitle-5.srt'] self.assertTrue(ecmd.called) ecmd.assert_called_once_with(cmd)