mirror of
https://github.com/yanyiwu/cppjieba.git
synced 2025-07-18 00:00:12 +08:00
19 lines
729 B
C++
19 lines
729 B
C++
#include "src/KeywordExtractor.hpp"
|
|
#include "gtest/gtest.h"
|
|
|
|
using namespace CppJieba;
|
|
|
|
|
|
|
|
TEST(KeywordExtractorTest, Test1)
|
|
{
|
|
KeywordExtractor extractor("../dict/extra_dict/jieba.dict.small.utf8", "../dict/hmm_model.utf8", "../dict/idf.utf8", "../dict/stop_words.utf8");
|
|
string s("我是拖拉机学院手扶拖拉机专业的。不用多久，我就会升职加薪，当上总经理，迎娶白富美，走上人生巅峰。");
|
|
string res;
|
|
vector<pair<string, double> > wordweights;
|
|
size_t topN = 5;
|
|
extractor.extract(s, wordweights, topN);
|
|
res << wordweights;
|
|
ASSERT_EQ(res, "[\"白富美:11.7392\", \"升职:10.8562\", \"加薪:10.6426\", \"迎娶:10.0505\", \"手扶拖拉机:10.0089\"]");
|
|
}
|