mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-11-30 19:08:30 +08:00
e429965f32
issue：https://github.com/milvus-io/milvus/issues/22837 Add bfloat16 vector, add the index part of float16 vector. Signed-off-by: Writer-X <1256866856@qq.com>
47 lines
981 B
Go
47 lines
981 B
Go
package funcutil
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func Test_flattenedByteVectorsToByteVectors(t *testing.T) {
|
|
flattenedVectors := []byte{0, 1, 2, 3, 4, 5}
|
|
dimension := 3
|
|
|
|
actual := flattenedByteVectorsToByteVectors(flattenedVectors, dimension)
|
|
expected := [][]byte{
|
|
{0, 1, 2},
|
|
{3, 4, 5},
|
|
}
|
|
|
|
assert.Equal(t, expected, actual)
|
|
}
|
|
|
|
func Test_flattenedFloat16VectorsToByteVectors(t *testing.T) {
|
|
flattenedVectors := []byte{0, 1, 2, 3, 4, 5, 6, 7}
|
|
dimension := 2
|
|
|
|
actual := flattenedFloat16VectorsToByteVectors(flattenedVectors, dimension)
|
|
expected := [][]byte{
|
|
{0, 1, 2, 3},
|
|
{4, 5, 6, 7},
|
|
}
|
|
|
|
assert.Equal(t, expected, actual)
|
|
}
|
|
|
|
func Test_flattenedBFloat16VectorsToByteVectors(t *testing.T) {
|
|
flattenedVectors := []byte{0, 1, 2, 3, 4, 5, 6, 7}
|
|
dimension := 2
|
|
|
|
actual := flattenedBFloat16VectorsToByteVectors(flattenedVectors, dimension)
|
|
expected := [][]byte{
|
|
{0, 1, 2, 3},
|
|
{4, 5, 6, 7},
|
|
}
|
|
|
|
assert.Equal(t, expected, actual)
|
|
}
|