fixed test

Author: ukr-projects

docs/README.md

@@ -55,6 +55,45 @@
 ## 🗂️ Folder Structure
 
 ```
+yt-transcript-gpt/
+├── .devcontainer/
+│   └── devcontainer.json
+├── .github/
+│   ├── ISSUE_TEMPLATE/
+│   │   ├── bug_report.md
+│   │   └── feature_request.md
+│   ├── PULL_REQUEST_TEMPLATE.md
+│   ├── RELEASE_TEMPLATE.md
+│   └── workflows/
+│       └── ci.yml
+├── .gitignore
+├── assets/
+│   ├── screenshots/
+│   │   └── screenshot.png
+│   └── yt-transcript-gpt-banner.jpg
+├── docs/
+│   ├── CHANGELOG.md
+│   ├── CODE_OF_CONDUCT.md
+│   ├── CONTRIBUTING.md
+│   ├── README.md
+│   ├── SECURITY.md
+│   ├── STATUS.md
+│   └── USAGE.md
+├── LICENSE
+├── pyproject.toml
+├── requirements.txt
+├── src/
+│   ├── app/
+│   │   ├── __init__.py
+│   │   ├── gemini_ai.py
+│   │   ├── main.py
+│   │   ├── transcript_extractor.py
+│   │   ├── ui.py
+│   │   └── utils.py
+│   └── main.py
+└── tests/
+    ├── test_gemini_ai.py
+    └── test_transcript_extractor.py
 
 ```
 

tests/test_gemini_ai.py

@@ -20,9 +20,10 @@ def setUp(self, mock_configure, mock_model):
         self.transcript_text = "This is a test transcript."
 
     def test_initialization_with_key(self):
-        with patch("src.app.gemini_ai.genai.configure") as mock_configure, \
-             patch("src.app.gemini_ai.genai.GenerativeModel") as mock_model:
-            
+        with patch("src.app.gemini_ai.genai.configure") as mock_configure, patch(
+            "src.app.gemini_ai.genai.GenerativeModel"
+        ) as mock_model:
+
             gemini_ai = GeminiAI(api_key="test_key")
             mock_configure.assert_called_once_with(api_key="test_key")
             mock_model.assert_called_once_with("gemini-1.5-flash")
@@ -42,14 +43,14 @@ def test_generate_summary_success(self):
         mock_response = MagicMock()
         mock_response.text = "This is a summary."
         self.mock_model_instance.generate_content.return_value = mock_response
-        
+
         summary = self.gemini_ai.generate_summary(self.transcript_text)
         self.assertEqual(summary, "This is a summary.")
         self.mock_model_instance.generate_content.assert_called_once()
 
     def test_generate_summary_failure(self):
         self.mock_model_instance.generate_content.side_effect = Exception("API Error")
-        
+
         summary = self.gemini_ai.generate_summary(self.transcript_text)
         self.assertIn("Error generating summary", summary)
         self.assertIn("API Error", summary)
@@ -58,90 +59,90 @@ def test_extract_key_quotes_success(self):
         mock_response = MagicMock()
         mock_response.text = "These are key quotes."
         self.mock_model_instance.generate_content.return_value = mock_response
-        
+
         quotes = self.gemini_ai.extract_key_quotes(self.transcript_text)
         self.assertEqual(quotes, "These are key quotes.")
 
     def test_extract_key_quotes_failure(self):
         self.mock_model_instance.generate_content.side_effect = Exception("API Error")
-        
+
         quotes = self.gemini_ai.extract_key_quotes(self.transcript_text)
         self.assertIn("Error extracting quotes", quotes)
 
     def test_create_study_guide_success(self):
         mock_response = MagicMock()
         mock_response.text = "This is a study guide."
         self.mock_model_instance.generate_content.return_value = mock_response
-        
+
         study_guide = self.gemini_ai.create_study_guide(self.transcript_text)
         self.assertEqual(study_guide, "This is a study guide.")
 
     def test_create_study_guide_failure(self):
         self.mock_model_instance.generate_content.side_effect = Exception("API Error")
-        
+
         study_guide = self.gemini_ai.create_study_guide(self.transcript_text)
         self.assertIn("Error creating study guide", study_guide)
 
     def test_generate_qa_success(self):
         mock_response = MagicMock()
         mock_response.text = "Q: Test question?\nA: Test answer."
         self.mock_model_instance.generate_content.return_value = mock_response
-        
+
         qa = self.gemini_ai.generate_qa(self.transcript_text)
         self.assertEqual(qa, "Q: Test question?\nA: Test answer.")
 
     def test_generate_qa_failure(self):
         self.mock_model_instance.generate_content.side_effect = Exception("API Error")
-        
+
         qa = self.gemini_ai.generate_qa(self.transcript_text)
         self.assertIn("Error generating Q&A", qa)
 
     def test_create_flashcards_success(self):
         mock_response = MagicMock()
         mock_response.text = "FRONT: Test term\nBACK: Test definition\n---"
         self.mock_model_instance.generate_content.return_value = mock_response
-        
+
         flashcards = self.gemini_ai.create_flashcards(self.transcript_text)
         self.assertEqual(flashcards, "FRONT: Test term\nBACK: Test definition\n---")
 
     def test_create_flashcards_failure(self):
         self.mock_model_instance.generate_content.side_effect = Exception("API Error")
-        
+
         flashcards = self.gemini_ai.create_flashcards(self.transcript_text)
         self.assertIn("Error creating flashcards", flashcards)
 
     def test_highlight_insights_success(self):
         mock_response = MagicMock()
         mock_response.text = "🔍 Key Insights: Test insights"
         self.mock_model_instance.generate_content.return_value = mock_response
-        
+
         insights = self.gemini_ai.highlight_insights(self.transcript_text)
         self.assertEqual(insights, "🔍 Key Insights: Test insights")
 
     def test_highlight_insights_failure(self):
         self.mock_model_instance.generate_content.side_effect = Exception("API Error")
-        
+
         insights = self.gemini_ai.highlight_insights(self.transcript_text)
         self.assertIn("Error extracting insights", insights)
 
     def test_chat_with_transcript_success(self):
         mock_response = MagicMock()
         mock_response.text = "This is an answer."
         self.mock_model_instance.generate_content.return_value = mock_response
-        
+
         answer = self.gemini_ai.chat_with_transcript(
             self.transcript_text, "What is this?"
         )
         self.assertEqual(answer, "This is an answer.")
 
     def test_chat_with_transcript_failure(self):
         self.mock_model_instance.generate_content.side_effect = Exception("API Error")
-        
+
         answer = self.gemini_ai.chat_with_transcript(
             self.transcript_text, "What is this?"
         )
         self.assertIn("Error in chat", answer)
 
 
 if __name__ == "__main__":
-    unittest.main()
+    unittest.main()

tests/test_transcript_extractor.py

@@ -40,26 +40,32 @@ def test_get_transcript_youtube_api_failure(self, mock_api):
 
     @patch("src.app.transcript_extractor.tempfile.TemporaryDirectory")
     @patch("src.app.transcript_extractor.os.listdir")
-    @patch("builtins.open", new_callable=mock_open, read_data="WEBVTT\n\n00:00:00.000 --> 00:00:01.000\nhello from ytdlp")
+    @patch(
+        "builtins.open",
+        new_callable=mock_open,
+        read_data="WEBVTT\n\n00:00:00.000 --> 00:00:01.000\nhello from ytdlp",
+    )
     @patch("src.app.transcript_extractor.yt_dlp")
-    def test_get_transcript_ytdlp_success(self, mock_ytdlp, mock_file, mock_listdir, mock_tempdir):
+    def test_get_transcript_ytdlp_success(
+        self, mock_ytdlp, mock_file, mock_listdir, mock_tempdir
+    ):
         # Mock the temporary directory
         mock_temp_path = "/tmp/test"
         mock_tempdir.return_value.__enter__.return_value = mock_temp_path
-        
+
         # Mock the yt-dlp instance
         mock_ydl_instance = MagicMock()
         mock_ydl_instance.extract_info.return_value = {
             "title": "Test Video",
             "duration": 60,
             "subtitles": {},
-            "automatic_captions": {}
+            "automatic_captions": {},
         }
         mock_ytdlp.YoutubeDL.return_value.__enter__.return_value = mock_ydl_instance
-        
+
         # Mock file listing to return a VTT file
         mock_listdir.return_value = ["test_video.en.vtt"]
-        
+
         transcript = self.extractor.get_transcript_ytdlp("dQw4w9WgXcQ")
         self.assertIsNotNone(transcript)
         if transcript:
@@ -111,7 +117,7 @@ def test_parse_vtt_content(self):
 
 00:00:01.000 --> 00:00:02.000
 this is a test"""
-        
+
         transcript = self.extractor._parse_vtt_content(vtt_content)
         self.assertEqual(len(transcript), 2)
         self.assertEqual(transcript[0]["text"], "hello world")
@@ -127,7 +133,7 @@ def test_parse_srt_content(self):
 2
 00:00:01,000 --> 00:00:02,000
 this is a test"""
-        
+
         transcript = self.extractor._parse_srt_content(srt_content)
         self.assertEqual(len(transcript), 2)
         self.assertEqual(transcript[0]["text"], "hello world")
@@ -149,4 +155,4 @@ def test_parse_srt_timestamp(self):
 
 
 if __name__ == "__main__":
-    unittest.main()
+    unittest.main()