Fixed bug in test suite

adipai · Oct 17, 2023 · a71ebac · a71ebac
1 parent a411a44
commit a71ebac
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/test/test_util.py b/test/test_util.py
@@ -37,8 +37,8 @@ def test_create_colored_tags(self):
         """
         Test case 2
         """
-        expected_result = '<span style="background-color: #FF1493; color: #FFFFFF;             \
-        padding: 5px; border-radius: 5px;">Musical</span>'
+        expected_result = '<span style="background-color: #FF1493; color: #FFFFFF; \
+            padding: 5px; border-radius: 5px;">Musical</span>'
         result = create_colored_tags(['Musical'])
         self.assertTrue(result == expected_result)
 
@@ -49,7 +49,15 @@ def test_create_movie_genres(self):
         expected_result = {'Toy Story (1995)': ['Animation', 'Comedy', 'Family'], \
                            'Jumanji (1995)': [
             'Adventure', 'Fantasy', 'Family']}
-        movie_genre_df = pd.read_csv('../data/movies.csv').head(n=2)
+
+        data = [["862", "Toy Story (1995)", "Animation|Comedy|Family", \
+                 "tt0114709", " ", "/rhIRbceoE9lR4veEXuwCC2wARtG.jpg", "81"], \
+                ["8844", "Jumanji (1995)", "Adventure|Fantasy|Family", "tt0113497", " ", \
+                  "/vzmL6fP7aPKNKPRTFnZmiUfciyV.jpg", "104"]]
+
+        movie_genre_df = pd.DataFrame(data, columns=[
+            'movieId', 'title', 'genres', 'imdb_id', 'overview', 'poster_path', 'runtime'])
+
         result = create_movie_genres(movie_genre_df)
         self.assertTrue(result == expected_result)