Add tests for fuzzy search

2015-12-20 17:00:46 +01:00 · 2015-12-20 17:00:46 +01:00 · 84c755cdb0
commit 84c755cdb0
parent f7698f1f15
1 changed files with 130 additions and 0 deletions
--- a/search/searchers/search_fuzzy_test.go
+++ b/search/searchers/search_fuzzy_test.go
@ -0,0 +1,130 @@
+//  Copyright (c) 2015 Couchbase, Inc.
+//  Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file
+//  except in compliance with the License. You may obtain a copy of the License at
+//    http://www.apache.org/licenses/LICENSE-2.0
+//  Unless required by applicable law or agreed to in writing, software distributed under the
+//  License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
+//  either express or implied. See the License for the specific language governing permissions
+//  and limitations under the License.
+
+package searchers
+
+import (
+	"testing"
+
+	"github.com/blevesearch/bleve/search"
+)
+
+func TestFuzzySearch(t *testing.T) {
+	twoDocIndexReader, err := twoDocIndex.Reader()
+	if err != nil {
+		t.Error(err)
+	}
+
+	defer func() {
+		err := twoDocIndexReader.Close()
+		if err != nil {
+			t.Fatal(err)
+		}
+	}()
+
+	fuzzySearcherbeet, err := NewFuzzySearcher(twoDocIndexReader, "beet", 0, 1, "desc", 1.0, true)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	fuzzySearcherdouches, err := NewFuzzySearcher(twoDocIndexReader, "douches", 0, 2, "desc", 1.0, true)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	fuzzySearcheraplee, err := NewFuzzySearcher(twoDocIndexReader, "aplee", 0, 2, "desc", 1.0, true)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	fuzzySearcherprefix, err := NewFuzzySearcher(twoDocIndexReader, "water", 3, 2, "desc", 1.0, true)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	tests := []struct {
+		searcher search.Searcher
+		results  []*search.DocumentMatch
+	}{
+		{
+			searcher: fuzzySearcherbeet,
+			results: []*search.DocumentMatch{
+				&search.DocumentMatch{
+					ID:    "1",
+					Score: 1.0,
+				},
+				&search.DocumentMatch{
+					ID:    "2",
+					Score: 0.5,
+				},
+				&search.DocumentMatch{
+					ID:    "3",
+					Score: 0.5,
+				},
+				&search.DocumentMatch{
+					ID:    "4",
+					Score: 0.9999999838027345,
+				},
+			},
+		},
+		{
+			searcher: fuzzySearcherdouches,
+			results:  []*search.DocumentMatch{},
+		},
+		{
+			searcher: fuzzySearcheraplee,
+			results: []*search.DocumentMatch{
+				&search.DocumentMatch{
+					ID:    "3",
+					Score: 0.9581453659370776,
+				},
+			},
+		},
+		{
+			searcher: fuzzySearcherprefix,
+			results: []*search.DocumentMatch{
+				&search.DocumentMatch{
+					ID:    "5",
+					Score: 1.916290731874155,
+				},
+			},
+		},
+	}
+
+	for testIndex, test := range tests {
+		defer func() {
+			err := test.searcher.Close()
+			if err != nil {
+				t.Fatal(err)
+			}
+		}()
+
+		next, err := test.searcher.Next()
+		i := 0
+		for err == nil && next != nil {
+			if i < len(test.results) {
+				if next.ID != test.results[i].ID {
+					t.Errorf("expected result %d to have id %s got %s for test %d", i, test.results[i].ID, next.ID, testIndex)
+				}
+				if next.Score != test.results[i].Score {
+					t.Errorf("expected result %d to have score %v got %v for test %d", i, test.results[i].Score, next.Score, testIndex)
+					t.Logf("scoring explanation: %s", next.Expl)
+				}
+			}
+			next, err = test.searcher.Next()
+			i++
+		}
+		if err != nil {
+			t.Fatalf("error iterating searcher: %v for test %d", err, testIndex)
+		}
+		if len(test.results) != i {
+			t.Errorf("expected %d results got %d for test %d", len(test.results), i, testIndex)
+		}
+	}
+}