From 643bbc2a749f9a37a915c9daedaf544478711e42 Mon Sep 17 00:00:00 2001 From: Peng Wu Date: Mon, 18 Apr 2011 14:37:21 +0800 Subject: refine insert/set freq --- tests/storage/test_ngram.cpp | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) (limited to 'tests/storage') diff --git a/tests/storage/test_ngram.cpp b/tests/storage/test_ngram.cpp index 1dd65a8..841a0f4 100644 --- a/tests/storage/test_ngram.cpp +++ b/tests/storage/test_ngram.cpp @@ -7,16 +7,19 @@ int main(int argc, char * argv[]){ const guint32 total_freq = 16; assert(single_gram.set_total_freq(total_freq)); - phrase_token_t tokens[6] = { 2, 6, 4, 3, 1, 3}; guint32 freqs[6] = { 1, 2, 4, 8, 16, 32}; + guint32 freq; + for(size_t i = 0; i < 6 ;++i){ - single_gram.set_freq(tokens[i], freqs[i]); + if ( single_gram.get_freq(tokens[i], freq)) + single_gram.set_freq(tokens[i], freqs[i]); + else + single_gram.insert_freq(tokens[i], freqs[i]); } - guint32 freq; single_gram.get_freq(3, freq); assert(freq == 32); @@ -38,7 +41,7 @@ int main(int argc, char * argv[]){ Bigram bigram; assert(bigram.attach(NULL, "/tmp/system.db")); bigram.store(1, &single_gram); - single_gram.set_freq(5, 8); + assert(single_gram.insert_freq(5, 8)); single_gram.set_total_freq(32); bigram.store(2, &single_gram); -- cgit