Function TestTopkSampleNeighbors

graphlearn/src/service/test/client_test.cpp:284–299

}

void TestTopkSampleNeighbors(Client* client) {
  SamplingRequest req("click", "TopkSampler", 3);
  int64_t ids[10] = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9};
  req.Set(ids, 10);

  SamplingResponse res;
  Status s = client->Sampling(&req, &res);
  std::cout << "SampleTopkNeighbors: " << s.ToString() << std::endl;

  const int64_t* nbrs = res.GetNeighborIds();
  int32_t size = res.TotalNeighborCount();
  std::cout << "TotalNeighborCount: " << size << std::endl;
  for (int32_t i = 0; i < size; ++i) {
    std::cout << nbrs[i] << std::endl;
  }
}

void TestFullSampleNeighbors(Client* client) {
  SamplingRequest req("click", "FullSampler", 3);