15 lines
852 B
Go
15 lines
852 B
Go
package retrieval_test
|
|
|
|
import (
|
|
"git.cin.su/m/Assistant/runtime/retrieval"
|
|
"testing"
|
|
)
|
|
|
|
func TestScopeAndVersionFilterBeforeRanking(t *testing.T) {
|
|
docs := []retrieval.RetrievalDocument{{ID: "wrong-company", PrincipalID: "a", ContextScopeRef: "b", VersionRef: "v", Lane: retrieval.LaneSemantic, Relevance: 100}, {ID: "right", PrincipalID: "a", ContextScopeRef: "a", VersionRef: "v", Lane: retrieval.LaneNegativeExperience, Relevance: 1}, {ID: "wrong-version", PrincipalID: "a", ContextScopeRef: "a", VersionRef: "old", Lane: retrieval.LaneSemantic, Relevance: 100}}
|
|
got := retrieval.FilterAndRank(docs, retrieval.Query{PrincipalID: "a", ContextScopeRef: "a", VersionRef: "v", Lanes: []retrieval.Lane{retrieval.LaneSemantic, retrieval.LaneNegativeExperience}, Limit: 10})
|
|
if len(got) != 1 || got[0].ID != "right" {
|
|
t.Fatalf("got %#v", got)
|
|
}
|
|
}
|