KeyArgo
/
youtube-transcript-api
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

added more tests for SRTFormatter

This commit is contained in:
Liam Sy 2022-10-25 10:58:44 -04:00
parent a78f493386
commit 88bab9e77c
1 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
youtube_transcript_api/test/test_formatters.py
View File

@ -29,13 +29,26 @@ class TestFormatters(TestCase):
with self.assertRaises(NotImplementedError): with self.assertRaises(NotImplementedError):
Formatter().format_transcripts([self.transcript]) Formatter().format_transcripts([self.transcript])
def test_srt_formatter(self): def test_srt_formatter_starting(self):
content = SRTFormatter().format_transcript(self.transcript) content = SRTFormatter().format_transcript(self.transcript)
lines = content.split('\n') lines = content.split('\n')
# test starting lines # test starting lines
self.assertEqual(lines[0], "1") self.assertEqual(lines[0], "1")
self.assertEqual(lines[1], "00:00:00,000 --> 00:00:01,500") 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 # test end lines
self.assertEqual(lines[-2], self.transcript[-1]['text']) self.assertEqual(lines[-2], self.transcript[-1]['text'])