Function querySingleVectorError

query/vector/vector_test.go:229–267

(t *testing.T, vector, pred string, validateError bool)

}

func querySingleVectorError(t *testing.T, vector, pred string, validateError bool) ([]float32, error) {

	vectorQuery := fmt.Sprintf(`
	 {
		 vector(func: similar_to(%v, 1, "%v")) {
				uid
				%v
		  }
	 }`, pred, vector, pred)

	resp, err := client.Query(vectorQuery)
	if validateError {
		require.NoError(t, err)
	} else if err != nil {
		return []float32{}, 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
	}

	if len(data.Vector) == 0 {
		return []float32{}, nil
	}

	return data.Vector[0].VTest, nil
}

func querySingleVector(t *testing.T, vector, pred string) ([]float32, error) {
	return querySingleVectorError(t, vector, pred, true)