e151e136d6
Closes #18 Co-authored-by: Jules Pénuchot <jules@penuchot.com> See merge request oss/rotgen!50
75 lines
3.3 KiB
C++
75 lines
3.3 KiB
C++
//==================================================================================================
|
|
/*
|
|
ROTGEN - Runtime Overlay for Eigen
|
|
Copyright : CODE RECKONS
|
|
SPDX-License-Identifier: BSL-1.0
|
|
*/
|
|
//==================================================================================================
|
|
#include <rotgen/rotgen.hpp>
|
|
|
|
#include "unit/common/references.hpp"
|
|
#include "unit/tests.hpp"
|
|
|
|
TTS_CASE_TPL("rowwise API", rotgen::tests::types)
|
|
|
|
<typename T, typename O>(tts::type<tts::types<T, O>>)
|
|
{
|
|
using eigen_mat_t =
|
|
Eigen::Matrix<T, Eigen::Dynamic, Eigen::Dynamic, O::value>;
|
|
eigen_mat_t eigen_ref = eigen_mat_t::Random(4, 4);
|
|
auto eigen_ref_rw = eigen_ref.rowwise();
|
|
|
|
rotgen::matrix<T, rotgen::Dynamic, rotgen::Dynamic, O::value> rotgen_mat(4,
|
|
4);
|
|
rotgen::tests::prepare([&](auto r, auto c) { return eigen_ref(r, c); },
|
|
rotgen_mat);
|
|
|
|
auto rotgen_rw = rotgen::rowwise(rotgen_mat);
|
|
|
|
for (rotgen::Index i = 0; i < rows(rotgen_mat); ++i)
|
|
TTS_EQUAL(rotgen::sum(rotgen_rw)(i), eigen_ref_rw.sum()(i));
|
|
for (rotgen::Index i = 0; i < rows(rotgen_mat); ++i)
|
|
TTS_EQUAL(rotgen::mean(rotgen_rw)(i), eigen_ref_rw.mean()(i));
|
|
for (rotgen::Index i = 0; i < rows(rotgen_mat); ++i)
|
|
TTS_EQUAL(rotgen::prod(rotgen_rw)(i), eigen_ref_rw.prod()(i));
|
|
for (rotgen::Index i = 0; i < rows(rotgen_mat); ++i)
|
|
TTS_EQUAL(rotgen::maxCoeff(rotgen_rw)(i), eigen_ref_rw.maxCoeff()(i));
|
|
for (rotgen::Index i = 0; i < rows(rotgen_mat); ++i)
|
|
TTS_EQUAL(rotgen::minCoeff(rotgen_rw)(i), eigen_ref_rw.minCoeff()(i));
|
|
for (rotgen::Index i = 0; i < rows(rotgen_mat); ++i)
|
|
TTS_EQUAL(rotgen::squaredNorm(rotgen_rw)(i), eigen_ref_rw.squaredNorm()(i));
|
|
for (rotgen::Index i = 0; i < rows(rotgen_mat); ++i)
|
|
TTS_EQUAL(rotgen::norm(rotgen_rw)(i), eigen_ref_rw.norm()(i));
|
|
};
|
|
|
|
TTS_CASE_TPL("colwise API", rotgen::tests::types)
|
|
|
|
<typename T, typename O>(tts::type<tts::types<T, O>>)
|
|
{
|
|
using eigen_mat_t =
|
|
Eigen::Matrix<T, Eigen::Dynamic, Eigen::Dynamic, O::value>;
|
|
eigen_mat_t eigen_ref = eigen_mat_t::Random(4, 4);
|
|
auto eigen_ref_rw = eigen_ref.colwise();
|
|
|
|
rotgen::matrix<T, rotgen::Dynamic, rotgen::Dynamic, O::value> rotgen_mat(4,
|
|
4);
|
|
rotgen::tests::prepare([&](auto r, auto c) { return eigen_ref(r, c); },
|
|
rotgen_mat);
|
|
|
|
auto rotgen_rw = rotgen::colwise(rotgen_mat);
|
|
|
|
for (rotgen::Index i = 0; i < cols(rotgen_mat); ++i)
|
|
TTS_EQUAL(rotgen::sum(rotgen_rw)(i), eigen_ref_rw.sum()(i));
|
|
for (rotgen::Index i = 0; i < cols(rotgen_mat); ++i)
|
|
TTS_EQUAL(rotgen::mean(rotgen_rw)(i), eigen_ref_rw.mean()(i));
|
|
for (rotgen::Index i = 0; i < cols(rotgen_mat); ++i)
|
|
TTS_EQUAL(rotgen::prod(rotgen_rw)(i), eigen_ref_rw.prod()(i));
|
|
for (rotgen::Index i = 0; i < cols(rotgen_mat); ++i)
|
|
TTS_EQUAL(rotgen::maxCoeff(rotgen_rw)(i), eigen_ref_rw.maxCoeff()(i));
|
|
for (rotgen::Index i = 0; i < cols(rotgen_mat); ++i)
|
|
TTS_EQUAL(rotgen::minCoeff(rotgen_rw)(i), eigen_ref_rw.minCoeff()(i));
|
|
for (rotgen::Index i = 0; i < cols(rotgen_mat); ++i)
|
|
TTS_EQUAL(rotgen::squaredNorm(rotgen_rw)(i), eigen_ref_rw.squaredNorm()(i));
|
|
for (rotgen::Index i = 0; i < cols(rotgen_mat); ++i)
|
|
TTS_EQUAL(rotgen::norm(rotgen_rw)(i), eigen_ref_rw.norm()(i));
|
|
};
|