Compiled

2007-06-23 16:17:56 +00:00
parent e0a324e2f3
commit de3c55dfe9
7 changed files with 2485 additions and 0 deletions
@@ -0,0 +1,82 @@
+/*
+ * =====================================================================================
+ *
+ *       Filename:  sparse_matrix_unit.cc
+ *
+ *    Description:  
+ *
+ *        Version:  1.0
+ *        Created:  06/22/2007 09:35:38 AM EDT
+ *       Revision:  none
+ *       Compiler:  gcc
+ *
+ *         Author:  Nikolaos Vasiloglou (NV), nvasil@ieee.org
+ *        Company:  Georgia Tech Fastlab-ESP Lab
+ *
+ * =====================================================================================
+ */
+#include "fastlib/fastlib.h"
+#include "base/test.h"
+#include "u/nvasil/sparse/sparse_matrix.h"
+
+namespace sparse {
+template<typename T>
+class MatrixTest {
+ public:
+	void Init(){
+	  size_=100;
+		non_zero_=10;
+		matrix_.Init(100, 10);
+		for(index_t i=0; i<size_; i++) {
+		  for(index_t j=0; j<non_zero_; j++) {
+			  matrix_.set(i, j, float32(i*size_+j));
+			}
+		}
+		matrixfile_="sparse_matrix.txt";
+		xfile_="xfile.txt";
+		bfile_="bfile.txt";
+	};
+	void Destruct(){
+	  matrix_.Destruct();
+	}
+  void TestAccess(){
+	  for(index_t i=0; i<size_; i++) {
+		  for(index_t j=0; j<non_zero_; j++) {
+			  TEST_ASSERT(matrix_.get(i, j)==float32(i*size_+j));
+			}
+		}
+	}
+  
+  void TestConjugateGradient() {
+		matrix_.Destruct();
+		matrix_.Init(matrixfile_);
+		T *b=ReadVectorFromFile<T>(bfile_);
+		T *x=ReadVectorFromFile<T>(xfile_);
+	  T *xconjg=NewVector<T>(size_);	
+	  ConjugateGradient(matrix_, b,  xconjg);
+		for(index_t i=0; i<size_; i++) {
+		  TEST_DOUBLE_APPROX(xconjg[i], x[i], 0.0001);
+		}
+	}
+  void TestAll() {
+	  Init();
+		TestAccess();
+		Destruct();
+		Init();
+    TestConjugateGradient();
+		Destruct();
+	}	
+ private:
+	Matrix<T>  matrix_;
+	index_t size_;
+	index_t non_zero_;
+  string matrixfile_;
+  string xfile_;
+  string bfile_;	
+};
+};
+
+int main(int argc, char *argv[]) {
+	sparse::MatrixTest<float32> matrix_test;
+	matrix_test.TestAll();
+}