Update: Refactored the code

tests/test_transcript_generator.py

@@ -0,0 +1,19 @@
+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()