class MatrixMultiplicationTestCase(unittest.TestCase): def setUp(self): self.matrix_Ab = RowMatrix(matrix_rdd,'test_data',1000,10) def test_mat_rtimes(self): vec = np.random.rand(10) p = self.matrix_Ab.rtimes_vec(vec) p_true = np.dot( A, vec ) self.assertTrue( np.linalg.norm(p-p_true) < 1e-5 ) def test_mat_ltimes(self): vec = np.random.rand(1000) p = self.matrix_Ab.ltimes_vec(vec) p_true = np.dot( vec, A ) self.assertTrue( np.linalg.norm(p-p_true) < 1e-5 ) def test_get_b(self): b = self.matrix_Ab.get_b() self.assertTrue( np.linalg.norm(b - Ab[:,-1]) < 1e-5 )
class MatrixMultiplicationTestCase(unittest.TestCase): def setUp(self): self.matrix_Ab = RowMatrix(matrix_rdd, 'test_data', 1000, 10) def test_mat_rtimes(self): vec = np.random.rand(10) p = self.matrix_Ab.rtimes_vec(vec) p_true = np.dot(A, vec) self.assertTrue(np.linalg.norm(p - p_true) < 1e-5) def test_mat_ltimes(self): vec = np.random.rand(1000) p = self.matrix_Ab.ltimes_vec(vec) p_true = np.dot(vec, A) self.assertTrue(np.linalg.norm(p - p_true) < 1e-5) def test_get_b(self): b = self.matrix_Ab.get_b() self.assertTrue(np.linalg.norm(b - Ab[:, -1]) < 1e-5)