fixed Suggester.suggestNgram error which it occurs when the first wor…

…d begins with a non-alphabetic character.
high-moctane · Oct 1, 2019 · 0adb536 · 0adb536
1 parent 56c9c77
commit 0adb536
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/suggester.go b/suggester.go
@@ -93,6 +93,10 @@ func (sg *Suggester) suggestNgram(words []string) (candidates []string, err erro
 	// open data
 	n := len(words) + 1
 	initial := strings.ToLower(string([]rune(words[0])[0]))
+	if (initial < "A" || "Z" <= initial) && (initial < "a" || "z" <= initial) {
+		return
+	}
+
 	f, err := os.Open(sg.filePath(n, initial))
 	if err != nil {
 		return

diff --git a/suggester_test.go b/suggester_test.go
@@ -194,6 +194,14 @@ func TestSuggester_SuggestNgram(t *testing.T) {
 			[]string{"I felt paint"},
 			nil,
 		},
+		{
+			[]string{"0000000000"},
+			nil,
+		},
+		{
+			[]string{"🤔"},
+			nil,
+		},
 	}
 
 	for idx, test := range tests {