import unittest

from src.transcript_generator import TranscriptGenerator


class TranscriptGeneratorTest(unittest.TestCase):
    def setUp(self):
        self.transcript_generator = TranscriptGenerator()

    def test_generate(self):
        input_sources = [
            'https://www.youtube.com/watch?v=BVVo1BaL6JA',
            '/path/to/your/local/file.mp3'
        ]
        self.transcript_generator.generate(input_sources)


if __name__ == "__main__":
    unittest.main()