Function rangeTest

test/range.cpp:63–106

template<typename T>
void rangeTest(const uint x, const uint y, const uint z, const uint w,
               const uint dim) {
    SUPPORTED_TYPE_CHECK(T);

    dim4 idims(x, y, z, w);

    af_array outArray = 0;

    ASSERT_SUCCESS(af_range(&outArray, idims.ndims(), idims.get(), dim,
                            (af_dtype)dtype_traits<T>::af_type));

    // Get result
    T* outData = new T[idims.elements()];
    ASSERT_SUCCESS(af_get_data_ptr((void*)outData, outArray));

    // Compare result
    for (int w = 0; w < (int)idims[3]; w++) {
        for (int z = 0; z < (int)idims[2]; z++) {
            for (int y = 0; y < (int)idims[1]; y++) {
                for (int x = 0; x < (int)idims[0]; x++) {
                    T val(0);
                    if (dim == 0) {
                        val = x;
                    } else if (dim == 1) {
                        val = y;
                    } else if (dim == 2) {
                        val = z;
                    } else if (dim == 3) {
                        val = w;
                    }
                    dim_t idx = w * idims[0] * idims[1] * idims[2] +
                                z * idims[0] * idims[1] + y * idims[0] + x;

                    ASSERT_EQ(val, outData[idx]) << "at: " << idx;
                }
            }
        }
    }

    // Delete
    delete[] outData;

    if (outArray != 0) af_release_array(outArray);
}

#define RANGE_INIT(desc, x, y, z, w, rep) \
    TYPED_TEST(Range, desc) { rangeTest<TypeParam>(x, y, z, w, rep); }