Example #1
0
 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)