Testing functional API only (oss/rotgen#17)
Co-authored-by: Jules Pénuchot <jules@penuchot.com> See merge request oss/rotgen!42
This commit is contained in:
parent
6fa95fb22d
commit
4a7aa08cdb
11 changed files with 192 additions and 192 deletions
|
|
@ -5,81 +5,85 @@
|
|||
SPDX-License-Identifier: BSL-1.0
|
||||
*/
|
||||
//==================================================================================================
|
||||
#pragma once
|
||||
|
||||
#include <rotgen/rotgen.hpp>
|
||||
|
||||
#include "tts.hpp"
|
||||
#include <Eigen/Dense>
|
||||
#include <vector>
|
||||
#include <tuple>
|
||||
|
||||
namespace rotgen::tests
|
||||
{
|
||||
template<typename T> void check_shape_functions(T original)
|
||||
template<typename T> void check_shape_functions(T rotgen_input)
|
||||
{
|
||||
using mat_t =
|
||||
using eigen_mat_t =
|
||||
matrix<typename T::value_type, Dynamic, Dynamic, T::storage_order>;
|
||||
|
||||
mat_t result(original.cols(), original.rows());
|
||||
prepare([&](auto r, auto c) { return original(c, r); }, result);
|
||||
eigen_mat_t eigen_result(rotgen_input.cols(), rotgen_input.rows());
|
||||
prepare([&](auto r, auto c) { return rotgen_input(c, r); }, eigen_result);
|
||||
|
||||
TTS_EQUAL(transpose(original), result);
|
||||
TTS_EQUAL(conjugate(original), original);
|
||||
TTS_EQUAL(adjoint(original), result);
|
||||
TTS_EQUAL(transpose(rotgen_input), eigen_result);
|
||||
TTS_EQUAL(conjugate(rotgen_input), rotgen_input);
|
||||
TTS_EQUAL(adjoint(rotgen_input), eigen_result);
|
||||
|
||||
if constexpr (T::is_defined_static)
|
||||
{
|
||||
if constexpr (T::RowsAtCompileTime == T::ColsAtCompileTime)
|
||||
{
|
||||
mat_t ref = original;
|
||||
transposeInPlace(original);
|
||||
TTS_EQUAL(original, result);
|
||||
eigen_mat_t ref = rotgen_input;
|
||||
transposeInPlace(rotgen_input);
|
||||
TTS_EQUAL(rotgen_input, eigen_result);
|
||||
|
||||
adjointInPlace(original);
|
||||
TTS_EQUAL(original, ref);
|
||||
adjointInPlace(rotgen_input);
|
||||
TTS_EQUAL(rotgen_input, ref);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
if (original.rows() == original.cols())
|
||||
if (rotgen_input.rows() == rotgen_input.cols())
|
||||
{
|
||||
mat_t ref = original;
|
||||
transposeInPlace(original);
|
||||
TTS_EQUAL(original, result);
|
||||
eigen_mat_t eigen_ref = rotgen_input;
|
||||
transposeInPlace(rotgen_input);
|
||||
TTS_EQUAL(rotgen_input, eigen_result);
|
||||
|
||||
adjointInPlace(original);
|
||||
TTS_EQUAL(original, ref);
|
||||
adjointInPlace(rotgen_input);
|
||||
TTS_EQUAL(rotgen_input, eigen_ref);
|
||||
}
|
||||
}
|
||||
|
||||
if constexpr (!rotgen::use_expression_templates)
|
||||
{
|
||||
TTS_EXPECT(verify_rotgen_reentrance(original.transpose()));
|
||||
TTS_EXPECT(verify_rotgen_reentrance(original.conjugate()));
|
||||
TTS_EXPECT(verify_rotgen_reentrance(original.adjoint()));
|
||||
TTS_EXPECT(verify_rotgen_reentrance(transpose(rotgen_input)));
|
||||
TTS_EXPECT(verify_rotgen_reentrance(conjugate(rotgen_input)));
|
||||
TTS_EXPECT(verify_rotgen_reentrance(adjoint(rotgen_input)));
|
||||
}
|
||||
}
|
||||
|
||||
template<typename T> void check_reduction_functions(T const& input)
|
||||
template<typename T> void check_reduction_functions(T const& rotgen_input)
|
||||
{
|
||||
using EigenMatrix =
|
||||
using eigen_mat_t =
|
||||
Eigen::Matrix<typename T::value_type, Eigen::Dynamic, Eigen::Dynamic>;
|
||||
|
||||
EigenMatrix ref(input.rows(), input.cols());
|
||||
prepare([&](auto r, auto c) { return input(r, c); }, ref);
|
||||
eigen_mat_t eigen_ref(rotgen_input.rows(), rotgen_input.cols());
|
||||
prepare([&](auto r, auto c) { return rotgen_input(r, c); }, eigen_ref);
|
||||
|
||||
TTS_ULP_EQUAL(sum(input), ref.sum(), 2);
|
||||
TTS_ULP_EQUAL(prod(input), ref.prod(), 2);
|
||||
TTS_ULP_EQUAL(mean(input), ref.mean(), 2);
|
||||
TTS_EQUAL(trace(input), ref.trace());
|
||||
TTS_EQUAL(minCoeff(input), ref.minCoeff());
|
||||
TTS_EQUAL(maxCoeff(input), ref.maxCoeff());
|
||||
TTS_ULP_EQUAL(sum(rotgen_input), eigen_ref.sum(), 2);
|
||||
TTS_ULP_EQUAL(prod(rotgen_input), eigen_ref.prod(), 2);
|
||||
TTS_ULP_EQUAL(mean(rotgen_input), eigen_ref.mean(), 2);
|
||||
TTS_EQUAL(trace(rotgen_input), eigen_ref.trace());
|
||||
TTS_EQUAL(minCoeff(rotgen_input), eigen_ref.minCoeff());
|
||||
TTS_EQUAL(maxCoeff(rotgen_input), eigen_ref.maxCoeff());
|
||||
|
||||
{
|
||||
int row, col, ref_row, ref_col;
|
||||
|
||||
TTS_EQUAL(minCoeff(input, &row, &col), ref.minCoeff(&ref_row, &ref_col));
|
||||
TTS_EQUAL(minCoeff(rotgen_input, &row, &col),
|
||||
eigen_ref.minCoeff(&ref_row, &ref_col));
|
||||
TTS_EQUAL(row, ref_row);
|
||||
TTS_EQUAL(col, ref_col);
|
||||
|
||||
TTS_EQUAL(maxCoeff(input, &row, &col), ref.maxCoeff(&ref_row, &ref_col));
|
||||
TTS_EQUAL(maxCoeff(rotgen_input, &row, &col),
|
||||
eigen_ref.maxCoeff(&ref_row, &ref_col));
|
||||
TTS_EQUAL(row, ref_row);
|
||||
TTS_EQUAL(col, ref_col);
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue