From 88bab9e77caa151ddaa8bd5ff16d11b420c0d9b2 Mon Sep 17 00:00:00 2001 From: Liam Sy Date: Tue, 25 Oct 2022 10:58:44 -0400 Subject: [PATCH] added more tests for SRTFormatter --- youtube_transcript_api/test/test_formatters.py | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) diff --git a/youtube_transcript_api/test/test_formatters.py b/youtube_transcript_api/test/test_formatters.py index ce7d869..321d74d 100644 --- a/youtube_transcript_api/test/test_formatters.py +++ b/youtube_transcript_api/test/test_formatters.py @@ -29,13 +29,26 @@ class TestFormatters(TestCase): with self.assertRaises(NotImplementedError): Formatter().format_transcripts([self.transcript]) - def test_srt_formatter(self): + def test_srt_formatter_starting(self): content = SRTFormatter().format_transcript(self.transcript) lines = content.split('\n') # test starting lines self.assertEqual(lines[0], "1") self.assertEqual(lines[1], "00:00:00,000 --> 00:00:01,500") + + def test_srt_formatter_middle(self): + content = SRTFormatter().format_transcript(self.transcript) + lines = content.split('\n') + + #test middle lines + self.assertEqual(lines[4], "2") + self.assertEqual(lines[5], "00:00:01,500 --> 00:00:02,500") + self.assertEqual(lines[6], self.transcript[1]['text']) + + def test_srt_formatter_ending(self): + content = SRTFormatter().format_transcript(self.transcript) + lines = content.split('\n') # test end lines self.assertEqual(lines[-2], self.transcript[-1]['text'])