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_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)