add asserts to gen ngram

diff --git a/utils/training/gen_ngram.cpp b/utils/training/gen_ngram.cpp
index 6109e92..3203cd1 100644 (file)
--- a/utils/training/gen_ngram.cpp
+++ b/utils/training/gen_ngram.cpp
@@ -126,9 +126,9 @@ int main(int argc, char * argv[]){
            guint32 freq, total_freq;
            //increase freq
            if (single_gram->get_freq(cur_token, freq))
-                single_gram->set_freq(cur_token, freq + 1);
+                assert(single_gram->set_freq(cur_token, freq + 1));
             else
-                single_gram->insert_freq(cur_token, 1);
+                assert(single_gram->insert_freq(cur_token, 1));
            //increase total freq
            single_gram->get_total_freq(total_freq);
            single_gram->set_total_freq(total_freq + 1);