Function TestProdAggregateNodes

graphlearn/src/service/test/client_test.cpp:226–248

}

void TestProdAggregateNodes(Client* client) {
  AggregatingRequest req("movie", "ProdAggregator");
  int64_t ids[10] = {0,  1, 2,  3, 4, 5,  6, 7, 8, 9};
  int32_t segment_ids[10] = {0, 1, 1, 2, 2, 2, 3, 3, 3, 3};
  int32_t num_segments = 4;
  req.Set(ids, segment_ids, 10, 4);

  AggregatingResponse res;
  Status s = client->Aggregating(&req, &res);
  std::cout << "ProdAggregateNodes: " << s.ToString() << std::endl;

  int32_t size = res.NumSegments();

  if (res.EmbeddingDim() > 0) {
    int32_t float_num = res.EmbeddingDim();
    const float* floats = res.Embeddings();
    std::cout << "floats: ";
    for (int32_t i = 0; i < size * float_num; ++i) {
      std::cout << floats[i] << ' ';
    }
    std::cout << std::endl;
  }
}

void TestRandomSampleNeighbors(Client* client) {
  SamplingRequest req("click", "RandomSampler", 3);