#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