Function TEST_P

test/nearest_neighbour.cpp:454–472

                         testNameGenerator<KNearestNeighborsTest>);

TEST_P(NearestNeighborsTest, SingleQTests) {
    UNSUPPORTED_BACKEND(AF_BACKEND_ONEAPI);
    nearest_neighbors_params params = GetParam();
    array query = array(params.qdims_, params.query_.data());
    array train = array(params.tdims_, params.train_.data());

    const int k        = params.k_;
    const int feat_dim = params.feat_dim_;

    array indices, distances;

    nearestNeighbour(indices, distances, query, train, feat_dim, k, AF_SSD);

    array indices_gold(params.idims_, params.indices_.data());
    array distances_gold(params.ddims_, params.dists_.data());

    ASSERT_ARRAYS_EQ(indices_gold, indices);
    ASSERT_ARRAYS_NEAR(distances_gold, distances, 1e-5);
}

TEST_P(KNearestNeighborsTest, SingleQTests) {
    UNSUPPORTED_BACKEND(AF_BACKEND_ONEAPI);