write test chewing

4 files changed, 77 insertions, 7 deletions

diff --git a/src/libpinyin.ver b/src/libpinyin.ver
index 0bc8c09..2221ce6 100644
--- a/src/libpinyin.ver
+++ b/src/libpinyin.ver
@@ -21,6 +21,8 @@ LIBPINYIN {
         pinyin_translate_token;
         pinyin_alloc_auxiliary_arrays;
         pinyin_free_auxiliary_arrays;
+        pinyin_parse_chewing;
+        pinyin_parse_more_chewings;
 
     local:
 	*;
diff --git a/src/pinyin.cpp b/src/pinyin.cpp
index 8e5773d..45da2fa 100644
--- a/src/pinyin.cpp
+++ b/src/pinyin.cpp
@@ -235,9 +235,9 @@ bool pinyin_parse_chewing(pinyin_context_t * context,
     return chewing_len == parse_len;
 }
 
-bool pinyin_parse_more_chewing(pinyin_context_t * context,
-                               const char * chewings,
-                               PinyinKeyVector * pinyin_keys){
+bool pinyin_parse_more_chewings(pinyin_context_t * context,
+                                const char * chewings,
+                                PinyinKeyVector pinyin_keys){
     int chewing_len = strlen(chewings);
     PinyinKeyPosVector poses;
     poses = g_array_new(FALSE, FALSE, sizeof(PinyinKeyPos));
@@ -247,7 +247,7 @@ bool pinyin_parse_more_chewing(pinyin_context_t * context,
           poses, chewings, chewing_len);
 
     g_array_free(poses, TRUE);
-    return pinyin_len == parse_len;
+    return chewing_len == parse_len;
 }
 
 
diff --git a/src/pinyin.h b/src/pinyin.h
index 2ac1001..7b3bacb 100644
--- a/src/pinyin.h
+++ b/src/pinyin.h
@@ -77,9 +77,9 @@ bool pinyin_parse_more_doubles(pinyin_context_t * context,
 bool pinyin_parse_chewing(pinyin_context_t * context,
                           const char * onechewing,
                           PinyinKey * onekey);
-bool pinyin_parse_more_chewing(pinyin_context_t * context,
-                               const char * chewings,
-                               PinyinKeyVector * pinyin_keys);
+bool pinyin_parse_more_chewings(pinyin_context_t * context,
+                                const char * chewings,
+                                PinyinKeyVector pinyin_keys);
 
 bool pinyin_get_candidates(pinyin_context_t * context,
                            size_t offset,
diff --git a/tests/test_chewing.cpp b/tests/test_chewing.cpp
index e69de29..060d482 100644
--- a/tests/test_chewing.cpp
+++ b/tests/test_chewing.cpp
@@ -0,0 +1,68 @@
+/* 
+ *  libpinyin
+ *  Library to deal with pinyin.
+ *  
+ *  Copyright (C) 2011 Peng Wu <alexepico@gmail.com>
+ *  
+ *  This program is free software; you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License as published by
+ *  the Free Software Foundation; either version 2 of the License, or
+ *  (at your option) any later version.
+ * 
+ *  This program is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ *  GNU General Public License for more details.
+ *  
+ *  You should have received a copy of the GNU General Public License
+ *  along with this program; if not, write to the Free Software
+ *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+ */
+
+
+#include <stdlib.h>
+#include "pinyin.h"
+
+int main(int argc, char * argv[]){
+    pinyin_context_t * context =
+        pinyin_init("../data", "../data");
+
+    PinyinKeyVector pinyin_keys = NULL;
+    CandidateConstraints constraints = NULL;
+    MatchResults match_results = NULL;
+
+    pinyin_alloc_auxiliary_arrays
+        (context, &pinyin_keys, &constraints, &match_results);
+
+    char* linebuf = NULL;
+    size_t size = 0;
+    ssize_t read;
+    while( (read = getline(&linebuf, &size, stdin)) != -1 ){
+        if ( '\n' == linebuf[strlen(linebuf) - 1] ) {
+            linebuf[strlen(linebuf) - 1] = '\0';
+        }
+
+	if ( strcmp ( linebuf, "quit" ) == 0)
+            break;
+
+        pinyin_parse_more_chewings(context, linebuf, pinyin_keys);
+        pinyin_update_constraints(context, pinyin_keys, constraints);
+        pinyin_get_guessed_tokens(context, pinyin_keys, constraints,
+                                  match_results);
+
+        char * sentence = NULL;
+        pinyin_get_sentence (context, match_results, &sentence);
+        printf("%s\n", sentence);
+        g_free(sentence);
+
+        pinyin_train(context, pinyin_keys, constraints, match_results);
+        pinyin_reset(context, pinyin_keys, constraints, match_results);
+        pinyin_save(context);
+    }
+
+    pinyin_free_auxiliary_arrays
+        (context, &pinyin_keys, &constraints, &match_results);
+    pinyin_fini(context);
+    free(linebuf);
+    return 0;
+}