arrayfire/test/corrcoef.cpp at master · arrayfire/arrayfire · GitHub

Repository navigation

History

93 lines (78 loc) · 2.76 KB

Raw

/*******************************************************

* 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 <arrayfire.h>

#include <gtest/gtest.h>

#include <testHelpers.hpp>

#include <af/dim4.hpp>

#include <af/traits.hpp>

#include <algorithm>

#include <ctime>

#include <string>

#include <vector>

using af::array;

using af::cfloat;

using af::corrcoef;

using af::dim4;

using std::string;

using std::vector;

template<typename T>

class CorrelationCoefficient : public ::testing::Test {

public:

virtual void SetUp() {}

};

// create a list of types to be tested

typedef ::testing::Types<float, double, int, uint, intl, uintl, char, schar,

uchar>

TestTypes;

// register the type list

TYPED_TEST_SUITE(CorrelationCoefficient, TestTypes);

template<typename T>

struct f32HelperType {

typedef

typename cond_type<is_same_type<T, double>::value, double, float>::type

type;

};

template<typename T>

struct c32HelperType {

typedef typename cond_type<is_same_type<T, cfloat>::value, cfloat,

typename f32HelperType<T>::type>::type type;

};

template<typename T>

struct elseType {

typedef typename cond_type<is_same_type<T, uintl>::value ||

is_same_type<T, intl>::value,

double, T>::type type;

};

template<typename T>

struct ccOutType {

typedef typename cond_type<

is_same_type<T, float>::value || is_same_type<T, int>::value ||

is_same_type<T, uint>::value || is_same_type<T, schar>::value ||

is_same_type<T, uchar>::value || is_same_type<T, short>::value ||

is_same_type<T, ushort>::value || is_same_type<T, char>::value,

float, typename elseType<T>::type>::type type;

};

TYPED_TEST(CorrelationCoefficient, All) {

typedef typename ccOutType<TypeParam>::type outType;

SUPPORTED_TYPE_CHECK(TypeParam);

SUPPORTED_TYPE_CHECK(outType);

vector<dim4> numDims;

vector<vector<int>> in;

vector<vector<float>> tests;

readTestsFromFile<int, float>(

string(TEST_DIR "/corrcoef/mat_10x10_scalar.test"), numDims, in, tests);

vector<TypeParam> input1(in[0].begin(), in[0].end());

vector<TypeParam> input2(in[1].begin(), in[1].end());

array a(numDims[0], &(input1.front()));

array b(numDims[1], &(input2.front()));

outType c = corrcoef<outType>(a, b);

vector<outType> currGoldBar(tests[0].begin(), tests[0].end());

ASSERT_NEAR(::real(currGoldBar[0]), ::real(c), 1.0e-3);

ASSERT_NEAR(::imag(currGoldBar[0]), ::imag(c), 1.0e-3);

}

arrayfire/test/corrcoef.cpp at master · arrayfire/arrayfire · GitHub

Repository navigation

Expand file tree

Breadcrumbs

corrcoef.cpp

Latest commit

History

Breadcrumbs

corrcoef.cpp

File metadata and controls