add support for count unique

carpedm20 · Jan 5, 2022 · 5aebf67 · 5aebf67
1 parent e35fc45
commit 5aebf67
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/emoji/core.py b/emoji/core.py
@@ -293,8 +293,13 @@ def distinct_emoji_lis(string, language=None):
     return distinct_list
 
 
-def emoji_count(string):
-    """Returns the count of emojis in a string."""
+def emoji_count(string, unique=False):
+    """Returns the count of emojis in a string.
+
+    :param unique: (optional) True if count only unique emojis
+    """
+    if unique:
+        return len(distinct_emoji_lis(string))
     return len(emoji_lis(string))
 
 

diff --git a/tests/test_core.py b/tests/test_core.py
@@ -256,6 +256,7 @@ def test_emoji_count():
     assert emoji.emoji_count('Hi, I am fine. 😁') == 1
     assert emoji.emoji_count('Hi') == 0
     assert emoji.emoji_count('Hello 🇫🇷👌') == 2
+    assert emoji.emoji_count('Hello 🇵🇱🍺🇵🇱', unique=True) == 2
 
 
 def test_replace_emoji():