Function testData

test/confidence_connected.cpp:115–157

template<typename T>
void testData(CCCTestParams params) {
    SUPPORTED_TYPE_CHECK(T);

    vector<dim4> numDims;
    vector<vector<T>> in;
    vector<vector<T>> tests;

    string file = string(TEST_DIR) + "/confidence_cc/" + string(params.prefix) +
                  "_" + to_string(params.radius) + "_" +
                  to_string(params.multiplier) + ".test";
    readTests<T, T, int>(file, numDims, in, tests);

    dim4 dims         = numDims[0];
    af_array inArray  = 0;
    af_array seedxArr = 0, seedyArr = 0;

    vector<uint> seedCoords(in[1].begin(), in[1].end());
    const unsigned *seedxy = seedCoords.data();

    dim4 seedDims(1);
    ASSERT_SUCCESS(af_create_array(&seedxArr, seedxy + 0, seedDims.ndims(),
                                   seedDims.get(), u32));
    ASSERT_SUCCESS(af_create_array(&seedyArr, seedxy + 1, seedDims.ndims(),
                                   seedDims.get(), u32));
    ASSERT_SUCCESS(af_create_array(&inArray, &(in[0].front()), dims.ndims(),
                                   dims.get(),
                                   (af_dtype)af::dtype_traits<T>::af_type));

    af_array outArray = 0;
    ASSERT_SUCCESS(af_confidence_cc(&outArray, inArray, seedxArr, seedyArr,
                                    params.radius, params.multiplier,
                                    params.iterations, params.replace));
    int device = 0;
    ASSERT_SUCCESS(af_get_device(&device));
    ASSERT_SUCCESS(af_sync(device));

    ASSERT_VEC_ARRAY_EQ(tests[0], dims, outArray);

    ASSERT_SUCCESS(af_release_array(inArray));
    ASSERT_SUCCESS(af_release_array(outArray));
    ASSERT_SUCCESS(af_release_array(seedxArr));
    ASSERT_SUCCESS(af_release_array(seedyArr));
}

class ConfidenceConnectedDataTest
    : public testing::TestWithParam<CCCTestParams> {};