Fix tests for tags generation

Author: fspv

leetcode_anki/helpers/leetcode.py

@@ -83,7 +83,9 @@ class LeetcodeData:
     names.
     """
 
-    def __init__(self, start: int, stop: int, page_size: int = 1000, list_id: str = "") -> None:
+    def __init__(
+        self, start: int, stop: int, page_size: int = 1000, list_id: str = ""
+    ) -> None:
         """
         Initialize leetcode API and disk cache for API responses
         """
@@ -354,7 +356,7 @@ async def tags(self, problem_slug: str) -> List[str]:
         """
         data = self._get_problem_data(problem_slug)
         tags = list(map(lambda x: x.slug, data.topic_tags))
-        tags.append(data.difficulty)
+        tags.append(f"difficulty-{data.difficulty.lower()}-tag")
         return tags
 
     async def freq_bar(self, problem_slug: str) -> float:

test/helpers/test_leetcode.py

@@ -245,7 +245,10 @@ async def test_dislikes(self) -> None:
     async def test_tags(self) -> None:
         self._leetcode_data._cache["test"] = QUESTION_DETAIL
 
-        assert (await self._leetcode_data.tags("test")) == ["test-tag"]
+        assert (await self._leetcode_data.tags("test")) == [
+            "test-tag",
+            "difficulty-hard-tag",
+        ]
 
     @pytest.mark.asyncio
     @mock.patch(