Function TestFullSampleNeighbors

graphlearn/src/service/test/client_test.cpp:301–322

}

void TestFullSampleNeighbors(Client* client) {
  SamplingRequest req("click", "FullSampler", 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 << "FullSampleNeighbors: " << 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;
  }

  int32_t batch_size = res.BatchSize();
  const int32_t* degrees = res.GetDegrees();
  for (int32_t i = 0; i < batch_size; ++i) {
    std::cout << degrees[i] << std::endl;
  }
}

void TestNodeWeightNegativeSample(Client* client) {
  SamplingRequest req("user", "NodeWeightNegativeSampler", 3);