common

Documentation

Index

• Constants
• Variables
• func CheckErr(t *testing.T, actualErr error, expErrNil bool, expErrorMsg ...string)
• func CheckIndex(t *testing.T, actualIdxDesc client.IndexDescription, idx index.Index, ...)
• func CheckInsertResult(t *testing.T, expIDs column.Column, insertRes client.InsertResult)
• func CheckOutputFields(t *testing.T, expFields []string, actualColumns []column.Column)
• func CheckQueryResult(t *testing.T, expColumns []column.Column, actualColumns []column.Column)
• func CheckSearchResult(t *testing.T, actualSearchResults []client.ResultSet, expNq int, expTopK int)
• func EqualArrayColumn(t *testing.T, columnA column.Column, columnB column.Column)
• func EqualColumn(t *testing.T, columnA column.Column, columnB column.Column)
• func GenBFloat16Vector(dim int) []byte
• func GenBinaryVector(dim int) []byte
• func GenFloat16Vector(dim int) []byte
• func GenFloatVector(dim int) []float32
• func GenInvalidNames() []string
• func GenLongString(n int) string
• func GenRandomString(prefix string, n int) string
• func GenSparseVector(maxLen int) entity.SparseEmbedding
• func GenValidNames() []string
• func TNewCheckIndexOpt(totalRows int64) *checkIndexOpt
• type InvalidExprStruct

Constants

const (
	DefaultInt8FieldName        = "int8"
	DefaultInt16FieldName       = "int16"
	DefaultInt32FieldName       = "int32"
	DefaultInt64FieldName       = "int64"
	DefaultBoolFieldName        = "bool"
	DefaultFloatFieldName       = "float"
	DefaultDoubleFieldName      = "double"
	DefaultVarcharFieldName     = "varchar"
	DefaultJSONFieldName        = "json"
	DefaultArrayFieldName       = "array"
	DefaultFloatVecFieldName    = "floatVec"
	DefaultBinaryVecFieldName   = "binaryVec"
	DefaultFloat16VecFieldName  = "fp16Vec"
	DefaultBFloat16VecFieldName = "bf16Vec"
	DefaultSparseVecFieldName   = "sparseVec"
	DefaultDynamicNumberField   = "dynamicNumber"
	DefaultDynamicStringField   = "dynamicString"
	DefaultDynamicBoolField     = "dynamicBool"
	DefaultDynamicListField     = "dynamicList"
	DefaultBoolArrayField       = "boolArray"
	DefaultInt8ArrayField       = "int8Array"
	DefaultInt16ArrayField      = "int16Array"
	DefaultInt32ArrayField      = "int32Array"
	DefaultInt64ArrayField      = "int64Array"
	DefaultFloatArrayField      = "floatArray"
	DefaultDoubleArrayField     = "doubleArray"
	DefaultVarcharArrayField    = "varcharArray"
)

cost default field name

const (
	RowCount       = "row_count"
	DefaultTimeout = 120
	DefaultDim     = 128
	DefaultShards  = int32(2)
	DefaultNb      = 3000
	DefaultNq      = 5
	DefaultLimit   = 10
	TestCapacity   = 100 // default array field capacity
	TestMaxLen     = 100 // default varchar field max length
)

cost for test cases

const (
	MaxPartitionNum         = 1024
	DefaultDynamicFieldName = "$meta"
	QueryCountFieldName     = "count(*)"
	DefaultPartition        = "_default"
	DefaultIndexName        = "_default_idx_102"
	DefaultIndexNameBinary  = "_default_idx_100"
	DefaultRgName           = "__default_resource_group"
	DefaultDb               = "default"
	MaxDim                  = 32768
	MaxLength               = int64(65535)
	MaxCollectionNameLen    = 255
	DefaultRgCapacity       = 1000000
	RetentionDuration       = 40   // common.retentionDuration
	MaxCapacity             = 4096 // max array capacity
	DefaultPartitionNum     = 16   // default num_partitions
	MaxTopK                 = 16384
	MaxVectorFieldNum       = 4
	MaxShardNum             = 16
)

const default value from milvus config

const (
	IndexStateIndexStateNone index.IndexState = 0
	IndexStateUnissued       index.IndexState = 1
	IndexStateInProgress     index.IndexState = 2
	IndexStateFinished       index.IndexState = 3
	IndexStateFailed         index.IndexState = 4
	IndexStateRetry          index.IndexState = 5
)

const (
	DatabaseMaxCollections   = "database.max.collections"
	DatabaseResourceGroups   = "database.resource_groups"
	DatabaseReplicaNumber    = "database.replica.number"
	DatabaseForceDenyWriting = "database.force.deny.writing"
	DatabaseForceDenyReading = "database.force.deny.reading"
	DatabaseDiskQuotaMb      = "database.diskQuota.mb"
)

part database properties

Variables

var InvalidExpressions = []InvalidExprStruct{
	{Expr: "id in [0]", ErrNil: true, ErrMsg: "fieldName(id) not found"},
	{Expr: "int64 in not [0]", ErrNil: false, ErrMsg: "cannot parse expression"},
	{Expr: "int64 < floatVec", ErrNil: false, ErrMsg: "not supported"},
	{Expr: "floatVec in [0]", ErrNil: false, ErrMsg: "cannot be casted to FloatVector"},
	{Expr: fmt.Sprintf("%s == 1", DefaultJSONFieldName), ErrNil: true, ErrMsg: ""},
	{Expr: fmt.Sprintf("%s like 'a%%' ", DefaultJSONFieldName), ErrNil: true, ErrMsg: ""},
	{Expr: fmt.Sprintf("%s like `a%%` ", DefaultJSONFieldName), ErrNil: false, ErrMsg: "cannot parse expression"},
	{Expr: fmt.Sprintf("%s > 1", DefaultDynamicFieldName), ErrNil: true, ErrMsg: ""},
	{Expr: fmt.Sprintf("%s[\"dynamicList\"] == [2, 3]", DefaultDynamicFieldName), ErrNil: true, ErrMsg: ""},
	{Expr: fmt.Sprintf("%s['a'] == [2, 3]", DefaultJSONFieldName), ErrNil: true, ErrMsg: ""},
	{Expr: fmt.Sprintf("%s['number'] == [2, 3]", DefaultJSONFieldName), ErrNil: true, ErrMsg: ""},
	{Expr: fmt.Sprintf("%s[0] == [2, 3]", DefaultJSONFieldName), ErrNil: true, ErrMsg: ""},
	{Expr: fmt.Sprintf("json_contains (%s['number'], 2)", DefaultJSONFieldName), ErrNil: true, ErrMsg: ""},
	{Expr: fmt.Sprintf("json_contains (%s['list'], [2])", DefaultJSONFieldName), ErrNil: true, ErrMsg: ""},
	{Expr: fmt.Sprintf("json_contains_all (%s['list'], 2)", DefaultJSONFieldName), ErrNil: false, ErrMsg: "operation element must be an array"},
	{Expr: fmt.Sprintf("JSON_CONTAINS_ANY (%s['list'], 2)", DefaultJSONFieldName), ErrNil: false, ErrMsg: "operation element must be an array"},
	{Expr: fmt.Sprintf("json_contains_aby (%s['list'], 2)", DefaultJSONFieldName), ErrNil: false, ErrMsg: "function json_contains_aby(json, int64_t) not found."},
	{Expr: fmt.Sprintf("json_contains_aby (%s['list'], 2)", DefaultJSONFieldName), ErrNil: false, ErrMsg: "function json_contains_aby(json, int64_t) not found."},
	{Expr: fmt.Sprintf("%s[-1] > %d", DefaultInt8ArrayField, TestCapacity), ErrNil: false, ErrMsg: "cannot parse expression"},
	{Expr: fmt.Sprintf("%s[-1] > 1", DefaultJSONFieldName), ErrNil: false, ErrMsg: "invalid expression"},
}

Functions

func CheckErr

func CheckErr(t *testing.T, actualErr error, expErrNil bool, expErrorMsg ...string)

func CheckIndex

func CheckIndex(t *testing.T, actualIdxDesc client.IndexDescription, idx index.Index, opt *checkIndexOpt)

func CheckInsertResult

func CheckInsertResult(t *testing.T, expIDs column.Column, insertRes client.InsertResult)

CheckInsertResult check insert result, ids len (insert count), ids data (pks, but no auto ids)

func CheckOutputFields

func CheckOutputFields(t *testing.T, expFields []string, actualColumns []column.Column)

CheckOutputFields check query output fields

func CheckQueryResult

func CheckQueryResult(t *testing.T, expColumns []column.Column, actualColumns []column.Column)

CheckQueryResult check query result, column name, type and field

func CheckSearchResult

func CheckSearchResult(t *testing.T, actualSearchResults []client.ResultSet, expNq int, expTopK int)

CheckSearchResult check search result, check nq, topk, ids, score

func EqualArrayColumn

func EqualArrayColumn(t *testing.T, columnA column.Column, columnB column.Column)

EqualColumn assert field data is equal of two columns

func EqualColumn

func EqualColumn(t *testing.T, columnA column.Column, columnB column.Column)

EqualColumn assert field data is equal of two columns

func GenBFloat16Vector

func GenBFloat16Vector(dim int) []byte

func GenBinaryVector

func GenBinaryVector(dim int) []byte

func GenFloat16Vector

func GenFloat16Vector(dim int) []byte

func GenFloatVector

func GenFloatVector(dim int) []float32

func GenInvalidNames

func GenInvalidNames() []string

func GenLongString

func GenLongString(n int) string

GenLongString gen invalid long string

func GenRandomString

func GenRandomString(prefix string, n int) string

func GenSparseVector

func GenSparseVector(maxLen int) entity.SparseEmbedding

func GenValidNames

func GenValidNames() []string

func TNewCheckIndexOpt

func TNewCheckIndexOpt(totalRows int64) *checkIndexOpt

Types

type InvalidExprStruct

type InvalidExprStruct struct {
	Expr   string
	ErrNil bool
	ErrMsg string
}

InvalidExprStruct invalid expr