KOKO\Mimi
62 lines
1.9 KiB
C++
62 lines
1.9 KiB
C++
#include "math/dense_blas_internal.h"
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <array>
|
|
#include <cstddef>
|
|
#include <limits>
|
|
#include <stdexcept>
|
|
|
|
namespace fesa::dense_blas_internal {
|
|
namespace {
|
|
|
|
TEST(DenseBlasInternal, CDup003ConvertsZeroNormalAndOverflowLengths) {
|
|
const auto zero = ToMklSize(0U);
|
|
ASSERT_TRUE(zero.HasValue());
|
|
EXPECT_EQ(zero.Value(), 0);
|
|
|
|
const auto normal = ToMklSize(3U);
|
|
ASSERT_TRUE(normal.HasValue());
|
|
EXPECT_EQ(normal.Value(), 3);
|
|
|
|
const std::size_t overflow =
|
|
static_cast<std::size_t>((std::numeric_limits<MKL_INT>::max)()) + 1U;
|
|
const auto rejected = ToMklSize(overflow);
|
|
EXPECT_FALSE(rejected.HasValue());
|
|
EXPECT_FALSE(rejected.GetStatus().IsOk());
|
|
}
|
|
|
|
TEST(DenseBlasInternal, CDup003CopiesZeroAndNonzeroContiguousValues) {
|
|
EXPECT_NO_THROW(CopyValues(nullptr, 0U, nullptr));
|
|
|
|
const std::array<double, 3U> source{1.25, -2.5, 4.0};
|
|
std::array<double, 3U> destination{};
|
|
CopyValues(source.data(), source.size(), destination.data());
|
|
|
|
EXPECT_EQ(destination, source);
|
|
}
|
|
|
|
TEST(DenseBlasInternal, TransposeProductChecksShapeAndKnownRectangularAction) {
|
|
Matrix left{3U, 2U};
|
|
Matrix right{3U, 1U};
|
|
for (std::size_t row = 0; row < 3U; ++row) {
|
|
left(row, 0U) = static_cast<double>(row + 1U);
|
|
left(row, 1U) = 2.0;
|
|
right(row, 0U) = static_cast<double>(row);
|
|
}
|
|
const auto product = MultiplyTransposeLeft(left, right);
|
|
ASSERT_EQ(product.Rows(), 2U);
|
|
ASSERT_EQ(product.Columns(), 1U);
|
|
EXPECT_DOUBLE_EQ(product(0U, 0U), 8.0);
|
|
EXPECT_DOUBLE_EQ(product(1U, 0U), 6.0);
|
|
EXPECT_THROW(static_cast<void>(MultiplyTransposeLeft(left, Matrix{2U, 1U})),
|
|
std::invalid_argument);
|
|
const auto empty = MultiplyTransposeLeft(Matrix{0U, 2U}, Matrix{0U, 3U});
|
|
EXPECT_EQ(empty.Rows(), 2U);
|
|
EXPECT_EQ(empty.Columns(), 3U);
|
|
EXPECT_DOUBLE_EQ(empty(1U, 2U), 0.0);
|
|
}
|
|
|
|
} // namespace
|
|
} // namespace fesa::dense_blas_internal
|