def test_outer(self): mat1 = RowMatrix(self.sc.parallelize([(1, array([1, 2, 3])), (2, array([4, 5, 6]))])) resultA = mat1.gramian() resultB1 = mat1.gramian("accum") resultB2 = mat1.gramian("aggregate") truth = array([[17, 22, 27], [22, 29, 36], [27, 36, 45]]) assert array_equal(resultA, truth) assert array_equal(resultB1, truth) assert array_equal(resultB2, truth)
def test_outer(self): mat1 = RowMatrix( self.sc.parallelize([(1, array([1, 2, 3])), (2, array([4, 5, 6]))])) resultA = mat1.gramian() resultB1 = mat1.gramian("accum") resultB2 = mat1.gramian("aggregate") truth = array([[17, 22, 27], [22, 29, 36], [27, 36, 45]]) assert array_equal(resultA, truth) assert array_equal(resultB1, truth) assert array_equal(resultB2, truth)