Function TestVectorUpdate

query/vector/vector_test.go:717–749

(t *testing.T)

}

func TestVectorUpdate(t *testing.T) {
	pred := "vtest"
	dropPredicate(pred)

	setSchema(fmt.Sprintf(vectorSchemaWithIndex, pred, "4", "euclidean"))

	numVectors := 1000
	rdf, vectors := generateRandomVectors(1000, 10, "vtest")
	require.NoError(t, addTriplesToCluster(rdf))

	allVectors, err := queryAllVectorsPred(t, "vtest")
	require.NoError(t, err)

	require.Equal(t, vectors, allVectors)

	updateVectorQuery := func(idx int) {
		triple := strings.Split(rdf, "\n")[idx]
		updatedVec := updateVector(t, triple, "vtest")
		allVectors[idx] = updatedVec

		updatedVectors, err := queryMultipleVectorsUsingSimilarTo(t, allVectors[0], "vtest", 100)
		require.NoError(t, err)

		for _, i := range updatedVectors {
			require.Contains(t, allVectors, i)
		}
	}

	for i := 0; i < 1000; i++ {
		idx := rand.Intn(numVectors)
		updateVectorQuery(idx)
	}
}

func TestVectorWithoutQuote(t *testing.T) {
	pred := "test-ve"