Function arrayIndexTest

test/index.cpp:720–754

template<typename T>
void arrayIndexTest(string pTestFile, int dim) {
    SUPPORTED_TYPE_CHECK(T);

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

    readTests<T, T, int>(pTestFile, numDims, in, tests);

    dim4 dims0        = numDims[0];
    dim4 dims1        = numDims[1];
    af_array outArray = 0;
    af_array inArray  = 0;
    af_array idxArray = 0;

    ASSERT_SUCCESS(af_create_array(&inArray, &(in[0].front()), dims0.ndims(),
                                   dims0.get(),
                                   (af_dtype)dtype_traits<T>::af_type));

    ASSERT_SUCCESS(af_create_array(&idxArray, &(in[1].front()), dims1.ndims(),
                                   dims1.get(),
                                   (af_dtype)dtype_traits<T>::af_type));

    ASSERT_SUCCESS(af_lookup(&outArray, inArray, idxArray, dim));

    vector<T> currGoldBar = tests[0];
    dim4 goldDims         = dims0;
    goldDims[dim]         = dims1[0];

    ASSERT_VEC_ARRAY_EQ(currGoldBar, goldDims, outArray);

    ASSERT_SUCCESS(af_release_array(inArray));
    ASSERT_SUCCESS(af_release_array(idxArray));
    ASSERT_SUCCESS(af_release_array(outArray));
}

TYPED_TEST(lookup, Dim0) {
    arrayIndexTest<TypeParam>(string(TEST_DIR "/arrayindex/dim0.test"), 0);