def test_times_array(self): mat1 = RowMatrix(self.sc.parallelize([(1, array([1, 2, 3])), (2, array([4, 5, 6]))])) mat2 = array([[7, 8], [9, 10], [11, 12]]) truth = [array([58, 64]), array([139, 154])] rdd = mat1.times(mat2) result = rdd.rows().collect() assert array_equal(result, truth) assert array_equal(rdd.index, range(0, 2))
def test_times_array(self): mat1 = RowMatrix( self.sc.parallelize([(1, array([1, 2, 3])), (2, array([4, 5, 6]))])) mat2 = array([[7, 8], [9, 10], [11, 12]]) truth = [array([58, 64]), array([139, 154])] rdd = mat1.times(mat2) result = rdd.rows().collect() assert array_equal(result, truth) assert array_equal(rdd.index, range(0, 2))
def test_times_rdd(self): mat1 = RowMatrix( self.sc.parallelize([(1, array([1, 2, 3])), (2, array([4, 5, 6]))], 2)) mat2 = RowMatrix( self.sc.parallelize([(1, array([7, 8, 9])), (2, array([10, 11, 12]))], 2)) truth = array([[47, 52, 57], [64, 71, 78], [81, 90, 99]]) resultA = mat1.times(mat2) assert array_equal(resultA, truth)
def test_times_rdd(self): mat1 = RowMatrix(self.sc.parallelize([(1, array([1, 2, 3])), (2, array([4, 5, 6]))], 2)) mat2 = RowMatrix(self.sc.parallelize([(1, array([7, 8, 9])), (2, array([10, 11, 12]))], 2)) truth = array([[47, 52, 57], [64, 71, 78], [81, 90, 99]]) resultA = mat1.times(mat2) assert array_equal(resultA, truth)