/*******************************************************
 * Copyright (c) 2014, ArrayFire
 * All rights reserved.
 *
 * This file is distributed under 3-clause BSD license.
 * The complete license agreement can be obtained at:
 * http://arrayfire.com/licenses/BSD-3-Clause
 ********************************************************/

#include <gtest/gtest.h>
#include <testHelpers.hpp>
#include <af/arith.h>
#include <af/array.h>
#include <af/data.h>

using af::array;
using af::dtype_traits;
using af::randu;
using std::vector;

template<typename T>
class Compare : public ::testing::Test {};

typedef ::testing::Types<float, double, uint, int, intl, uintl, uchar, short,
                         ushort>
    TestTypes;
TYPED_TEST_CASE(Compare, TestTypes);

#define COMPARE(OP, Name)                                   \
    TYPED_TEST(Compare, Test_##Name) {                      \
        typedef TypeParam T;                                \
        SUPPORTED_TYPE_CHECK(T);                            \
        const int num = 1 << 20;                            \
        af_dtype ty   = (af_dtype)dtype_traits<T>::af_type; \
        array a       = randu(num, ty);                     \
        array b       = randu(num, ty);                     \
        array c       = a OP b;                             \
        vector<T> ha(num), hb(num);                         \
        vector<char> hc(num);                               \
        a.host(&ha[0]);                                     \
        b.host(&hb[0]);                                     \
        c.host(&hc[0]);                                     \
        for (int i = 0; i < num; i++) {                     \
            char res = ha[i] OP hb[i];                      \
            ASSERT_EQ((int)res, (int)hc[i]);                \
        }                                                   \
    }

COMPARE(==, eq)
COMPARE(!=, ne)
COMPARE(<=, le)
COMPARE(>=, ge)
COMPARE(<, lt)
COMPARE(>, gt)