| 200 | } |
| 201 | |
| 202 | void TestMinAggregateNodes(Client* client) { |
| 203 | AggregatingRequest req("movie", "MinAggregator"); |
| 204 | int64_t ids[10] = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9}; |
| 205 | int32_t segment_ids[10] = {0, 1, 1, 2, 2, 2, 3, 3, 3, 3}; |
| 206 | int32_t num_segments = 4; |
| 207 | req.Set(ids, segment_ids, 10, 4); |
| 208 | |
| 209 | AggregatingResponse res; |
| 210 | Status s = client->Aggregating(&req, &res); |
| 211 | std::cout << "MinAggregateNodes: " << s.ToString() << std::endl; |
| 212 | |
| 213 | int32_t size = res.NumSegments(); |
| 214 | |
| 215 | if (res.EmbeddingDim() > 0) { |
| 216 | int32_t float_num = res.EmbeddingDim(); |
| 217 | const float* floats = res.Embeddings(); |
| 218 | std::cout << "floats: "; |
| 219 | for (int32_t i = 0; i < size * float_num; ++i) { |
| 220 | std::cout << floats[i] << ' '; |
| 221 | } |
| 222 | std::cout << std::endl; |
| 223 | } |
| 224 | } |
| 225 | |
| 226 | void TestProdAggregateNodes(Client* client) { |
| 227 | AggregatingRequest req("movie", "ProdAggregator"); |
no test coverage detected