From 7395646d16af3bf1ab73f6678ce2d4bce57a2094 Mon Sep 17 00:00:00 2001
From: Sumin Byeon <suminb@gmail.com>
Date: Wed, 30 Oct 2019 23:56:41 +0900
Subject: [PATCH] Add a test case to cover combination mode

---
 tests/test_basic.py | 35 +++++++++++++++++++++++++++--------
 1 file changed, 27 insertions(+), 8 deletions(-)

diff --git a/tests/test_basic.py b/tests/test_basic.py
index 2f261dc..3132d80 100644
--- a/tests/test_basic.py
+++ b/tests/test_basic.py
@@ -47,11 +47,30 @@ def test_split_hanja():
     assert corpora[3] == u"이다."
 
 
-def test_translate():
-    assert hanja.translate(u"韓國語", mode="substitution") == u"한국어"
-    assert hanja.translate(u"한국어", mode="substitution") == u"한국어"
-    assert hanja.translate(u"利用해", mode="substitution") == u"이용해"
-    assert hanja.translate(u"連結된", mode="substitution") == u"연결된"
-    assert hanja.translate(u"1800年에", mode="substitution") == u"1800년에"
-    assert hanja.translate(u"그레고리曆", mode="substitution") == u"그레고리력"
-    assert hanja.translate(u"系列", mode="substitution") == u"계열"
+def test_translate_substitution_mode():
+    mode = "substitution"
+    assert hanja.translate(u"韓國語", mode=mode) == u"한국어"
+    assert hanja.translate(u"한국어", mode=mode) == u"한국어"
+    assert hanja.translate(u"利用해", mode=mode) == u"이용해"
+    assert hanja.translate(u"連結된", mode=mode) == u"연결된"
+    assert hanja.translate(u"1800年에", mode=mode) == u"1800년에"
+    assert hanja.translate(u"그레고리曆", mode=mode) == u"그레고리력"
+    assert hanja.translate(u"系列", mode=mode) == u"계열"
+
+
+def test_translate_combination_mode():
+    mode = "combination"
+    assert (
+        hanja.translate("韓國語", mode=mode)
+        == '<span class="hanja">韓國語</span><span class="hangul">(한국어)</span>'
+    )
+    assert (
+        hanja.translate("利用해", mode=mode)
+        == '<span class="hanja">利用</span><span class="hangul">(이용)</span>해'
+    )
+    assert (
+        hanja.translate("大韓民國은 民主共和國이다.", mode=mode)
+        == '<span class="hanja">大韓民國</span><span class="hangul">(대한민국)'
+        '</span>은 <span class="hanja">民主共和國</span><span class="hangul">'
+        "(민주공화국)</span>이다."
+    )