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

Update unittest for calc_distance (#7488)

Browse Source
This commit is contained in:
groot 2021-09-06 11:35:02 +08:00 committed by GitHub
parent 1f590d6c26
commit af7b7537f0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 31 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
internal/util/distance/calc_distance_test.go
View File

@ -137,6 +137,10 @@ func TestCalcFloatDistance(t *testing.T) {
distances, err := CalcFloatDistance(dim, left, right, "L2")
assert.Nil(t, err)
invalid := CreateFloatArray(rightNum, 10)
_, err = CalcFloatDistance(dim, left, invalid, "L2")
assert.Error(t, err)
for i := int64(0); i < leftNum; i++ {
for j := int64(0); j < rightNum; j++ {
v1 := left[i*dim : (i+1)*dim]
@ -224,13 +228,6 @@ func TestCalcHamming(t *testing.T) {
v2[2] = 27
n := CalcHamming(dim, v1, 0, v2, 0)
assert.Equal(t, n, int32(4))
hamming := make([]int32, 1)
hamming[0] = n
tanimoto, err := CalcTanimotoCoefficient(dim, hamming)
realTanimoto := float64(int32(dim)-n) / (float64(dim)*2.0 - float64(int32(dim)-n))
assert.Nil(t, err)
assert.Less(t, math.Abs(float64(tanimoto[0])-realTanimoto), float64(PRECISION))
}
func TestCalcHamminDistance(t *testing.T) {
@ -256,4 +253,31 @@ func TestCalcHamminDistance(t *testing.T) {
n = CalcHamming(dim, left, 1, left, 0)
assert.Equal(t, n, distances[2])
invalid := CreateBinaryArray(leftNum, 200)
_, e = CalcHammingDistance(dim, invalid, left)
assert.Error(t, e)
_, e = CalcHammingDistance(dim, left, invalid)
assert.Error(t, e)
}
func TestCalcTanimotoCoefficient(t *testing.T) {
var dim int64 = 22
hamming := make([]int32, 2)
hamming[0] = 4
hamming[1] = 17
tanimoto, err := CalcTanimotoCoefficient(dim, hamming)
for i := 0; i < len(hamming); i++ {
realTanimoto := float64(int32(dim)-hamming[i]) / (float64(dim)*2.0 - float64(int32(dim)-hamming[i]))
assert.Nil(t, err)
assert.Less(t, math.Abs(float64(tanimoto[i])-realTanimoto), float64(PRECISION))
}
_, err = CalcTanimotoCoefficient(-1, hamming)
assert.Error(t, err)
_, err = CalcTanimotoCoefficient(3, hamming)
assert.Error(t, err)
}