| 360 | : public ::testing::TestWithParam<nearest_neighbors_params> {}; |
| 361 | |
| 362 | nearest_neighbors_params single_knn_data(const string testname, |
| 363 | const int nquery, const int ntrain, |
| 364 | const int nfeat, const int k, |
| 365 | const int feat_dim) { |
| 366 | array indices, dists; |
| 367 | array query, train; |
| 368 | if (feat_dim == 0) { |
| 369 | query = constant(0, nfeat, nquery); |
| 370 | train = constant(1, nfeat, ntrain); |
| 371 | } else { |
| 372 | query = constant(0, nquery, nfeat); |
| 373 | train = constant(1, ntrain, nfeat); |
| 374 | } |
| 375 | |
| 376 | indices = constant(0, k, nquery, u32); |
| 377 | dists = constant(nfeat, k, nquery); |
| 378 | |
| 379 | return nearest_neighbors_params(testname, k, feat_dim, query, train, |
| 380 | indices, dists); |
| 381 | } |
| 382 | |
| 383 | nearest_neighbors_params knn_data(const string testname, const int nquery, |
| 384 | const int ntrain, const int nfeat, |
no test coverage detected