Merge pull request #117 from go-ego/range-pr

add empty dictionary function support
go-ego · Sep 19, 2021 · 569222e · 569222e
2 parents 0c45dbf + 1dbfdc5
commit 569222e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/dict_util.go b/dict_util.go
@@ -77,6 +77,12 @@ func (seg *Segmenter) RemoveToken(text string) error {
 	return seg.Dict.RemoveToken(token)
 }
 
+// Empty empty the seg dictionary
+func (seg *Segmenter) Empty() error {
+	seg.Dict = nil
+	return nil
+}
+
 // LoadDictMap load dictionary from []map[string]string
 func (seg *Segmenter) LoadDictMap(dict []map[string]string) error {
 	if seg.Dict == nil {

diff --git a/gse_test.go b/gse_test.go
@@ -254,6 +254,10 @@ func TestNum(t *testing.T) {
 	seg.Alpha = true
 	s = seg.CutSearch(text)
 	tt.Equal(t, "[t 1 2 3 t e s t 1 2 3   n u m 1 2 3 - 1]", s)
+
+	err = seg.Empty()
+	tt.Nil(t, err)
+	tt.Nil(t, seg.Dict)
 }
 
 func TestUrl(t *testing.T) {