From 9ee20a529348a5166bfedba23ccb45dec1fabf25 Mon Sep 17 00:00:00 2001 From: cloudaice Date: Sat, 11 May 2013 22:50:30 +0200 Subject: [PATCH] add generator test --- test/jieba_test.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/test/jieba_test.py b/test/jieba_test.py index 27edcb88..6a413a33 100644 --- a/test/jieba_test.py +++ b/test/jieba_test.py @@ -1,8 +1,9 @@ #-*-coding: utf-8 -*- import sys sys.path.append("../") -import jieba import unittest +import types +import jieba jieba.initialize() @@ -103,6 +104,7 @@ def tearDown(self): def testDefaultCut(self): for content in test_contents: result = jieba.cut(content) + assert isinstance(result, types.GeneratorType), "Test DefaultCut Generator error" result = list(result) assert isinstance(result, list), "Test DefaultCut error on content: %s" % content print >> sys.stderr, " , ".join(result) @@ -110,6 +112,7 @@ def testDefaultCut(self): def testCutAll(self): for content in test_contents: result = jieba.cut(content, cut_all=True) + assert isinstance(result, types.GeneratorType), "Test CutAll Generator error" result = list(result) assert isinstance(result, list), "Test CutAll error on content: %s" % content print >> sys.stderr, " , ".join(result) @@ -118,6 +121,7 @@ def testSetDictionary(self): jieba.set_dictionary("foobar.txt") for content in test_contents: result = jieba.cut(content) + assert isinstance(result, types.GeneratorType), "Test SetDictionary Generator error" result = list(result) assert isinstance(result, list), "Test SetDictionary error on content: %s" % content print >> sys.stderr, " , ".join(result) @@ -125,6 +129,7 @@ def testSetDictionary(self): def testCutForSearch(self): for content in test_contents: result = jieba.cut_for_search(content) + assert isinstance(result, types.GeneratorType), "Test CutForSearch Generator error" result = list(result) assert isinstance(result, list), "Test CutForSearch error on content: %s" % content print >> sys.stderr, " , ".join(result) @@ -133,6 +138,7 @@ def testPosseg(self): import jieba.posseg as pseg for content in test_contents: result = pseg.cut(content) + assert isinstance(result, types.GeneratorType), "Test Posseg Generator error" result = list(result) assert isinstance(result, list), "Test Posseg error on content: %s" % content print >> sys.stderr, " , ".join([w.word + " / " + w.flag for w in result])