def test_correct_file_extension(self): srt = SrtMod('lalala.srt') result = srt.check_file_extension() self.assertTrue(result)
def test_incorrect_file_extension(self): srt = SrtMod('lalala.txt') result = srt.check_file_extension() self.assertFalse(result)