diff options
Diffstat (limited to 'tests')
-rw-r--r-- | tests/storage/test_flexible_ngram.cpp | 1 | ||||
-rw-r--r-- | tests/storage/test_ngram.cpp | 13 |
2 files changed, 3 insertions, 11 deletions
diff --git a/tests/storage/test_flexible_ngram.cpp b/tests/storage/test_flexible_ngram.cpp index eefb5ac..ea17adb 100644 --- a/tests/storage/test_flexible_ngram.cpp +++ b/tests/storage/test_flexible_ngram.cpp @@ -39,6 +39,7 @@ int main(int argc, char * argv[]) { assert(bigram.attach("/tmp/training.db")); bigram.store(1, &single_gram); assert(single_gram.insert_array_item(5, 8)); + assert(single_gram.remove_array_item(1, freq)); assert(single_gram.set_array_header(32)); assert(single_gram.get_array_header(freq)); printf("new array header:%d\n", freq); diff --git a/tests/storage/test_ngram.cpp b/tests/storage/test_ngram.cpp index 044cf59..7fc1b85 100644 --- a/tests/storage/test_ngram.cpp +++ b/tests/storage/test_ngram.cpp @@ -31,17 +31,16 @@ int main(int argc, char * argv[]){ for ( size_t i = 0; i < array->len; ++i){ BigramPhraseItem * item = &g_array_index(array, BigramPhraseItem, i); printf("item:%d:%f\n", item->m_token, item->m_freq); - } - + } assert(single_gram.get_total_freq(freq)); assert(freq == total_freq); - Bigram bigram; assert(bigram.attach("/tmp/test.db", ATTACH_CREATE|ATTACH_READWRITE)); bigram.store(1, &single_gram); assert(single_gram.insert_freq(5, 8)); + assert(single_gram.remove_freq(1, freq)); single_gram.set_total_freq(32); bigram.store(2, &single_gram); @@ -62,14 +61,6 @@ int main(int argc, char * argv[]){ } printf("--------------------------------------------------------\n"); - single_gram.prune(); - g_array_set_size(array, 0); - range.m_range_begin = 0; range.m_range_end = 8; - single_gram.search(&range,array); - for ( size_t i = 0; i < array->len; ++i){ - BigramPhraseItem * item = &g_array_index(array, BigramPhraseItem, i); - printf("item:%d:%f\n", item->m_token, item->m_freq); - } assert(single_gram.get_total_freq(freq)); printf("total_freq:%d\n", freq); |