def test_final_shape(self):
stacker = MedianStacker(self.single_matrix_data)
self.assertEqual(stacker.final_shape()['shape'], (2, 3))
self.assertEqual(stacker.final_shape()['zero'], (0, 0))
stacker = MedianStacker(self.two_matrices_data)
self.assertEqual(stacker.final_shape()['shape'], (5, 5))
self.assertEqual(stacker.final_shape()['zero'], (3, 0))
stacker = MedianStacker(self.three_matrices_data)
self.assertEqual(stacker.final_shape()['shape'], (12, 9))
self.assertEqual(stacker.final_shape()['zero'], (3, 4))
def test_median(self):
stacker = MedianStacker([
self.two_matrices_data[0], self.two_matrices_data[0],
self.two_matrices_data[1]
])
expected = np.zeros(25).reshape(5, 5)
expected[3:5, 0:3] = np.arange(1, 7).reshape(2, 3)
assert_array_equal(stacker.median(), expected)
stacker = MedianStacker([
self.two_matrices_data[0], self.two_matrices_data[1],
self.two_matrices_data[1]
])
expected = np.zeros(25).reshape(5, 5)
expected[0:2, 2:5] = np.arange(3, 9).reshape(2, 3)
assert_array_equal(stacker.median(), expected)
def test_data_reposition(self):
stacker = MedianStacker([])
input = self.three_matrices_data[0]
data = input['data']
assert_array_equal(
stacker.data_reposition(input, {
'zero': (0, 0),
'shape': (2, 3)
}), data)
expected = np.zeros(3 * 3).reshape(3, 3)
expected[1:3, 0:3] = data
assert_array_equal(
stacker.data_reposition(input, {
'zero': (1, 0),
'shape': (3, 3)
}), expected)
expected = np.zeros(3 * 4).reshape(3, 4)
expected[1:3, 1:4] = data
assert_array_equal(
stacker.data_reposition(input, {
'zero': (1, 1),
'shape': (3, 4)
}), expected)
input = self.three_matrices_data[1]
data = input['data']
expected = np.zeros(4 * 6).reshape(4, 6)
expected[0:2, 2:5] = data
assert_array_equal(
stacker.data_reposition(input, {
'zero': (3, 0),
'shape': (4, 6)
}), expected)
input = self.three_matrices_data[2]
data = input['data']
expected = np.zeros(12 * 9).reshape(12, 9)
expected[10:12, 0:3] = data
actual = stacker.data_reposition(input, {
'zero': (3, 4),
'shape': (12, 9)
})
assert_array_equal(actual, expected)
def test_median_single(self):
stacker = MedianStacker(self.single_matrix_data)
assert_array_equal(stacker.median(),
self.single_matrix_data[0]['data'])
