From e884646c168e04845b80d2a3cf87abe9b4eb70f4 Mon Sep 17 00:00:00 2001
From: Maja <majam1616@gmail.com>
Date: Tue, 26 Apr 2022 00:35:01 -0400
Subject: [PATCH] added error message to assert statement

---
 youtube_transcript_api/_api.py          |  3 +++
 youtube_transcript_api/test/test_api.py | 19 +++++++++++++++++++
 2 files changed, 22 insertions(+)

diff --git a/youtube_transcript_api/_api.py b/youtube_transcript_api/_api.py
index a618b0e..dfb790d 100644
--- a/youtube_transcript_api/_api.py
+++ b/youtube_transcript_api/_api.py
@@ -92,6 +92,8 @@ class YouTubeTranscriptApi(object):
         video ids, which could not be retrieved
         :rtype ({str: [{'text': str, 'start': float, 'end': float}]}, [str]}):
         """
+        assert isinstance(video_ids, list), "`video_ids` must be a list of strings"
+
         data = {}
         unretrievable_videos = []
 
@@ -126,6 +128,7 @@ class YouTubeTranscriptApi(object):
         :return: a list of dictionaries containing the 'text', 'start' and 'duration' keys
         :rtype [{'text': str, 'start': float, 'end': float}]:
         """
+        assert isinstance(video_id, str), "`video_id` must be a string"
         return cls.list_transcripts(video_id, proxies, cookies).find_transcript(languages).fetch()
     
     @classmethod
diff --git a/youtube_transcript_api/test/test_api.py b/youtube_transcript_api/test/test_api.py
index ef23f9b..e356c72 100644
--- a/youtube_transcript_api/test/test_api.py
+++ b/youtube_transcript_api/test/test_api.py
@@ -1,3 +1,4 @@
+from ast import Assert
 from unittest import TestCase
 from mock import patch
 
@@ -254,6 +255,24 @@ class TestYouTubeTranscriptApi(TestCase):
                 {'text': 'just something shorter, I made up for testing', 'start': 5.7, 'duration': 3.239}
             ]
         )
+    
+    def test_get_transcript__assertionerror_if_input_not_string(self):
+        """
+        Raise Assertion error if not expected type
+        """
+        with self.assertRaises(AssertionError):
+            YouTubeTranscriptApi.get_transcript(['video_id_1', 'video_id_2'])
+            # YouTubeTranscriptApi.get_transcripts(['video_id_1', 'video_id_2'])
+    
+    def test_get_transcripts__assertionerror_if_input_not_list(self):
+        """
+        Raise Assertion error if not expected type
+        """
+        with self.assertRaises(AssertionError):
+            YouTubeTranscriptApi.get_transcripts('video_id_1')
+            # YouTubeTranscriptApi.get_transcripts(['video_id_1', 'video_id_2'])
+
+
 
     @patch('youtube_transcript_api.YouTubeTranscriptApi.get_transcript')
     def test_get_transcripts(self, mock_get_transcript):