164 lines
5.1 KiB
C++
164 lines
5.1 KiB
C++
|
|
#include "test_common.h"
|
|
#include "./utils/matrix.h"
|
|
|
|
using utils::Vf; using utils::Vd; using utils::Vi;
|
|
using utils::Mf; using utils::Md; using utils::Mi;
|
|
|
|
|
|
// tiny helper
|
|
template <typename T>
|
|
static bool mat_is_filled(const utils::Matrix<T>& M, T v) {
|
|
for (std::uint64_t i = 0; i < M.rows(); ++i)
|
|
for (std::uint64_t j = 0; j < M.cols(); ++j)
|
|
if (M(i,j) != v) return false;
|
|
return true;
|
|
}
|
|
|
|
// ------------------- basic construction -------------------
|
|
TEST_CASE(Matrix_Default_Construct) {
|
|
Md M;
|
|
CHECK_EQ(M.rows(), 0, "default rows should be 0");
|
|
CHECK_EQ(M.cols(), 0, "default cols should be 0");
|
|
}
|
|
|
|
TEST_CASE(Matrix_Filled_Construct) {
|
|
Md M(3, 4, 2.5);
|
|
CHECK_EQ(M.rows(), 3, "rows");
|
|
CHECK_EQ(M.cols(), 4, "cols");
|
|
CHECK(mat_is_filled(M, 2.5), "all elements should be 2.5");
|
|
}
|
|
|
|
// ------------------- element access / write -------------------
|
|
TEST_CASE(Matrix_Set_Get) {
|
|
Mi M(2, 3, 0);
|
|
M(0,0) = 42;
|
|
M(1,2) = -7;
|
|
CHECK_EQ(M(0,0), 42, "set/get (0,0)");
|
|
CHECK_EQ(M(1,2), -7, "set/get (1,2)");
|
|
}
|
|
|
|
// ------------------- resize semantics -------------------
|
|
TEST_CASE(Matrix_Resize_Grow) {
|
|
Mf M(2, 2, 1.0f);
|
|
M.resize(3, 4, 9.0f); // grow; newly appended elements get the fill value
|
|
CHECK_EQ(M.rows(), 3, "rows after resize");
|
|
CHECK_EQ(M.cols(), 4, "cols after resize");
|
|
CHECK(M(2,3) == 9.0f, "last element should be the fill value after grow");
|
|
}
|
|
|
|
TEST_CASE(Matrix_Resize_Shrink) {
|
|
Mi M(4, 4, 5);
|
|
M(0,0) = 11;
|
|
M.resize(2, 2, 999); // shrink; size reduces
|
|
CHECK_EQ(M.rows(), 2, "rows after shrink");
|
|
CHECK_EQ(M.cols(), 2, "cols after shrink");
|
|
// element mapping after shrink is implementation dependent; just check bounds usable
|
|
M(1,1) = 3;
|
|
CHECK_EQ(M(1,1), 3, "write after shrink works");
|
|
}
|
|
|
|
// ------------------- row helpers -------------------
|
|
TEST_CASE(Matrix_Get_Row) {
|
|
Mi M(3,4,0);
|
|
// set row 1 to [10,20,30,40]
|
|
for (std::uint64_t j=0;j<4;++j) M(1,j) = (j+1)*10;
|
|
auto r = M.get_row(1);
|
|
CHECK_EQ(r.size(), 4, "row size");
|
|
CHECK(r[0]==10 && r[1]==20 && r[2]==30 && r[3]==40, "row contents");
|
|
}
|
|
|
|
TEST_CASE(Matrix_Set_Row) {
|
|
Mi M(2,3,0);
|
|
utils::Vector<int64_t> v(3, 0);
|
|
v[0]=7; v[1]=8; v[2]=9;
|
|
M.set_row(0, v);
|
|
CHECK(M(0,0)==7 && M(0,1)==8 && M(0,2)==9, "set_row contents");
|
|
}
|
|
|
|
TEST_CASE(Matrix_Row_OutOfRange_Throws) {
|
|
Mi M(2,2,0);
|
|
bool threw=false;
|
|
try { (void)M.get_row(2); } catch(const std::out_of_range&) { threw=true; }
|
|
CHECK(threw, "get_row out-of-range should throw");
|
|
|
|
threw=false;
|
|
try {
|
|
utils::Vector<int64_t> v(3,1); // wrong size
|
|
M.set_row(1, v);
|
|
} catch(const std::runtime_error&) { threw=true; }
|
|
CHECK(threw, "set_row size mismatch should throw");
|
|
}
|
|
|
|
// ------------------- col helpers -------------------
|
|
TEST_CASE(Matrix_Get_Col) {
|
|
Mi M(3,2,0);
|
|
M(0,1)=5; M(1,1)=6; M(2,1)=7;
|
|
auto c = M.get_col(1);
|
|
CHECK_EQ(c.size(), 3, "col size");
|
|
CHECK(c[0]==5 && c[1]==6 && c[2]==7, "col contents");
|
|
}
|
|
|
|
TEST_CASE(Matrix_Set_Col) {
|
|
Mi M(3,2,0);
|
|
utils::Vector<int64_t> v(3, 0);
|
|
v[0]=1; v[1]=4; v[2]=9;
|
|
M.set_col(0, v);
|
|
CHECK(M(0,0)==1 && M(1,0)==4 && M(2,0)==9, "set_col contents");
|
|
}
|
|
|
|
TEST_CASE(Matrix_Col_OutOfRange_Throws) {
|
|
Mi M(2,2,0);
|
|
bool threw=false;
|
|
try { (void)M.get_col(2); } catch(const std::out_of_range&) { threw=true; }
|
|
CHECK(threw, "get_col out-of-range should throw");
|
|
|
|
threw=false;
|
|
try {
|
|
utils::Vector<int64_t> v(1,1); // wrong size
|
|
M.set_col(1, v);
|
|
} catch(const std::runtime_error&) { threw=true; }
|
|
CHECK(threw, "set_col size mismatch should throw");
|
|
}
|
|
|
|
// ------------------- swap rows/cols -------------------
|
|
TEST_CASE(Matrix_Swap_Rows) {
|
|
Mi M(2,3,0);
|
|
// row0: 1 2 3, row1: 4 5 6
|
|
for (std::uint64_t j=0;j<3;++j){ M(0,j)=j+1; M(1,j)=j+4; }
|
|
M.swap_rows(0,1);
|
|
CHECK(M(0,0)==4 && M(0,1)==5 && M(0,2)==6, "row0 after swap");
|
|
CHECK(M(1,0)==1 && M(1,1)==2 && M(1,2)==3, "row1 after swap");
|
|
}
|
|
|
|
TEST_CASE(Matrix_Swap_Cols) {
|
|
Mi M(3,3,0);
|
|
// col0: 9 8 7, col2: 1 2 3
|
|
M(0,0)=9; M(1,0)=8; M(2,0)=7;
|
|
M(0,2)=1; M(1,2)=2; M(2,2)=3;
|
|
M.swap_cols(0,2);
|
|
CHECK(M(0,0)==1 && M(1,0)==2 && M(2,0)==3, "col0 after swap");
|
|
CHECK(M(0,2)==9 && M(1,2)==8 && M(2,2)==7, "col2 after swap");
|
|
}
|
|
|
|
TEST_CASE(Matrix_Swap_OutOfRange_Throws) {
|
|
Mi M(2,2,0);
|
|
bool threw=false;
|
|
try { M.swap_rows(0,2); } catch(const std::out_of_range&) { threw=true; }
|
|
CHECK(threw, "swap_rows out-of-range should throw");
|
|
threw=false;
|
|
try { M.swap_cols(0,3); } catch(const std::out_of_range&) { threw=true; }
|
|
CHECK(threw, "swap_cols out-of-range should throw");
|
|
}
|
|
|
|
// ------------------- stream output (basic sanity) -------------------
|
|
TEST_CASE(Matrix_Stream_ToString) {
|
|
Mf M(2,2,0.0f);
|
|
M(0,0)=1.0f; M(0,1)=2.0f; M(1,0)=3.0f; M(1,1)=4.0f;
|
|
std::ostringstream os;
|
|
os << M;
|
|
auto s = os.str();
|
|
CHECK(s.find("[[") != std::string::npos, "starts with [[");
|
|
CHECK(s.find("1.000") != std::string::npos, "contains formatted 1.000");
|
|
CHECK(s.find("4.000") != std::string::npos, "contains formatted 4.000");
|
|
} |