Improving matrix class

1 files changed, 30 insertions, 19 deletions

diff --git a/test/matrixtest.cpp b/test/matrixtest.cpp
index aecf4c87..530d3530 100644
--- a/test/matrixtest.cpp
+++ b/test/matrixtest.cpp
@@ -6,38 +6,38 @@
  * ----------------------------------------------------------------------------
  */
 
+#include <algorithm>
+#include <cstdlib>
+#include <ctime>
+#include <vector>
+
 #include "Math/math.hpp"
 
 #include "gtest/gtest.h"
 
-int matrixAssign(int number)
+template<int Size> int matrixAssign(int number, int i, int j)
 {
-	yage::Matrix<4, 5, int> m;
-	m[2][3]=number;
+	yage::Matrix<Size, Size, int> m;
+	m[j][i]=number;
 
-	return m[2][3];
+	return m[j][i];
 }
 
-int matrixAddition(int num1, int num2)
+template<int Size> int matrixAddition(int num1, int num2)
 {
-	yage::Matrix<4, 4, int> m1, m2;
+	yage::Matrix<Size, Size, int> m1, m2;
 	m1[1][1]=num1;
 	m2[1][1]=num2;
 
-	yage::Matrix<4, 4, int> m3=m1+m2;
+	yage::Matrix<Size, Size, int> m3 = m1+m2;
 
 	return m3[1][1];
 }
 
-bool vectorDotProduct()
+int vectorDotProduct(const std::vector<int> &vec_contents_f, const std::vector<int> &vec_contents_s)
 {
-	yage::Vector<3, int> v1, v2;
-	v1[0] = 2;
-	v1[1] = 3;
-	v1[2] = 5;
-	v2[0] = 9;
-	v2[1] = 6;
-	v2[2] = 8;
+	yage::Vector<3, int> v1(vec_contents_f);
+	yage::Vector<3, int> v2(vec_contents_s);
 
 	int x = yage::matrix::dot(v1, v2);
 
@@ -53,21 +53,32 @@ bool matrixMultiplication()
 
 TEST(Matrix, Assign)
 {
-	ASSERT_EQ(29348, matrixAssign(29348));
+	int rand_num = rand();
+	ASSERT_EQ(rand_num, matrixAssign<10>(rand_num, 4, 2));
 }
 
 TEST(Matrix, Addition)
 {
-	ASSERT_EQ(5793+2389, matrixAddition(5793, 2389));
+	int rand_x = rand();
+	int rand_y = rand();
+	ASSERT_EQ(rand_x+rand_y, matrixAddition<10>(rand_x, rand_y));
 }
 
-TEST(Matrix, DotProduct)
+TEST(Vector, DotProduct)
 {
-	ASSERT_TRUE(vectorDotProduct());
+	std::vector<int> contents_i = { rand()%100, rand()%100, rand()%100 };
+	std::vector<int> contents_j = { rand()%100, rand()%100, rand()%100 };
+	int sum = 0;
+	for(std::size_t i = 0; i < contents_i.size(); ++i)
+	{
+		sum += contents_i[i]*contents_j[i];
+	}
+	ASSERT_EQ(sum, vectorDotProduct(contents_i, contents_j));
 }
 
 int main(int argc, char **argv)
 {
+	srand(static_cast<unsigned>(time(nullptr)));
 	testing::InitGoogleTest(&argc, argv);
 	return RUN_ALL_TESTS();
 }