added benchmark to collector
This commit is contained in:
parent
5ac9583370
commit
7284c10020
|
@ -10,6 +10,8 @@
|
||||||
package collectors
|
package collectors
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"math/rand"
|
||||||
|
"strconv"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"github.com/blevesearch/bleve/search"
|
"github.com/blevesearch/bleve/search"
|
||||||
|
@ -213,3 +215,26 @@ func TestTop10ScoresSkip10(t *testing.T) {
|
||||||
t.Errorf("expected highest score to be 9.5ß, got %f", results[0].Score)
|
t.Errorf("expected highest score to be 9.5ß, got %f", results[0].Score)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func BenchmarkTop10of100000Scores(b *testing.B) {
|
||||||
|
|
||||||
|
matches := make(search.DocumentMatchCollection, 0, 100000)
|
||||||
|
for i := 0; i < 100000; i++ {
|
||||||
|
matches = append(matches, &search.DocumentMatch{
|
||||||
|
ID: strconv.Itoa(i),
|
||||||
|
Score: rand.Float64(),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
searcher := &stubSearcher{
|
||||||
|
matches: matches,
|
||||||
|
}
|
||||||
|
|
||||||
|
collector := NewTopScorerCollector(10)
|
||||||
|
b.ResetTimer()
|
||||||
|
|
||||||
|
collector.Collect(searcher)
|
||||||
|
res := collector.Results()
|
||||||
|
for _, dm := range res {
|
||||||
|
b.Logf("%s - %f\n", dm.ID, dm.Score)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in New Issue
Block a user