Function DimCheck2D

test/index.cpp:328–374

template<typename T>
void DimCheck2D(const vector<vector<af_seq>> &seqs, string TestFile,
                size_t NDims) {
    SUPPORTED_TYPE_CHECK(T);

    vector<dim4> numDims;

    vector<vector<T>> hData;
    vector<vector<T>> tests;
    readTests<T, T, int>(TestFile, numDims, hData, tests);
    dim4 dimensions = numDims[0];

    af_array a = 0;
    ASSERT_SUCCESS(af_create_array(&a, &(hData[0].front()), NDims,
                                   dimensions.get(),
                                   (af_dtype)dtype_traits<T>::af_type));

    vector<af_array> indexed_arrays(seqs.size(), 0);
    for (size_t i = 0; i < seqs.size(); i++) {
        ASSERT_SUCCESS(
            af_index(&(indexed_arrays[i]), a, NDims, seqs[i].data()));
    }

    vector<T *> h_indexed(seqs.size(), NULL);
    for (size_t i = 0; i < seqs.size(); i++) {
        dim_t elems;
        ASSERT_SUCCESS(af_get_elements(&elems, indexed_arrays[i]));
        h_indexed[i] = new T[elems];
        ASSERT_SUCCESS(
            af_get_data_ptr((void *)h_indexed[i], indexed_arrays[i]));

        T *ptr = h_indexed[i];
        if (false == equal(ptr, ptr + tests[i].size(), tests[i].begin())) {
            cout << "index data: ";
            copy(ptr, ptr + tests[i].size(), ostream_iterator<T>(cout, ", "));
            cout << endl << "file data: ";
            copy(tests[i].begin(), tests[i].end(),
                 ostream_iterator<T>(cout, ", "));
            FAIL() << "indexed_array[" << i << "] FAILED" << endl;
        }
        delete[] h_indexed[i];
    }

    ASSERT_SUCCESS(af_release_array(a));
    for (size_t i = 0; i < indexed_arrays.size(); i++) {
        ASSERT_SUCCESS(af_release_array(indexed_arrays[i]));
    }
}

TYPED_TEST_SUITE(Indexing2D, AllTypes);