Test emoji.replace_emoji with function parameter

carpedm20 · Oct 2, 2021 · 5711c67 · 5711c67
1 parent 9983e5e
commit 5711c67
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/tests/test_core.py b/tests/test_core.py
@@ -60,13 +60,13 @@ def test_emojize_variant():
     assert emoji.emojize(':admission_tickets:', variant='text_type') == remove_variant(emoji.EMOJI_UNICODE['en'][':admission_tickets:']) + u'\ufe0e'
     assert emoji.emojize(':admission_tickets:', variant='emoji_type') == remove_variant(emoji.EMOJI_UNICODE['en'][':admission_tickets:']) + u'\ufe0f'
 
-    with pytest.raises(Exception):
+    with pytest.raises(ValueError):
         emoji.emojize(':admission_tickets:', variant=False)
 
-    with pytest.raises(Exception):
+    with pytest.raises(ValueError):
         emoji.emojize(':admission_tickets:', variant=True)
 
-    with pytest.raises(Exception):
+    with pytest.raises(ValueError):
         emoji.emojize(':admission_tickets:', variant='wrong')
 
     assert emoji.emojize(":football:", use_aliases=False) == ':football:'
@@ -149,11 +149,16 @@ def test_emoji_count():
 
 
 def test_replace_emoji():
-    assert emoji.replace_emoji('Hi, I am fine. 😁') == 'Hi, I am fine. '
+    assert emoji.replace_emoji(u'Hi, I am fine. 😁') == 'Hi, I am fine. '
     assert emoji.replace_emoji('Hi') == 'Hi'
     assert emoji.replace_emoji('Hello 🇫🇷👌') == 'Hello '
     assert emoji.replace_emoji('Hello 🇫🇷👌', 'x') == 'Hello xx'
 
+    def replace(emj, data):
+        assert emj in ["🇫🇷", "👌"]
+        return 'x'
+    assert emoji.replace_emoji('Hello 🇫🇷👌', replace) == 'Hello xx'
+
 
 def test_is_emoji():
     assert emoji.is_emoji('😁')