From f28c34b696c88b5d83131eaaa962f0e2e8b3090f Mon Sep 17 00:00:00 2001
From: Ayush Morbar <108860844+ayushmorbar@users.noreply.github.com>
Date: Tue, 14 May 2024 06:54:29 +0000
Subject: [PATCH 1/2] Add test files for IMDB module

---
 src/test/imdb_test.py | 61 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 61 insertions(+)
 create mode 100644 src/test/imdb_test.py

diff --git a/src/test/imdb_test.py b/src/test/imdb_test.py
new file mode 100644
index 00000000..a3d3f9f3
--- /dev/null
+++ b/src/test/imdb_test.py
@@ -0,0 +1,61 @@
+import unittest
+from scrape_up import imdb
+
+class IMDBTest(unittest.TestCase):
+    """
+    imdb module test.\n
+    | Method               | Details                                                             |
+    | ---------------------| ------------------------------------------------------------------- |
+    | `.top_rated()`       | Returns the list of top rated movies                                |
+    | `.scrape_genre_movies(genre)` | Returns the list of movies of a specific genre             |
+    | `.top_rated_shows()` | Returns the list of top rated TV shows                              |
+    """
+
+    def setUp(self):
+        self.scraper = imdb()
+
+    def test_top_rated(self):
+        response = self.scraper.top_rated()
+        self.assertGreater(len(response), 0, "Top rated movies list is empty")
+        self.assertIsInstance(response, list, "Top rated movies is not a list")
+        self.assertTrue(
+            all(
+                isinstance(movie, dict) and "title" in movie and "year" in movie and "duration" in movie and "rating" in movie
+                for movie in response
+            ),
+            "Incorrect format for top rated movies",
+        )
+
+    def test_scrape_genre_movies(self):
+        genre = "Adventure"
+        response = self.scraper.scrape_genre_movies(genre)
+
+        self.assertIsInstance(response, list, f"{genre} movies is not a list")
+        self.assertGreater(len(response), 0, f"{genre} movies list is empty")
+
+        for movie in response:
+            self.assertIsInstance(movie, dict)
+            self.assertIn("title", movie)
+            self.assertIn("year", movie)
+            self.assertIn("certificate", movie)
+            self.assertIn("time", movie)
+            self.assertIn("genre", movie)
+            self.assertIn("rating", movie)
+            self.assertIn("simple_desc", movie)
+            self.assertIn("votes", movie)
+
+    def test_top_rated_shows(self):
+        response = self.scraper.top_rated_shows()
+
+        self.assertIsInstance(response, list, "Top rated shows is not a list")
+        self.assertGreater(len(response), 0, "Top rated shows list is empty")
+
+        for show in response:
+            self.assertIsInstance(show, dict)
+            self.assertIn("title", show)
+            self.assertIn("year", show)
+            self.assertIn("episode", show)
+            self.assertIn("rating", show)
+
+if __name__ == '__main__':
+    unittest.main()
\ No newline at end of file

From f7a10e85ffad5e11d4715f54708e9eac5afe0886 Mon Sep 17 00:00:00 2001
From: Nikhil Raj <nikhil25803@gmail.com>
Date: Tue, 14 May 2024 13:56:52 +0530
Subject: [PATCH 2/2] Formatting and check.

---
 src/scrape_up/imdb/imdb.py | 3 ---
 src/test/imdb_test.py      | 4 ++--
 2 files changed, 2 insertions(+), 5 deletions(-)

diff --git a/src/scrape_up/imdb/imdb.py b/src/scrape_up/imdb/imdb.py
index e44c4519..e730bfe5 100644
--- a/src/scrape_up/imdb/imdb.py
+++ b/src/scrape_up/imdb/imdb.py
@@ -90,7 +90,6 @@ def scrape_genre_movies(self, genre):
         genre_data = scraper.scrape_genre_movies(genre)
 
         json_data = json.dumps(genre_data, indent=4)
-        print(json_data)
         ```
         Return\n
         ```python
@@ -103,8 +102,6 @@ def scrape_genre_movies(self, genre):
         try:
             url = "https://www.imdb.com/search/title/?genres={}&sort=user_rating,desc&title_type=feature&num_votes=25000,&pf_rd_m=A2FGELUUNOQJNL&pf_rd_p=5aab685f-35eb-40f3-95f7-c53f09d542c3&pf_rd_r=N97GEQS6R7J9EV7V770D&pf_rd_s=right-6&pf_rd_t=15506&pf_rd_i=top&ref_=chttp_gnr_16"
             formatted_url = url.format(genre)
-            print(formatted_url)
-
             resp = requests.get(formatted_url, headers=self.headers)
             content = BeautifulSoup(resp.content, "lxml")
             genres = [
diff --git a/src/test/imdb_test.py b/src/test/imdb_test.py
index a3d3f9f3..a2933931 100644
--- a/src/test/imdb_test.py
+++ b/src/test/imdb_test.py
@@ -1,5 +1,5 @@
 import unittest
-from scrape_up import imdb
+from scrape_up.imdb import IMDB
 
 class IMDBTest(unittest.TestCase):
     """
@@ -12,7 +12,7 @@ class IMDBTest(unittest.TestCase):
     """
 
     def setUp(self):
-        self.scraper = imdb()
+        self.scraper = IMDB()
 
     def test_top_rated(self):
         response = self.scraper.top_rated()