class TestSRTReader(unittest.TestCase): def setUp(self): test_content = StringIO(u"""1 00:00:03,490 --> 00:00:07,430 FISHER: All right. So, let's begin. This session is: Going Social 00:00:07,430 --> 00:00:11,600 with the YouTube APIs. I am Jeff Fisher, 2 00:00:11,600 --> 00:00:14,009 and this is Johann Hartmann, we're presenting today. 3 00:00:14,009 --> 00:00:15,889 [pause] """) self.reader = Reader(test_content) def test_read(self): captions = self.reader.read() self.assertTrue(captions is not None) self.assertEqual(len(captions), 4) first = captions[0] self.assertEqual(type(first.text), unicode) self.assertEqual(first.text, u"FISHER: All right. So, let's begin.\nThis session is: Going Social\n") self.assertEqual(first.start, core.get_date(second=3, millisecond=490)) self.assertEqual(first.end, core.get_date(second=7, millisecond=430)) self.assertEqual(first.duration, timedelta(seconds=3, milliseconds=940))
def setUp(self): test_content = StringIO(u"""1 00:00:03,490 --> 00:00:07,430 FISHER: All right. So, let's begin. This session is: Going Social 00:00:07,430 --> 00:00:11,600 with the YouTube APIs. I am Jeff Fisher, 2 00:00:11,600 --> 00:00:14,009 and this is Johann Hartmann, we're presenting today. 3 00:00:14,009 --> 00:00:15,889 [pause] """) self.reader = Reader(test_content)
class TestSRTWriter(unittest.TestCase): def setUp(self): test_content = StringIO(u"""1 00:00:03,490 --> 00:00:07,430 FISHER: All right. So, let's begin. This session is: Going Social 00:00:07,430 --> 00:00:11,600 with the YouTube APIs. I am Jeff Fisher, 2 00:00:11,600 --> 00:00:14,009 and this is Johann Hartmann, we're presenting today. 3 00:00:14,009 --> 00:00:15,889 [pause] """) self.reader = Reader(test_content) self.writer = Writer(StringIO()) def test_transformtext(self): captions = self.reader.read() self.writer.captions = captions text = self.writer.captions_to_text() should_be = u"""0\n00:00:03,490 --> 00:00:07,430\nFISHER: All right. So, let's begin.\nThis session is: Going Social""" self.assertTrue(text.startswith(should_be), "%s !startswith %s" % (text, should_be)) def test_format_time(self): caption = core.Caption() caption.start = core.get_date(second=3, millisecond=490) caption.end = core.get_date(second=7, millisecond=430) time_info = self.writer.format_time(caption) self.assertEqual(time_info['start'], '00:00:03,490') self.assertEqual(time_info['end'], '00:00:07,430')
class TestSRTReader(unittest.TestCase): def setUp(self): test_content = StringIO(u"""1 00:00:03,490 --> 00:00:07,430 FISHER: All right. So, let's begin. This session is: Going Social 00:00:07,430 --> 00:00:11,600 with the YouTube APIs. I am Jeff Fisher, 2 00:00:11,600 --> 00:00:14,009 and this is Johann Hartmann, we're presenting today. 3 00:00:14,009 --> 00:00:15,889 [pause] """) self.reader = Reader(test_content) def test_read(self): captions = self.reader.read() self.assertTrue(captions is not None) self.assertEqual(len(captions), 4) first = captions[0] self.assertEqual(type(first.text), unicode) self.assertEqual( first.text, u"FISHER: All right. So, let's begin.\nThis session is: Going Social\n" ) self.assertEqual(first.start, core.get_date(second=3, millisecond=490)) self.assertEqual(first.end, core.get_date(second=7, millisecond=430)) self.assertEqual(first.duration, timedelta(seconds=3, milliseconds=940))