diff --git a/test/test_util.py b/test/test_util.py index 9e3306a4..c62e471b 100644 --- a/test/test_util.py +++ b/test/test_util.py @@ -46,7 +46,8 @@ def test_create_movie_genres(self): """ Test case 3 """ - expected_result = {'Toy Story (1995)': ['Animation', 'Comedy', 'Family'], 'Jumanji (1995)': [ + 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) result = create_movie_genres(movie_genre_df)