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)
