diff --git a/src/PosTagger.hpp b/src/PosTagger.hpp index 63f13c0..b63f1de 100644 --- a/src/PosTagger.hpp +++ b/src/PosTagger.hpp @@ -17,13 +17,16 @@ namespace CppJieba public: PosTagger(){}; - PosTagger(const string& dictPath, const string& hmmFilePath, const string& charStatus, const string& startProb, const string& emitProb, const string& endProb, const string& transProb) + PosTagger( + const string& dictPath, + const string& hmmFilePath + ) { - LIMONP_CHECK(init(dictPath, hmmFilePath, charStatus, startProb, emitProb, endProb, transProb)); + LIMONP_CHECK(init(dictPath, hmmFilePath)); }; ~PosTagger(){}; public: - bool init(const string& dictPath, const string& hmmFilePath, const string& charStatus, const string& startProb, const string& emitProb, const string& endProb, const string& transProb) + bool init(const string& dictPath, const string& hmmFilePath) { LIMONP_CHECK(_dictTrie.init(dictPath)); LIMONP_CHECK(_segment.init(dictPath, hmmFilePath)); diff --git a/test/tagging_demo.cpp b/test/tagging_demo.cpp index 5b341f9..638352a 100644 --- a/test/tagging_demo.cpp +++ b/test/tagging_demo.cpp @@ -3,7 +3,7 @@ using namespace CppJieba; int main(int argc, char ** argv) { - PosTagger tagger("../dict/jieba.dict.utf8", "../dict/hmm_model.utf8", "", "", "", "", ""); + PosTagger tagger("../dict/jieba.dict.utf8", "../dict/hmm_model.utf8"); string s("我是蓝翔技工拖拉机学院手扶拖拉机专业的。不用多久,我就会升职加薪,当上总经理,出任CEO,迎娶白富美,走上人生巅峰。"); vector > res; tagger.tag(s, res); diff --git a/test/unittest/CMakeLists.txt b/test/unittest/CMakeLists.txt index c252213..a6c8de4 100644 --- a/test/unittest/CMakeLists.txt +++ b/test/unittest/CMakeLists.txt @@ -16,7 +16,13 @@ ADD_LIBRARY(gtest STATIC ${GTEST_ROOT_DIR}/src/gtest-all.cc) #ADD_EXECUTABLE(keyword.test gtest_main.cpp TKeywordExtractor.cpp) #TARGET_LINK_LIBRARIES(keyword.test gtest pthread) -ADD_EXECUTABLE(test.run gtest_main.cpp TKeywordExtractor.cpp TTrie.cpp TSegments.cpp ) +ADD_EXECUTABLE(test.run + gtest_main.cpp + TKeywordExtractor.cpp + TTrie.cpp + TSegments.cpp + TPosTagger.cpp +) TARGET_LINK_LIBRARIES(gtest pthread) TARGET_LINK_LIBRARIES(test.run gtest pthread) diff --git a/test/unittest/TPosTagger.cpp b/test/unittest/TPosTagger.cpp new file mode 100644 index 0000000..3f32e7e --- /dev/null +++ b/test/unittest/TPosTagger.cpp @@ -0,0 +1,17 @@ +#include "src/PosTagger.hpp" +#include "gtest/gtest.h" + +using namespace CppJieba; + +const char * const QUERY_TEST1 = "我是蓝翔技工拖拉机学院手扶拖拉机专业的。不用多久,我就会升职加薪,当上总经理,出任CEO,迎娶白富美,走上人生巅峰。"; +const char * const ANS_TEST1 = "[\"我:r\", \"是:v\", \"蓝翔:x\", \"技工:n\", \"拖拉机:n\", \"学院:n\", \"手扶拖拉机:n\", \"专业:n\", \"的:uj\", \"。:x\", \"不用:v\", \"多久:m\", \",:x\", \"我:r\", \"就:d\", \"会:v\", \"升职:v\", \"加薪:nr\", \",:x\", \"当上:t\", \"总经理:n\", \",:x\", \"出任:v\", \"CEO:x\", \",:x\", \"迎娶:v\", \"白富:x\", \"美:ns\", \",:x\", \"走上:v\", \"人生:n\", \"巅峰:n\", \"。:x\"]"; + +TEST(PosTaggerTest, Test1) +{ + PosTagger tagger("../dict/jieba.dict.utf8", "../dict/hmm_model.utf8"); + vector > res; + tagger.tag(QUERY_TEST1, res); + string s; + s << res; + ASSERT_TRUE(s == ANS_TEST1); +}