From a89cd24a6de01c42d71632ef47b72fd1f581d980 Mon Sep 17 00:00:00 2001
From: Parijat Mazumdar <mazumdarparijat@gmail.com>
Date: Tue, 18 Feb 2014 21:35:07 +0530
Subject: [PATCH] PCA unit test added

---
 src/shogun/preprocessor/PCA.cpp         |  3 +-
 tests/unit/preprocessor/PCA_unittest.cc | 44 +++++++++++++++++++++++++
 2 files changed, 46 insertions(+), 1 deletion(-)
 create mode 100644 tests/unit/preprocessor/PCA_unittest.cc

diff --git a/src/shogun/preprocessor/PCA.cpp b/src/shogun/preprocessor/PCA.cpp
index 6752f26fc83..cbc3670075e 100644
--- a/src/shogun/preprocessor/PCA.cpp
+++ b/src/shogun/preprocessor/PCA.cpp
@@ -200,7 +200,8 @@ SGMatrix<float64_t> CPCA::apply_to_feature_matrix(CFeatures* features)
 			m_transformation_matrix.num_rows, m_transformation_matrix.num_cols);
 		final_feature_matrix = transform_matrix.transpose()*feature_matrix_centered;
 	}
-	
+
+	SG_INFO("Copying eigen matrix to shogun matrix\n")	
 	SGMatrix<float64_t> result_matrix = SGMatrix<float64_t>(num_dim, num_vectors);
 	for (int32_t c=0; c<num_vectors; c++)
 	{
diff --git a/tests/unit/preprocessor/PCA_unittest.cc b/tests/unit/preprocessor/PCA_unittest.cc
new file mode 100644
index 00000000000..39b56e5e5c7
--- /dev/null
+++ b/tests/unit/preprocessor/PCA_unittest.cc
@@ -0,0 +1,44 @@
+/*
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * Written (W) 2014 Parijat Mazumdar
+ */
+
+#include <gtest/gtest.h>
+#include <shogun/mathematics/Math.h>
+#include <shogun/features/DenseFeatures.h>
+#include <shogun/lib/SGMatrix.h>
+#include <shogun/preprocessor/PCA.h>
+
+using namespace shogun;
+
+TEST(PCA, PCA_output_test)
+{
+	SGMatrix<float64_t> data(2,3);
+	data(0,0)=1.0*cos(M_PI/3.0);
+	data(0,1)=2.0*cos(M_PI/3.0);
+	data(0,2)=3.0*cos(M_PI/3.0);
+	data(1,0)=1.0*sin(M_PI/3.0);
+	data(1,1)=2.0*sin(M_PI/3.0);
+	data(1,2)=3.0*sin(M_PI/3.0);
+
+	CDenseFeatures<float64_t>* features=new CDenseFeatures<float64_t>(data);
+	CPCA* pca=new CPCA();
+	pca->set_target_dim(1);
+	pca->init(features);
+	
+	SGMatrix<float64_t> mat=pca->apply_to_feature_matrix(features);
+
+	EXPECT_EQ(1,mat.num_rows);
+	EXPECT_EQ(3,mat.num_cols);
+
+	EXPECT_NEAR(-1,mat(0,0),0.0000001);
+	EXPECT_NEAR(0,mat(0,1),0.0000001);
+	EXPECT_NEAR(1,mat(0,2),0.0000001);
+
+	SG_UNREF(pca);
+	SG_UNREF(features);
+}