21 lines
564 B
C++
21 lines
564 B
C++
#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}));
|
|
}
|