add more test cases

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);