class TestShifting(unittest.TestCase): def setUp(self): self.item = WebVTTItem(1, text="Hello world !") self.item.shift(minutes=1) self.item.end.shift(seconds=20) def test_shift_up(self): self.item.shift(1, 2, 3, 4) self.assertEqual(self.item.start, (1, 3, 3, 4)) self.assertEqual(self.item.end, (1, 3, 23, 4)) self.assertEqual(self.item.duration, (0, 0, 20, 0)) self.assertEqual(self.item.characters_per_second, 0.65) def test_shift_down(self): self.item.shift(5) self.item.shift(-1, -2, -3, -4) self.assertEqual(self.item.start, (3, 58, 56, 996)) self.assertEqual(self.item.end, (3, 59, 16, 996)) self.assertEqual(self.item.duration, (0, 0, 20, 0)) self.assertEqual(self.item.characters_per_second, 0.65) def test_shift_by_ratio(self): self.item.shift(ratio=2) self.assertEqual(self.item.start, {'minutes': 2}) self.assertEqual(self.item.end, {'minutes': 2, 'seconds': 40}) self.assertEqual(self.item.duration, (0, 0, 40, 0)) self.assertEqual(self.item.characters_per_second, 0.325)
class TestTagRemoval(unittest.TestCase): def setUp(self): self.item = WebVTTItem(1, text="Hello world !") self.item.shift(minutes=1) self.item.end.shift(seconds=20) def test_italics_tag(self): self.item.text = "<i>Hello world !</i>" self.assertEqual(self.item.text_without_tags,'Hello world !') def test_bold_tag(self): self.item.text = "<b>Hello world !</b>" self.assertEqual(self.item.text_without_tags,'Hello world !') def test_underline_tag(self): self.item.text = "<u>Hello world !</u>" self.assertEqual(self.item.text_without_tags,'Hello world !') def test_color_tag(self): self.item.text = '<font color="#ff0000">Hello world !</font>' self.assertEqual(self.item.text_without_tags,'Hello world !') def test_all_tags(self): self.item.text = '<b>Bold</b>, <i>italic</i>, <u>underlined</u>\n' + \ '<font color="#ff0000">red text</font>' + \ ', <b>one,<i> two,<u> three</u></i></b>.' self.assertEqual(self.item.text_without_tags,'Bold, italic, underlined' + \ '\nred text, one, two, three.')
class TestTagRemoval(TestCase): def setUp(self): self.item = WebVTTItem(1, text='Hello world !') self.item.shift(minutes=1) self.item.end.shift(seconds=20) def test_italics_tag(self): self.item.text = '<i>Hello world !</i>' self.assertEqual(self.item.text_without_tags, 'Hello world !') def test_bold_tag(self): self.item.text = '<b>Hello world !</b>' self.assertEqual(self.item.text_without_tags, 'Hello world !') def test_underline_tag(self): self.item.text = '<u>Hello world !</u>' self.assertEqual(self.item.text_without_tags, 'Hello world !') def test_color_tag(self): self.item.text = '<font color="#ff0000">Hello world !</font>' self.assertEqual(self.item.text_without_tags, 'Hello world !') def test_tag_as_text(self): self.item.text = '<This is not a tag>' self.assertEqual(self.item.text_without_tags, 'This is not a tag') def test_split_tag_as_text(self): self.item.text = '<This is\n not a tag>' self.assertEqual(self.item.text_without_tags, 'This is\n not a tag') def test_all_tags(self): self.item.text = '<b>Bold</b>, <i>italic</i>, <u>underlined</u>\n' + \ '<font color="#ff0000">red text</font>' + \ ', <b>one,<i> two,<u> three</u></i></b>.' self.assertEqual( self.item.text_without_tags, 'Bold, italic, underlined\nred text, one, two, three.') def test_bracket_tag(self): self.item.text = '[b]Hello world ![/b]' self.assertEqual(self.item.text_without_brackets, 'Hello world !') def test_key_tag(self): self.item.text = '{b}Hello world !{/b}' self.assertEqual(self.item.text_without_keys, 'Hello world !') def test_replacements(self): self.item.text = 'P & G, A + B' self.assertEqual( self.item.text_with_replacements([('&', 'and'), ('+', 'plus')]), 'P and G, A plus B') def test_regex_eplacements(self): self.item.text = '\\tag21 This is a test!' self.assertEqual( self.item.text_with_replacements([(compile(r'\\tag\d+ '), '')]), 'This is a test!')
class TestOperators(unittest.TestCase): def setUp(self): self.item = WebVTTItem(1, text="Hello world !") self.item.shift(minutes=1) self.item.end.shift(seconds=20) def test_cmp(self): self.assertEqual(self.item, self.item)
class TestDuration(unittest.TestCase): def setUp(self): self.item = WebVTTItem(1, text="Hello world !") self.item.shift(minutes=1) self.item.end.shift(seconds=20) def test_duration(self): self.assertEqual(self.item.duration, (0, 0, 20, 0))
class TestCPS(unittest.TestCase): def setUp(self): self.item = WebVTTItem(1, text="Hello world !") self.item.shift(minutes=1) self.item.end.shift(seconds=20) def test_characters_per_second(self): self.assertEqual(self.item.characters_per_second, 0.65) def test_text_change(self): self.item.text = "Hello world !\nHello world again !" self.assertEqual(self.item.characters_per_second, 1.6) def test_zero_duration(self): self.item.start.shift(seconds = 20) self.assertEqual(self.item.characters_per_second, 0.0) def test_tags(self): self.item.text = '<b>bold</b>, <i>italic</i>, <u>underlined</u>\n' + \ '<font color="#ff0000">red text</font>' + \ ', <b>one,<i> two,<u> three</u></i></b>' self.assertEqual(self.item.characters_per_second, 2.45)
def setUp(self): self.item = WebVTTItem(1, text="Hello world !") self.item.shift(minutes=1) self.item.end.shift(seconds=20) self.string = '1\n00:01:00.000 --> 00:01:20.000\nHello world !\n' self.bad_string = 'foobar' self.coordinates = ('1\n00:01:00.000 --> 00:01:20.000 X1:000 X2:000 ' 'Y1:050 Y2:100\nHello world !\n') self.vtt = ('00:01:00.000 --> 00:01:20.000 D:vertical A:start ' 'L:12%\nHello world !\n') self.string_index = 'foo\n00:01:00.000 --> 00:01:20.000\nHello !\n' self.dots = '1\n00:01:00.000 --> 00:01:20.000\nHello world !\n' self.no_index = '00:01:00,000 --> 00:01:20,000\nHello world !\n' self.junk_after_timestamp = ('1\n00:01:00,000 --> 00:01:20,000?\n' 'Hello world !\n')
def test_multiple_item(self): vtt_file = WebVTTFile([ WebVTTItem(1, {'seconds': 0}, {'seconds': 3}, 'Hello'), WebVTTItem(1, {'seconds': 1}, {'seconds': 2}, 'World !') ]) self.assertEquals(vtt_file.text, 'Hello\nWorld !')
def test_from_string(self): self.assertEqual(WebVTTItem.from_string(self.string), self.item) self.assertRaises(InvalidItem, WebVTTItem.from_string, self.bad_string)
def test_no_index(self): item = WebVTTItem.from_string(self.no_index) self.assertEquals(item.index, None) self.assertEquals(item.text, 'Hello world !')
def test_junk_after_timestamp(self): item = WebVTTItem.from_string(self.junk_after_timestamp) self.assertEquals(item, self.item)
def test_dots(self): self.assertEqual(WebVTTItem.from_string(self.dots), self.item)
def test_string_index(self): item = WebVTTItem.from_string(self.string_index) self.assertEquals(item.index, 'foo') self.assertEquals(item.text, 'Hello !')
def test_vtt_positioning(self): vtt = WebVTTItem.from_string(self.vtt) self.assertEqual(vtt.position, 'D:vertical A:start L:12%') self.assertEqual(vtt.index, None) self.assertEqual(vtt.text, 'Hello world !')
def test_idempotence(self): vtt = WebVTTItem.from_string(self.vtt) self.assertEqual(str(vtt), self.vtt) item = WebVTTItem.from_string(self.coordinates) self.assertEqual(str(item), self.coordinates[2:])
def test_single_item(self): vtt_file = WebVTTFile( [WebVTTItem(1, {'seconds': 1}, {'seconds': 2}, 'Hello')]) self.assertEquals(vtt_file.text, 'Hello')
def test_coordinates(self): item = WebVTTItem.from_string(self.coordinates) self.assertEqual(item, self.item) self.assertEqual(item.position, 'X1:000 X2:000 Y1:050 Y2:100')
def setUp(self): self.item = WebVTTItem()
class TestSerialAndParsing(unittest.TestCase): def setUp(self): self.item = WebVTTItem(1, text="Hello world !") self.item.shift(minutes=1) self.item.end.shift(seconds=20) self.string = '1\n00:01:00.000 --> 00:01:20.000\nHello world !\n' self.bad_string = 'foobar' self.coordinates = ('1\n00:01:00.000 --> 00:01:20.000 X1:000 X2:000 ' 'Y1:050 Y2:100\nHello world !\n') self.vtt = ('00:01:00.000 --> 00:01:20.000 D:vertical A:start ' 'L:12%\nHello world !\n') self.string_index = 'foo\n00:01:00.000 --> 00:01:20.000\nHello !\n' self.dots = '1\n00:01:00.000 --> 00:01:20.000\nHello world !\n' self.no_index = '00:01:00,000 --> 00:01:20,000\nHello world !\n' self.junk_after_timestamp = ('1\n00:01:00,000 --> 00:01:20,000?\n' 'Hello world !\n') def test_serialization(self): self.assertEqual(str(self.item), self.string[2:]) def test_from_string(self): self.assertEqual(WebVTTItem.from_string(self.string), self.item) self.assertRaises(InvalidItem, WebVTTItem.from_string, self.bad_string) def test_coordinates(self): item = WebVTTItem.from_string(self.coordinates) self.assertEqual(item, self.item) self.assertEqual(item.position, 'X1:000 X2:000 Y1:050 Y2:100') def test_vtt_positioning(self): vtt = WebVTTItem.from_string(self.vtt) self.assertEqual(vtt.position, 'D:vertical A:start L:12%') self.assertEqual(vtt.index, None) self.assertEqual(vtt.text, 'Hello world !') def test_idempotence(self): vtt = WebVTTItem.from_string(self.vtt) self.assertEqual(str(vtt), self.vtt) item = WebVTTItem.from_string(self.coordinates) self.assertEqual(str(item), self.coordinates[2:]) def test_dots(self): self.assertEqual(WebVTTItem.from_string(self.dots), self.item) # Bug reported in https://github.com/byroot/pysrt/issues/16 def test_paring_error(self): self.assertRaises(InvalidItem, WebVTTItem.from_string, '1\n' '00:01:00,000 -> 00:01:20,000 X1:000 X2:000 ' 'Y1:050 Y2:100\nHello world !\n') def test_string_index(self): item = WebVTTItem.from_string(self.string_index) self.assertEquals(item.index, 'foo') self.assertEquals(item.text, 'Hello !') def test_no_index(self): item = WebVTTItem.from_string(self.no_index) self.assertEquals(item.index, None) self.assertEquals(item.text, 'Hello world !') def test_junk_after_timestamp(self): item = WebVTTItem.from_string(self.junk_after_timestamp) self.assertEquals(item, self.item)
def setUp(self): self.item = WebVTTItem(1, text='Hello world!') self.item.shift(minutes=1) self.item.end.shift(seconds=20)
class TestSerialAndParsing(TestCase): def setUp(self): self.item = WebVTTItem(1, text="Hello world !") self.item.shift(minutes=1) self.item.end.shift(seconds=20) self.string = '1\n00:01:00.000 --> 00:01:20.000\nHello world !\n' self.bad_string = 'foobar' self.coordinates = ('1\n00:01:00.000 --> 00:01:20.000 X1:000 X2:000 ' 'Y1:050 Y2:100\nHello world !\n') self.vtt = ('00:01:00.000 --> 00:01:20.000 D:vertical A:start ' 'L:12%\nHello world !\n') self.string_index = 'foo\n00:01:00.000 --> 00:01:20.000\nHello !\n' self.dots = '1\n00:01:00.000 --> 00:01:20.000\nHello world !\n' self.no_index = '00:01:00,000 --> 00:01:20,000\nHello world !\n' self.junk_after_timestamp = ('1\n00:01:00,000 --> 00:01:20,000?\n' 'Hello world !\n') def test_serialization(self): self.assertEqual(str(self.item), self.string[2:]) def test_from_string(self): self.assertEqual(WebVTTItem.from_string(self.string), self.item) self.assertRaises(InvalidItem, WebVTTItem.from_string, self.bad_string) def test_coordinates(self): item = WebVTTItem.from_string(self.coordinates) self.assertEqual(item, self.item) self.assertEqual(item.position, 'X1:000 X2:000 Y1:050 Y2:100') def test_vtt_positioning(self): vtt = WebVTTItem.from_string(self.vtt) self.assertEqual(vtt.position, 'D:vertical A:start L:12%') self.assertEqual(vtt.index, None) self.assertEqual(vtt.text, 'Hello world !') def test_idempotence(self): vtt = WebVTTItem.from_string(self.vtt) self.assertEqual(str(vtt), self.vtt) item = WebVTTItem.from_string(self.coordinates) self.assertEqual(str(item), self.coordinates[2:]) def test_dots(self): self.assertEqual(WebVTTItem.from_string(self.dots), self.item) # Bug reported in https://github.com/byroot/pysrt/issues/16 def test_paring_error(self): self.assertRaises( InvalidItem, WebVTTItem.from_string, '1\n' '00:01:00,000 -> 00:01:20,000 X1:000 X2:000 ' 'Y1:050 Y2:100\nHello world !\n') def test_string_index(self): item = WebVTTItem.from_string(self.string_index) self.assertEqual(item.index, 'foo') self.assertEqual(item.text, 'Hello !') def test_no_index(self): item = WebVTTItem.from_string(self.no_index) self.assertEqual(item.index, None) self.assertEqual(item.text, 'Hello world !') def test_junk_after_timestamp(self): item = WebVTTItem.from_string(self.junk_after_timestamp) self.assertEqual(item, self.item) def test_is_unicode(self): # This tests makes no sense for Py3 if is_py3: return # Defining non_unicode encodings list as defined in # https://docs.python.org/2/library/codecs.html#standard-encodings non_unicode_encodings = [ 'big5', 'big5hkscs', 'cp1250', 'cp1251', 'cp1252', 'cp1253', 'cp1254', 'cp1255', 'cp1256', 'cp1257', 'cp1258', 'euc_jp', 'euc_jis_2004', 'euc_jisx0213', 'euc_kr', 'gb2312', 'gbk', 'gb18030', 'hz', 'iso2022_jp', 'iso2022_jp_1', 'iso2022_jp_2', 'iso2022_jp_2004', 'iso2022_jp_3', 'iso2022_jp_ext', 'iso2022_kr', 'latin_1', 'iso8859_2', 'iso8859_3', 'iso8859_4', 'iso8859_5', 'iso8859_6', 'iso8859_7', 'iso8859_8', 'iso8859_9', 'iso8859_10', 'iso8859_11', 'iso8859_13', 'iso8859_13', 'iso8859_14', 'iso8859_15', 'iso8859_16', 'johab', 'koi8_r', 'koi8_u', 'mac_cyrillic', 'mac_greek', 'mac_iceland', 'mac_latin2', 'mac_roman', 'mac_turkish', 'ptcp154', 'shift_jis', 'shift_jis_2004', 'shift_jisx0213'] for enc in non_unicode_encodings: non_unicode = cencode('non_unicode\n', enc) self.no_unicode_item = WebVTTItem(1, text=non_unicode) self.assertRaises(NotImplementedError, WebVTTItem.__str__, self.no_unicode_item)
def setUp(self): self.item = WebVTTItem(1, text="Hello world !") self.item.shift(minutes=1) self.item.end.shift(seconds=20)
def test_shift(self): vtt_file = WebVTTFile([WebVTTItem()]) vtt_file.shift(1, 1, 1, 1) self.assertEqual(vtt_file[0].end, (1, 1, 1, 1)) vtt_file.shift(ratio=2) self.assertEqual(vtt_file[0].end, (2, 2, 2, 2))