func TestNegativeQueryBoost(t *testing.T) { Test(t, func(t *T) { q := search.NewTermQuery(index.NewTerm("foo", "bar")) q.SetBoost(-42) t.Assert(-42 == q.Boost()) directory := NewDirectory() defer directory.Close() analyzer := analysis.NewMockAnalyzerWithRandom(Random()) conf := NewIndexWriterConfig(TEST_VERSION_CURRENT, analyzer) writer, err := index.NewIndexWriter(directory, conf) if err != nil { t.Error(err) } defer writer.Close() d := index.NewDocument() d.Add(NewTextField("foo", "bar", true)) writer.AddDocument(d.Fields()) writer.Close() // ensure index is written reader, err := index.OpenDirectoryReader(directory) if err != nil { t.Error(err) } defer reader.Close() searcher := NewSearcher(reader) res, err := searcher.Search(q, nil, 1000) if err != nil { t.Error(err) } hits := res.ScoreDocs t.Assert(1 == len(hits)) t.Assert2(hits[0].Score < 0, fmt.Sprintf("score is not negative: %v", hits[0].Score)) explain, err := searcher.Explain(q, hits[0].Doc) if err != nil { t.Error(err) } t.Assert2(isSimilar(hits[0].Score, explain.Value(), 0.01), "score doesn't match explanation") t.Assert2(explain.IsMatch(), "explain doesn't think doc is a match") }) }
func TestKeywordSearch(t *testing.T) { d, err := store.OpenFSDirectory("testdata/belfrysample") if err != nil { t.Error(err) } r, err := index.OpenDirectoryReader(d) if err != nil { t.Error(err) } if r == nil { t.Error("DirectoryReader cannot be opened.") } if len(r.Leaves()) < 1 { t.Error("Should have one leaf.") } ss := NewIndexSearcher(r) docs, err := ss.SearchTop(NewTermQuery(index.NewTerm("content", "bat")), 10) if err != nil { t.Error(err) } assertEquals(t, 8, docs.TotalHits) doc, err := r.Document(docs.ScoreDocs[0].Doc) if err != nil { t.Error(err) } assertEquals(t, "Bat recycling", doc.Get("title")) }
func main() { util.SetDefaultInfoStream(util.NewPrintStreamInfoStream(os.Stdout)) index.DefaultSimilarity = func() index.Similarity { return search.NewDefaultSimilarity() } directory, _ := store.OpenFSDirectory("test_index") analyzer := std.NewStandardAnalyzer() conf := index.NewIndexWriterConfig(util.VERSION_LATEST, analyzer) writer, _ := index.NewIndexWriter(directory, conf) d := document.NewDocument() d.Add(document.NewTextFieldFromString("foo", "bar", document.STORE_YES)) writer.AddDocument(d.Fields()) writer.Close() // ensure index is written reader, _ := index.OpenDirectoryReader(directory) searcher := search.NewIndexSearcher(reader) q := search.NewTermQuery(index.NewTerm("foo", "bar")) res, _ := searcher.Search(q, nil, 1000) fmt.Printf("Found %v hit(s).\n", res.TotalHits) for _, hit := range res.ScoreDocs { fmt.Printf("Doc %v score: %v\n", hit.Doc, hit.Score) doc, _ := reader.Document(hit.Doc) fmt.Printf("foo -> %v\n", doc.Get("foo")) } }
func main() { log.Print("Oepning FSDirectory...") // path := "src/github.com/balzaczyy/golucene/search/testdata/win8/belfrysample" path := "core/search/testdata/win8/belfrysample" // path := "/private/tmp/kc/index/belfrysample" // path := "c:/tmp/kc/index/belfrysample" d, err := store.OpenFSDirectory(path) if err != nil { panic(err) } log.Print("Opening DirectoryReader...") r, err := index.OpenDirectoryReader(d) if err != nil { panic(err) } if r == nil { panic("DirectoryReader cannot be opened.") } if len(r.Leaves()) < 1 { panic("Should have one leaf.") } log.Print("Initializing IndexSearcher...") ss := search.NewIndexSearcher(r) log.Print("Searching...") docs, err := ss.SearchTop(search.NewTermQuery(index.NewTerm("content", "bat")), 10) if err != nil { panic(err) } log.Println("Hits:", docs.TotalHits) doc, err := r.Document(docs.ScoreDocs[0].Doc) if err != nil { panic(err) } log.Println("Hit 1's title: ", doc.Get("title")) }
func TestBasicIndexAndSearch(t *testing.T) { q := search.NewTermQuery(index.NewTerm("foo", "bar")) q.SetBoost(-42) os.RemoveAll(".gltest") directory, err := store.OpenFSDirectory(".gltest") It(t).Should("has no error: %v", err).Assert(err == nil) It(t).Should("has valid directory").Assert(directory != nil) fmt.Println("Directory", directory) defer directory.Close() analyzer := std.NewStandardAnalyzer() conf := index.NewIndexWriterConfig(util.VERSION_LATEST, analyzer) writer, err := index.NewIndexWriter(directory, conf) It(t).Should("has no error: %v", err).Assert(err == nil) d := docu.NewDocument() d.Add(docu.NewTextFieldFromString("foo", "bar", docu.STORE_YES)) err = writer.AddDocument(d.Fields()) It(t).Should("has no error: %v", err).Assert(err == nil) err = writer.Close() // ensure index is written It(t).Should("has no error: %v", err).Assert(err == nil) reader, err := index.OpenDirectoryReader(directory) It(t).Should("has no error: %v", err).Assert(err == nil) defer reader.Close() searcher := search.NewIndexSearcher(reader) res, err := searcher.Search(q, nil, 1000) It(t).Should("has no error: %v", err).Assert(err == nil) hits := res.ScoreDocs It(t).Should("expect 1 hits, but %v only.", len(hits)).Assert(len(hits) == 1) It(t).Should("expect score to be negative (got %v)", hits[0].Score).Verify(hits[0].Score < 0) explain, err := searcher.Explain(q, hits[0].Doc) It(t).Should("has no error: %v", err).Assert(err == nil) It(t).Should("score doesn't match explanation (%v vs %v)", hits[0].Score, explain.Value()).Verify(isSimilar(hits[0].Score, explain.Value(), 0.001)) It(t).Should("explain doesn't think doc is a match").Verify(explain.IsMatch()) }
func main() { fmt.Printf("tests_codec: %v\n", os.Getenv("tests_codec")) index.DefaultSimilarity = func() index.Similarity { return search.NewDefaultSimilarity() } // This controls how suite-level rules are nested. It is important // that _all_ rules declared in testcase are executed in proper // order if they depend on each other. ClassRuleChain(ClassEnvRule) BeforeSuite(nil) Test(nil, func(t *T) { q := search.NewTermQuery(index.NewTerm("foo", "bar")) q.SetBoost(-42) t.Assert(-42 == q.Boost()) directory := NewDirectory() fmt.Println("Directory", directory) defer directory.Close() analyzer := analysis.NewMockAnalyzerWithRandom(Random()) conf := NewIndexWriterConfig(TEST_VERSION_CURRENT, analyzer) writer, err := index.NewIndexWriter(directory, conf) if err != nil { t.Error(err) } if writer != nil { defer writer.Close() } // d := index.NewDocument() // d.Add(NewTextField("foo", "bar", true)) // writer.AddDocument(d.Fields()) // writer.Close() // ensure index is written // reader, err := index.OpenDirectoryReader(directory) // if err != nil { // t.Error(err) // } // defer reader.Close() // searcher := NewSearcher(reader) // res, err := searcher.Search(q, nil, 1000) // if err != nil { // t.Error(err) // } // hits := res.ScoreDocs // t.Assert(1 == len(hits)) // t.Assert2(hits[0].Score < 0, fmt.Sprintf("score is not negative: %v", hits[0].Score)) // explain, err := searcher.Explain(q, hits[0].Doc) // if err != nil { // t.Error(err) // } // t.Assert2(isSimilar(hits[0].Score, explain.Value(), 0.01), "score doesn't match explanation") // t.Assert2(explain.IsMatch(), "explain doesn't think doc is a match") }) AfterSuite(nil) }