don't raise error when data file is not found

high-moctane · Jan 30, 2020 · 846571e · 846571e
1 parent 7ae9ff7
commit 846571e
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 0 deletions.
diff --git a/nextword.go b/nextword.go
@@ -129,6 +129,9 @@ func (nw *Nextword) searchNgram(ngram []string) (candidates []string, err error)
 
 	// open
 	path := filepath.Join(nw.params.DataPath, fname)
+	if _, err = os.Stat(path); err != nil {
+		return nil, nil
+	}
 	f, err := os.Open(path)
 	if err != nil {
 		return

diff --git a/nextword_test.go b/nextword_test.go
@@ -160,6 +160,30 @@ func TestNextword_Suggest(t *testing.T) {
 			nil,
 			nil,
 		},
+
+		// not alphabet
+		{
+			"-",
+			&NextwordParams{
+				DataPath:     NextwordTestDataPath,
+				CandidateNum: 10,
+				Greedy:       false,
+			},
+			nil,
+			nil,
+		},
+
+		// not alphabet
+		{
+			"- ",
+			&NextwordParams{
+				DataPath:     NextwordTestDataPath,
+				CandidateNum: 10,
+				Greedy:       false,
+			},
+			nil,
+			nil,
+		},
 	}
 
 	for idx, test := range tests {