#include "multidimensional.hpp"
#include <gtest/gtest.h>

TEST(Multidimensional, CreateMatrix) {
    const auto m = arrays::create_matrix(2, 3, 7);
    ASSERT_EQ(m.size(), 2U);
    ASSERT_EQ(m[0].size(), 3U);
    EXPECT_EQ(m[1][2], 7);
}

TEST(Multidimensional, Transpose) {
    const arrays::Matrix m{{1, 2}, {3, 4}};
    const auto t = arrays::transpose(m);
    EXPECT_EQ(t, (arrays::Matrix{{1, 3}, {2, 4}}));
}

TEST(Multidimensional, RowSums) {
    const arrays::Matrix m{{1, 2, 3}, {4, 5, 6}};
    EXPECT_EQ(arrays::row_sums(m), (std::vector<int>{6, 15}));
}