admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2025-12-08 01:58:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add runtime stats unittest

Browse Source 
Signed-off-by: sunby <bingyi.sun@zilliz.com>
This commit is contained in:
sunby 2020-12-25 16:08:35 +08:00 committed by yefu.chen
parent a03ab05c6c
commit e21397d68a
1 changed files with 30 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
internal/master/runtime_stats_test.go
View File

@ -41,3 +41,33 @@ func TestRuntimeStats_UpdateFieldStats(t *testing.T) {
assert.EqualValues(t, 1, found) assert.EqualValues(t, 1, found)
} }
} }
func TestRuntimeStats_GetTotalNumOfRelatedSegments(t *testing.T) {
runtimeStats := NewRuntimeStats()
runtimeStats.collStats = make(map[UniqueID]*CollRuntimeStats)
runtimeStats.collStats[1] = &CollRuntimeStats{
fieldIndexStats: map[UniqueID][]*FieldIndexRuntimeStats{
100: {
{1, []*commonpb.KeyValuePair{{Key: "k1", Value: "v1"}}, 10},
{3, []*commonpb.KeyValuePair{{Key: "k1", Value: "v1"}}, 20},
{2, []*commonpb.KeyValuePair{{Key: "k2", Value: "v2"}}, 20},
},
200: {
{1, []*commonpb.KeyValuePair{}, 20},
},
},
}
runtimeStats.collStats[2] = &CollRuntimeStats{
fieldIndexStats: map[UniqueID][]*FieldIndexRuntimeStats{
100: {
{1, []*commonpb.KeyValuePair{{Key: "k1", Value: "v1"}}, 10},
},
},
}
assert.EqualValues(t, 30, runtimeStats.GetTotalNumOfRelatedSegments(1, 100, []*commonpb.KeyValuePair{{Key: "k1", Value: "v1"}}))
assert.EqualValues(t, 20, runtimeStats.GetTotalNumOfRelatedSegments(1, 100, []*commonpb.KeyValuePair{{Key: "k2", Value: "v2"}}))
assert.EqualValues(t, 20, runtimeStats.GetTotalNumOfRelatedSegments(1, 200, []*commonpb.KeyValuePair{}))
assert.EqualValues(t, 10, runtimeStats.GetTotalNumOfRelatedSegments(2, 100, []*commonpb.KeyValuePair{{Key: "k1", Value: "v1"}}))
}