Spaces:
Running
Running
File size: 3,328 Bytes
b110593 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 |
// _ _
// __ _____ __ ___ ___ __ _| |_ ___
// \ \ /\ / / _ \/ _` \ \ / / |/ _` | __/ _ \
// \ V V / __/ (_| |\ V /| | (_| | || __/
// \_/\_/ \___|\__,_| \_/ |_|\__,_|\__\___|
//
// Copyright © 2016 - 2024 Weaviate B.V. All rights reserved.
//
// CONTACT: [email protected]
//
package aggregator
import (
"testing"
"github.com/stretchr/testify/assert"
"github.com/weaviate/weaviate/entities/aggregation"
)
func TestTextAggregator_TopOccurrencesCalculation(t *testing.T) {
testCases := []struct {
name string
texts []string
expectedCount int
expectedTopOccurrences []aggregation.TextOccurrence
}{
{
name: "All texts occurring once",
texts: []string{"b_occurs1", "c_occurs1", "g_occurs1", "f_occurs1", "a_occurs1", "d_occurs1", "e_occurs1"},
expectedCount: 7,
expectedTopOccurrences: []aggregation.TextOccurrence{
{Value: "a_occurs1", Occurs: 1},
{Value: "b_occurs1", Occurs: 1},
{Value: "c_occurs1", Occurs: 1},
{Value: "d_occurs1", Occurs: 1},
{Value: "e_occurs1", Occurs: 1},
},
},
{
name: "All texts occurring different number of times",
texts: []string{
"b_occurs2", "e_occurs5", "d_occurs4", "c_occurs3", "g_occurs7", "e_occurs5", "d_occurs4",
"f_occurs6", "g_occurs7", "c_occurs3", "b_occurs2", "g_occurs7", "f_occurs6", "g_occurs7", "d_occurs4",
"a_occurs1", "f_occurs6", "g_occurs7", "g_occurs7", "f_occurs6", "d_occurs4", "e_occurs5", "g_occurs7",
"c_occurs3", "f_occurs6", "e_occurs5", "f_occurs6", "e_occurs5",
},
expectedCount: 28,
expectedTopOccurrences: []aggregation.TextOccurrence{
{Value: "g_occurs7", Occurs: 7},
{Value: "f_occurs6", Occurs: 6},
{Value: "e_occurs5", Occurs: 5},
{Value: "d_occurs4", Occurs: 4},
{Value: "c_occurs3", Occurs: 3},
},
},
{
name: "Some texts occurring same number of times",
texts: []string{
"a_occurs4", "b_occurs3", "g_occurs4", "f_occurs3", "a_occurs4", "e_occurs2", "a_occurs4",
"c_occurs2", "g_occurs4", "f_occurs3", "b_occurs3", "c_occurs2", "a_occurs4", "f_occurs3", "g_occurs4",
"b_occurs3", "d_occurs1", "e_occurs2", "g_occurs4",
},
expectedCount: 19,
expectedTopOccurrences: []aggregation.TextOccurrence{
{Value: "a_occurs4", Occurs: 4},
{Value: "g_occurs4", Occurs: 4},
{Value: "b_occurs3", Occurs: 3},
{Value: "f_occurs3", Occurs: 3},
{Value: "c_occurs2", Occurs: 2},
},
},
{
name: "Fewer texts than limit",
texts: []string{"b_occurs3", "d_occurs3", "c_occurs1", "d_occurs3", "b_occurs3", "b_occurs3", "a_occurs1", "d_occurs3"},
expectedCount: 8,
expectedTopOccurrences: []aggregation.TextOccurrence{
{Value: "b_occurs3", Occurs: 3},
{Value: "d_occurs3", Occurs: 3},
{Value: "a_occurs1", Occurs: 1},
{Value: "c_occurs1", Occurs: 1},
},
},
}
for _, tc := range testCases {
t.Run(tc.name, func(t *testing.T) {
agg := newTextAggregator(5)
for _, text := range tc.texts {
agg.AddText(text)
}
res := agg.Res()
assert.Equal(t, tc.expectedCount, res.Count)
assert.Equal(t, tc.expectedTopOccurrences, res.Items)
})
}
}
|