Fix concordance bug for Python2 and add test

konlpy/utils.py

@@ -65,7 +65,7 @@ def concordance(phrase, text, show=False):
     indexes = [i for i, term in enumerate(terms) if phrase in term]
     if show:
         for i in indexes:
-            print(i, ' '.join(terms[max(0, i-3):i+3]))
+            print('%d\t%s' % (i, ' '.join(terms[max(0, i-3):i+3])))
     return indexes
 
 def concat(phrase):

test/test_utils.py

@@ -10,3 +10,18 @@ def test_utils_pprint(capsys): # Fixture `capsys` allows stdout/stderr captures
         assert out == u"[저는 소프트웨어 관련학과 입니다.]\n"
     else:
         assert out == u"['저는 소프트웨어 관련학과 입니다.']\n"
+
+def test_utils_concordance():
+    from konlpy.corpus import kolaw
+    from konlpy.utils  import concordance
+    doc = kolaw.open('constitution.txt').read()
+    ccd = concordance(u'대한민국', doc, show=True)
+    assert ccd == [0, 9, 98, 100, 110, 126, 133, 147, 787, 1836, 3620]
+
+def test_utils_concordance_show(capsys):
+    from konlpy.corpus import kolaw
+    from konlpy.utils  import concordance
+    doc = kolaw.open('constitution.txt').read()
+    ccd = concordance(u'대한민국', doc, show=True)
+    out, err = capsys.readouterr()
+    assert out == u"0\t대한민국헌법 유구한 역사와\n9\t대한국민은 3·1운동으로 건립된 대한민국임시정부의 법통과 불의에\n98\t총강 제1조 ① 대한민국은 민주공화국이다. ②대한민국의\n100\t① 대한민국은 민주공화국이다. ②대한민국의 주권은 국민에게\n110\t나온다. 제2조 ① 대한민국의 국민이 되는\n126\t의무를 진다. 제3조 대한민국의 영토는 한반도와\n133\t부속도서로 한다. 제4조 대한민국은 통일을 지향하며,\n147\t추진한다. 제5조 ① 대한민국은 국제평화의 유지에\n787\t군무원이 아닌 국민은 대한민국의 영역안에서는 중대한\n1836\t파견 또는 외국군대의 대한민국 영역안에서의 주류에\n3620\t경제 제119조 ① 대한민국의 경제질서는 개인과\n"