Add test to GrangerCausality

Author: aoqiz

tests/TestGranger.py

@@ -19,7 +19,6 @@ def syn_data_3d(self):
         # generate transition matrix, time lag 2
         np.random.seed(0)
         A = 0.2 * np.random.rand(3,6)
-        print('True matrix is \n {}'.format(A))
         # generate time series
         T = 1000
         data = np.random.rand(3, T)
@@ -35,7 +34,6 @@ def syn_data_2d(self):
         A = 0.5*np.random.rand(2,4)
         A[0,1] = 0
         A[0,3] = 0
-        print('True matrix is \n {}'.format(A))
         # generate time series
         T = 100
         data = np.random.rand(2, T)
@@ -54,8 +52,10 @@ def test_granger_test(self):
         dataset = self.syn_data_2d()
         G = Granger()
         p_value_matrix, adj_matrix = G.granger_test_2d(data=dataset)
-        print('P-value matrix is \n {}'.format(p_value_matrix))
-        print('Adjacency matrix is \n {}'.format(adj_matrix))
+        p_value_matrix_truth = np.array([[0, 0.5989, 0, 0.5397], [0.0006, 0, 0.0014, 0]])
+        adj_matrix_truth = np.array([[1, 0, 1, 0], [1, 1, 1, 1]])
+        self.assertEqual((np.round(p_value_matrix, 4) - p_value_matrix_truth).all(), 0)
+        self.assertEqual((adj_matrix - adj_matrix_truth).all(), 0)
 
     # example2
     # for data with multi-dimensional variables, granger lasso regression.
@@ -66,7 +66,10 @@ def test_granger_lasso(self):
         dataset = self.syn_data_3d()
         G = Granger()
         coeff = G.granger_lasso(data=dataset)
-        print('Estimated matrix is \n {}'.format(coeff))
+        coeff_truth = np.array([[0.09, 0.1101, 0.1527, 0.1127, 0.0226, 0.1538],
+                                [0.1004, 0.15, 0.1757, 0.1037, 0.1612, 0.0987],
+                                [0.1155, 0.1485, 0, 0.039, -0., 0.1085]])
+        self.assertEqual((np.round(coeff, 4) - coeff_truth).all(), 0)
 
 
 if __name__ == '__main__':