Update test_util.py

refactor the code
adipai · Oct 17, 2023 · a411a44 · a411a44
1 parent c6e2d86
commit a411a44
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/test/test_util.py b/test/test_util.py
@@ -5,10 +5,12 @@
 import sys
 import unittest
 import warnings
-sys.path.append(str(Path(__file__).resolve().parents[1]))
+from pathlib import Path
 import pandas as pd
+sys.path.append(str(Path(__file__).resolve().parents[1]))
 #pylint: disable=wrong-import-position
-from src.recommenderapp.utils import create_colored_tags, beautify_feedback_data, create_movie_genres, send_email_to_user
+from src.recommenderapp.utils import create_colored_tags, \
+    beautify_feedback_data, create_movie_genres, send_email_to_user
 #pylint: enable=wrong-import-position
 
 warnings.filterwarnings("ignore")
@@ -35,15 +37,17 @@ 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)
 
     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)