def test_throw_exception_on_loading_unknown_subtitle(self): try: unknown_file_path = os.path.join(self.__resource_tmp, "subtitle_test.unknown") Path(unknown_file_path).touch() Undertest.shift_subtitle(unknown_file_path, 2, "", "") except Exception as e: self.assertTrue(isinstance(e, UnsupportedFormatException)) else: self.fail("Should have thrown exception")
def test_shift_stl_subtitle(self): shifted_stl_file_path = os.path.join(self.__resource_tmp, "subtitle_test.srt") Undertest.shift_subtitle( self.__stl_file_path, 2, shifted_stl_file_path, suffix="_test" ) with open(shifted_stl_file_path) as shifted: for j, ls in enumerate(shifted): pass shifted_line_num = j + 1 self.assertEqual(32, shifted_line_num)
def test_shift_sami_subtitle(self): shifted_sami_file_path = os.path.join(self.__resource_tmp, "subtitle_test.sami") Undertest.shift_subtitle( self.__sami_file_path, 2, shifted_sami_file_path, suffix="_test" ) with open(self.__sami_file_path) as original: for i, lo in enumerate(original): pass with open(shifted_sami_file_path) as shifted: for j, ls in enumerate(shifted): pass original_line_num = i + 1 shifted_line_num = j + 1 self.assertEqual(original_line_num, shifted_line_num)
def test_shift_srt_subtitle_without_destination_file_path(self): shifted_srt_file_path = os.path.join(self.__resource_tmp, "subtitle_test.srt") another_shifted_srt_file_path = os.path.join( self.__resource_tmp, "subtitle_test_aligned.srt") Undertest.shift_subtitle(self.__srt_file_path, 2, shifted_srt_file_path, suffix="_test") Undertest.shift_subtitle(shifted_srt_file_path, 2, None, suffix="_aligned") with open(shifted_srt_file_path) as original: for i, lo in enumerate(original): pass with open(another_shifted_srt_file_path) as shifted: for j, ls in enumerate(shifted): pass original_line_num = i + 1 shifted_line_num = j + 1 self.assertEqual(original_line_num, shifted_line_num)