SemanticSearchPOC / test /acceptance /objects /skip_vector_index_test.go
KevinStephenson
Adding in weaviate code
b110593
raw
history blame
1.99 kB
// _ _
// __ _____ __ ___ ___ __ _| |_ ___
// \ \ /\ / / _ \/ _` \ \ / / |/ _` | __/ _ \
// \ V V / __/ (_| |\ V /| | (_| | || __/
// \_/\_/ \___|\__,_| \_/ |_|\__,_|\__\___|
//
// Copyright © 2016 - 2024 Weaviate B.V. All rights reserved.
//
// CONTACT: [email protected]
//
package test
import (
"testing"
"github.com/go-openapi/strfmt"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"github.com/weaviate/weaviate/client/objects"
"github.com/weaviate/weaviate/entities/models"
"github.com/weaviate/weaviate/test/helper"
)
func TestSkipVectorIndex(t *testing.T) {
// Import a class with vectorizer 'none' and 'skipVectorIndex: true', import
// objects without vectors.
t.Run("create schema", func(t *testing.T) {
createObjectClass(t, &models.Class{
Class: "TestSkipVectorIndex",
VectorIndexConfig: map[string]interface{}{
"skip": true,
},
Vectorizer: "none",
Properties: []*models.Property{
{
Name: "name",
DataType: []string{"text"},
},
},
})
})
id := strfmt.UUID("d1d58565-3c9b-4ca6-ac7f-43f739700a1d")
t.Run("create object", func(t *testing.T) {
params := objects.NewObjectsCreateParams().WithBody(
&models.Object{
ID: id,
Class: "TestSkipVectorIndex",
Properties: map[string]interface{}{"name": "Jane Doe"},
})
_, err := helper.Client(t).Objects.ObjectsCreate(params, nil)
require.Nil(t, err, "creation should succeed")
})
t.Run("get obj by ID", func(t *testing.T) {
params := objects.NewObjectsGetParams().WithID(id)
obj, err := helper.Client(t).Objects.ObjectsGet(params, nil)
require.Nil(t, err, "object can be retrieved by id")
assert.Equal(t, "Jane Doe", obj.Payload.Properties.(map[string]interface{})["name"].(string))
})
t.Run("tear down", func(t *testing.T) {
deleteObjectClass(t, "TestSkipVectorIndex")
})
}