Function TEST

test/getting_started.cpp:20–81

using std::vector;

TEST(GettingStarted, SNIPPET_getting_started_gen) {
    //! [ex_getting_started_constructors]
    // Arrays may be created using the array constructor and dimensioned
    // as 1D, 2D, 3D; however, the values in these arrays will be undefined
    array undefined_1D(100);         // 1D array with 100 elements
    array undefined_2D(10, 100);     // 2D array of size 10 x 100
    array undefined_3D(10, 10, 10);  // 3D array of size 10 x 10 x 10
    //! [ex_getting_started_constructors]

    //! [ex_getting_started_gen]
    // Generate an array of size three filled with zeros.
    // If no data type is specified, ArrayFire defaults to f32.
    // The constant function generates the data on the device.
    array zeros = constant(0, 3);

    // Generate a 1x4 array of uniformly distributed [0,1] random numbers
    // The randu function generates the data on the device.
    array rand1 = randu(1, 4);

    // Generate a 2x2 array (or matrix, if you prefer) of random numbers
    // sampled from a normal distribution.
    // The randn function generates data on the device.
    array rand2 = randn(2, 2);

    // Generate a 3x3 identity matrix. The data is generated on the device.
    array iden = identity(3, 3);

    // Lastly, create a 2x1 array (column vector) of uniformly distributed
    // 32-bit complex numbers (c32 data type):
    array randcplx = randu(2, 1, c32);
    //! [ex_getting_started_gen]

    {
        vector<float> output;
        output.resize(zeros.elements());
        zeros.host(&output.front());
        ASSERT_EQ(f32, zeros.type());
        for (dim_t i = 0; i < zeros.elements(); i++)
            ASSERT_FLOAT_EQ(0, output[i]);
    }

    if (!noDoubleTests(f64)) {
        array ones = constant(1, 3, 2, f64);
        vector<double> output(ones.elements());
        ones.host(&output.front());
        ASSERT_EQ(f64, ones.type());
        for (dim_t i = 0; i < ones.elements(); i++)
            ASSERT_FLOAT_EQ(1, output[i]);
    }

    {
        vector<float> output;
        output.resize(iden.elements());
        iden.host(&output.front());
        for (dim_t i = 0; i < iden.dims(0); i++)
            for (dim_t j = 0; j < iden.dims(1); j++)
                if (i == j)
                    ASSERT_FLOAT_EQ(1, output[i * iden.dims(0) + j]);