86 lines
2.8 KiB
Python
86 lines
2.8 KiB
Python
from unittest import TestCase
|
|
|
|
import json
|
|
|
|
import pprint
|
|
|
|
from youtube_transcript_api.formatters import (
|
|
Formatter,
|
|
JSONFormatter,
|
|
TextFormatter,
|
|
WebVTTFormatter,
|
|
PrettyPrintFormatter, FormatterLoader
|
|
)
|
|
|
|
|
|
class TestFormatters(TestCase):
|
|
def setUp(self):
|
|
self.transcript = [
|
|
{'text': 'Test line 1', 'start': 0.0, 'duration': 1.50},
|
|
{'text': 'line between', 'start': 1.5, 'duration': 2.0},
|
|
{'text': 'testing the end line', 'start': 2.5, 'duration': 3.25}
|
|
]
|
|
|
|
def test_base_formatter_valid_type(self):
|
|
with self.assertRaises(TypeError) as err:
|
|
Formatter({"test": []})
|
|
expected_err = "'transcript' must be of type: List"
|
|
self.assertEqual(expected_err, str(err.exception))
|
|
|
|
def test_base_formatter_format_call(self):
|
|
with self.assertRaises(NotImplementedError) as err:
|
|
Formatter(self.transcript).format()
|
|
|
|
expected_err = "A subclass of Formatter must implement their own " \
|
|
".format() method."
|
|
self.assertEqual(expected_err, str(err.exception))
|
|
|
|
def test_webvtt_formatter_starting(self):
|
|
content = WebVTTFormatter(self.transcript).format()
|
|
lines = content.split('\n')
|
|
|
|
# test starting lines
|
|
self.assertEqual(lines[0], "WEBVTT")
|
|
self.assertEqual(lines[1], "")
|
|
|
|
def test_webvtt_formatter_ending(self):
|
|
content = WebVTTFormatter(self.transcript).format()
|
|
lines = content.split('\n')
|
|
|
|
# test ending lines
|
|
self.assertEqual(lines[-2], self.transcript[-1]['text'])
|
|
self.assertEqual(lines[-1], "")
|
|
|
|
def test_pretty_print_formatter(self):
|
|
content = PrettyPrintFormatter(self.transcript).format()
|
|
|
|
self.assertEqual(content, pprint.pformat(self.transcript))
|
|
|
|
def test_json_formatter(self):
|
|
content = JSONFormatter(self.transcript).format()
|
|
|
|
self.assertEqual(json.loads(content), self.transcript)
|
|
|
|
def test_text_formatter(self):
|
|
content = TextFormatter(self.transcript).format()
|
|
lines = content.split('\n')
|
|
|
|
self.assertEqual(lines[0], self.transcript[0]["text"])
|
|
self.assertEqual(lines[-1], self.transcript[-1]["text"])
|
|
|
|
def test_formatter_loader(self):
|
|
loader = FormatterLoader('json')
|
|
formatter = loader.load(self.transcript)
|
|
|
|
self.assertTrue(isinstance(formatter, JSONFormatter))
|
|
|
|
def test_formatter_loader__default_formatter(self):
|
|
loader = FormatterLoader()
|
|
formatter = loader.load(self.transcript)
|
|
|
|
self.assertTrue(isinstance(formatter, PrettyPrintFormatter))
|
|
|
|
def test_formatter_loader__unknown_format(self):
|
|
with self.assertRaises(FormatterLoader.UnknownFormatterType):
|
|
FormatterLoader('png')
|