Function testVectorGraphQlMutationAndQuery

query/vector/vector_graphql_test.go:219–247

(t *testing.T, hc *dgraphapi.HTTPClient)

}

func testVectorGraphQlMutationAndQuery(t *testing.T, hc *dgraphapi.HTTPClient) {
	var vectors [][]float32
	numProjects := 100
	projects := generateProjects(numProjects)
	for _, project := range projects {
		vectors = append(vectors, project.TitleV)
		addProject(t, hc, project)
	}

	for _, project := range projects {
		p := queryProjectUsingTitle(t, hc, project.Title)
		require.Equal(t, project.Title, p.Title)
		require.Equal(t, project.TitleV, p.TitleV)
	}

	for _, project := range projects {
		p := queryProjectUsingTitle(t, hc, project.Title)
		require.Equal(t, project.Title, p.Title)
		require.Equal(t, project.TitleV, p.TitleV)
	}

	// query similar project by embedding
	for _, project := range projects {
		similarProjects := queryProjectsSimilarByEmbedding(t, hc, project.TitleV, numProjects)
		for _, similarVec := range similarProjects {
			require.Contains(t, vectors, similarVec.TitleV)
		}
	}
}