add find_common_tags example

Author: KRRT7

src/dsa/various.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import re
 
 
@@ -114,3 +116,15 @@ def find_shortest_path(self, start: str, end: str) -> list[str]:
                     queue.append(new_path)
 
         return []  # No path found
+
+
+def find_common_tags(articles: list[dict[str, list[str]]]) -> set[str]:
+    if not articles:
+        return set()
+
+    common_tags = set(articles[0].get("tags", []))
+    for article in articles[1:]:
+        common_tags.intersection_update(article.get("tags", []))
+        if not common_tags:
+            break
+    return common_tags

tests/test_common_tags.py

@@ -0,0 +1,21 @@
+from src.dsa.various import find_common_tags
+
+def test_common_tags_1() -> None:
+    articles_1 = [
+        {"title": "Article 1", "tags": ["Python", "AI", "ML"]},
+        {"title": "Article 2", "tags": ["Python", "Data Science", "AI"]},
+        {"title": "Article 3", "tags": ["Python", "AI", "Big Data"]},
+    ]
+
+    expected = {"Python", "AI"}
+
+    assert find_common_tags(articles_1) == expected
+
+    articles_2 = [
+        {"title": "Article 1", "tags": ["Python", "AI", "ML"]},
+        {"title": "Article 2", "tags": ["Python", "Data Science", "AI"]},
+        {"title": "Article 3", "tags": ["Python", "AI", "Big Data"]},
+        {"title": "Article 4", "tags": ["Python", "AI", "ML"]},
+    ]
+
+    assert find_common_tags(articles_2) == expected