diff options
Diffstat (limited to 'tests/storage/test_ngram.cpp')
-rw-r--r-- | tests/storage/test_ngram.cpp | 72 |
1 files changed, 27 insertions, 45 deletions
diff --git a/tests/storage/test_ngram.cpp b/tests/storage/test_ngram.cpp index 0b8f199..1dd65a8 100644 --- a/tests/storage/test_ngram.cpp +++ b/tests/storage/test_ngram.cpp @@ -43,55 +43,37 @@ int main(int argc, char * argv[]){ bigram.store(2, &single_gram); - printf("--------------------------------------------------------\n"); - SingleGram * system, * user; - bigram.load(1, system, user); - assert(NULL == system); - g_array_set_size(array, 0); - range.m_range_begin = 0; range.m_range_end = 8; - user->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); - } - delete user; - printf("--------------------------------------------------------\n"); - bigram.load(2, system, user); - assert(NULL == system); - g_array_set_size(array, 0); - range.m_range_begin = 0; range.m_range_end = 8; - user->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); - } - delete user; + SingleGram * system, * user; + for ( int m = 1; m <= 2; ++m ){ + printf("--------------------------------------------------------\n"); + bigram.load(m, system, user); + assert(NULL == system); + g_array_set_size(array, 0); + range.m_range_begin = 0; range.m_range_end = 8; + user->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); + } + delete user; + } bigram.attach("/tmp/system.db", NULL); - printf("--------------------------------------------------------\n"); - bigram.load(1, system, user); - assert(NULL == user); - g_array_set_size(array, 0); - range.m_range_begin = 0; range.m_range_end = 8; - system->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); - } - delete system; - - printf("--------------------------------------------------------\n"); - bigram.load(2, system, user); - assert(NULL == user); - g_array_set_size(array, 0); - range.m_range_begin = 0; range.m_range_end = 8; - system->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); + + for ( int m = 1; m <=2; ++m ){ + printf("--------------------------------------------------------\n"); + bigram.load(m, system, user); + assert(NULL == user); + g_array_set_size(array, 0); + range.m_range_begin = 0; range.m_range_end = 8; + system->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); + } + delete system; } - delete system; printf("--------------------------------------------------------\n"); single_gram.prune(); |