Function queryAllVectorsPred

query/vector/vector_test.go:273–306

(t *testing.T, pred string)

}

func queryAllVectorsPred(t *testing.T, pred string) ([][]float32, error) {
	vectorQuery := fmt.Sprintf(`
	 {
		 vector(func: has(%v)) {
				uid
				%v
		  }
	 }`, pred, pred)

	resp, err := client.Query(vectorQuery)
	require.NoError(t, err)

	type VectorData struct {
		UID   string    `json:"uid"`
		VTest []float32 `json:"vtest"`
	}

	type Data struct {
		Vector []VectorData `json:"vector"`
	}

	var data Data

	err = json.Unmarshal([]byte(resp.Json), &data)
	if err != nil {
		return [][]float32{}, err
	}

	var vectors [][]float32
	for _, vector := range data.Vector {
		vectors = append(vectors, vector.VTest)
	}
	return vectors, nil
}

func generateRandomVector(size int) []float32 {
	vector := make([]float32, size)