fix broken tests
This commit is contained in:
parent
949accc3e4
commit
e6ee7a11c4
|
@ -175,7 +175,7 @@ func TestSimpleFragmenter(t *testing.T) {
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
fragmenter := NewSimpleFragmenter()
|
fragmenter := NewSimpleFragmenterWithSize(100)
|
||||||
for _, test := range tests {
|
for _, test := range tests {
|
||||||
fragments := fragmenter.Fragment(test.orig, test.ot)
|
fragments := fragmenter.Fragment(test.orig, test.ot)
|
||||||
if !reflect.DeepEqual(fragments, test.fragments) {
|
if !reflect.DeepEqual(fragments, test.fragments) {
|
||||||
|
|
|
@ -116,7 +116,14 @@ func (s *SimpleHighlighter) BestFragmentsInField(dm *DocumentMatch, doc *documen
|
||||||
// now that we have the best fragments, we can format them
|
// now that we have the best fragments, we can format them
|
||||||
formattedFragments := make([]string, len(bestFragments))
|
formattedFragments := make([]string, len(bestFragments))
|
||||||
for i, fragment := range bestFragments {
|
for i, fragment := range bestFragments {
|
||||||
formattedFragments[i] = s.sep + s.formatter.Format(fragment, dm.Locations[field]) + s.sep
|
formattedFragments[i] = ""
|
||||||
|
if fragment.start != 0 {
|
||||||
|
formattedFragments[i] += s.sep
|
||||||
|
}
|
||||||
|
formattedFragments[i] += s.formatter.Format(fragment, dm.Locations[field])
|
||||||
|
if fragment.end != len(fragment.orig) {
|
||||||
|
formattedFragments[i] += s.sep
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if dm.Fragments == nil {
|
if dm.Fragments == nil {
|
||||||
|
|
|
@ -42,7 +42,7 @@ func TestSimpleHighlighter(t *testing.T) {
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
expectedFragment := DEFAULT_SEPARATOR + "the " + DEFAULT_ANSI_HIGHLIGHT + "quick" + reset + " brown " + DEFAULT_ANSI_HIGHLIGHT + "fox" + reset + " jumps over the lazy dog" + DEFAULT_SEPARATOR
|
expectedFragment := "the " + DEFAULT_ANSI_HIGHLIGHT + "quick" + reset + " brown " + DEFAULT_ANSI_HIGHLIGHT + "fox" + reset + " jumps over the lazy dog"
|
||||||
doc := document.NewDocument("a").AddField(document.NewTextField("desc", []uint64{}, []byte("the quick brown fox jumps over the lazy dog")))
|
doc := document.NewDocument("a").AddField(document.NewTextField("desc", []uint64{}, []byte("the quick brown fox jumps over the lazy dog")))
|
||||||
|
|
||||||
fragment := highlighter.BestFragmentInField(&docMatch, doc, "desc")
|
fragment := highlighter.BestFragmentInField(&docMatch, doc, "desc")
|
||||||
|
@ -143,6 +143,7 @@ Etiam vel augue vel nisl commodo suscipit et ac nisl. Quisque eros diam, porttit
|
||||||
}
|
}
|
||||||
|
|
||||||
highlighter := NewSimpleHighlighter()
|
highlighter := NewSimpleHighlighter()
|
||||||
|
highlighter.SetFragmenter(NewSimpleFragmenterWithSize(100))
|
||||||
fragments := highlighter.BestFragmentsInField(&docMatch, doc, "full", 5)
|
fragments := highlighter.BestFragmentsInField(&docMatch, doc, "full", 5)
|
||||||
|
|
||||||
if !reflect.DeepEqual(fragments, expectedFragments) {
|
if !reflect.DeepEqual(fragments, expectedFragments) {
|
||||||
|
|
Loading…
Reference in New Issue
Block a user