Function TestVectorDelete

query/vector/vector_test.go:661–715

(t *testing.T)

}

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

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

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

	query := `{
		 vector(func: has(vtest)) {
				count(uid)
			 }
	 }`

	result := processQueryNoErr(t, query)
	require.JSONEq(t, fmt.Sprintf(`{"data": {"vector":[{"count":%d}]}}`, numVectors), result)

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

	require.Equal(t, vectors, allVectors)

	triples := strings.Split(rdf, "\n")

	deleteTriple := func(idx int) string {
		triple := triples[idx]

		deleteTriplesInCluster(triple)
		uid := strings.Split(triple, " ")[0]
		query = fmt.Sprintf(`{
		  vector(func: uid(%s)) {
		   vtest
		  }
		}`, uid[1:len(uid)-1])

		result = processQueryNoErr(t, query)
		require.JSONEq(t, `{"data": {"vector":[]}}`, result)
		return triple

	}

	for i := 0; i < len(triples)-2; i++ {
		triple := deleteTriple(i)
		vector, err := querySingleVector(t, strings.Split(triple, `"`)[1], "vtest")
		require.NoError(t, err)
		require.Contains(t, allVectors, vector)
	}

	triple := deleteTriple(len(triples) - 2)
	// after deleteing all vectors, we should get an empty array of vectors in response when we do silimar_to query
	_, err = querySingleVectorError(t, strings.Split(triple, `"`)[1], "vtest", false)
	require.NoError(t, err)
}

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