add test cases to bi-gram

1 files changed, 5 insertions, 0 deletions

diff --git a/tests/storage/test_ngram.cpp b/tests/storage/test_ngram.cpp
index 7fc1b85..9577167 100644
--- a/tests/storage/test_ngram.cpp
+++ b/tests/storage/test_ngram.cpp
@@ -74,4 +74,9 @@ int main(int argc, char * argv[]){
 	phrase_token_t * token = &g_array_index(items, phrase_token_t, i);
 	printf("item:%d\n", *token);
     }
+
+    assert(bigram.load_db("/tmp/test.db"));
+    assert(bigram.save_db("/tmp/test.db"));
+
+    return 0;
 }