def test_get_data_transposed(self):
a = MatrixPointer([1, 3, 5, 2, 4, 6], (2, 3), transpose=True)
self.assertEqual(a.get_data(), [1, 2, 3, 4, 5, 6])
def test_reduce_sqr_diff(self):
a = MatrixPointer([1, 2, 3, 4], (2, 2))
b = MatrixPointer([4, 3, 2, 1], (2, 2))
c = a.reduce(lambda accumulator, x, y: accumulator + (x - y)**2, 0, b)
self.assertEqual(c, 20)
def test_map_sub(self):
a = MatrixPointer([1, 2, 3, 4], (2, 2))
b = MatrixPointer([4, 3, 2, 1], (2, 2))
c = a.map(lambda x, y: x - y, b)
self.assertEqual(c, [-3, -1, 1, 3])
def test_reduce_sum(self):
a = MatrixPointer([1, 2, 3, 4], (2, 2))
b = MatrixPointer([4, 3, 2, 1], (2, 2))
c = a.reduce(lambda accumulator, x, y: accumulator + x + y, 0, b)
self.assertEqual(c, sum([5]*4))
def test_map_sum(self):
a = MatrixPointer([1, 2, 3, 4], (2, 2))
b = MatrixPointer([4, 3, 2, 1], (2, 2))
c = a.map(lambda x, y: x + y, b)
self.assertEqual(c, [5]*4)
