def test_padding_vec3_4(self): dtype = np.dtype((int, (3,))) a = ChunkGrid(2, dtype, np.zeros(3)) expected = np.zeros((6, 6, 6, 3), dtype=int) for n, i in enumerate(np.ndindex(6, 6, 6)): pos = np.array(i, dtype=int) - 2 value = np.array([1, 2, 3]) + (n + 1) * 3 a.set_value(pos, value) s = np.sum(pos < 0) + np.sum(2 <= pos) if s >= 2: continue expected[i] = value data = a.padding_at((0, 0, 0), 2, corners=False, edges=False) self.assertEqual(expected.shape, data.shape) self.assertEqual(str(expected), str(data)) expected2 = expected[1:-1, 1:-1, 1:-1] data2 = a.padding_at((0, 0, 0), 1, corners=False, edges=False) self.assertEqual(expected2.shape, data2.shape) self.assertEqual(str(expected2), str(data2)) expected3_tmp = np.zeros((6, 6, 6, 3), dtype=int) for n, i in enumerate(np.ndindex(6, 6, 6)): value = np.array([1, 2, 3]) + (n + 1) * 3 expected3_tmp[i] = value expected3 = np.zeros((8, 8, 8, 3), dtype=int) expected3[1:-1, 1:-1, 1:-1] = expected3_tmp data3 = a.padding_at((0, 0, 0), 3, corners=False, edges=False) self.assertEqual(expected3.shape, data3.shape) self.assertEqual(str(expected3), str(data3))
def test_padding_2(self): a = ChunkGrid(2, int, 0) expected = np.zeros((6, 6, 6), dtype=int) for n, i in enumerate(np.ndindex(6, 6, 6)): pos = np.array(i, dtype=int) - 2 a.set_value(pos, n + 1) s = np.sum(pos < 0) + np.sum(2 <= pos) if s >= 2: continue expected[i] = n + 1 data = a.padding_at((0, 0, 0), 2, corners=False, edges=False) self.assertEqual(expected.shape, data.shape) self.assertEqual(str(expected), str(data)) expected2 = expected[1:-1, 1:-1, 1:-1] data2 = a.padding_at((0, 0, 0), 1, corners=False, edges=False) self.assertEqual(expected2.shape, data2.shape) self.assertEqual(str(expected2), str(data2)) expected3_tmp = np.zeros((6, 6, 6), dtype=int) for n, i in enumerate(np.ndindex(6, 6, 6)): expected3_tmp[i] = n + 1 expected3 = np.zeros((8, 8, 8), dtype=int) expected3[1:-1, 1:-1, 1:-1] = expected3_tmp data3 = a.padding_at((0, 0, 0), 3, corners=False, edges=False) self.assertEqual(expected3.shape, data3.shape) self.assertEqual(str(expected3), str(data3))
def test_padding_1(self): a = ChunkGrid(2, int, 0) expected = np.zeros((6, 6, 6), dtype=int) for n, i in enumerate(np.ndindex(6, 6, 6)): pos = np.array(i, dtype=int) - 2 a.set_value(pos, n + 1) expected[i] = n + 1 data = a.padding_at((0, 0, 0), 2) self.assertEqual(expected.shape, data.shape) self.assertEqual(str(expected), str(data)) expected2 = expected[1:-1, 1:-1, 1:-1] data2 = a.padding_at((0, 0, 0), 1) self.assertEqual(expected2.shape, data2.shape) self.assertEqual(str(expected2), str(data2)) expected3_tmp = np.zeros((6, 6, 6), dtype=int) for n, i in enumerate(np.ndindex(6, 6, 6)): expected3_tmp[i] = n + 1 expected3 = np.zeros((8, 8, 8), dtype=int) expected3[1:-1, 1:-1, 1:-1] = expected3_tmp data3 = a.padding_at((0, 0, 0), 3) self.assertEqual(expected3.shape, data3.shape) self.assertEqual(str(expected3), str(data3))
def test_padding_vec3_1(self): dtype = np.dtype((int, (3,))) a = ChunkGrid(2, dtype, np.zeros(3)) a.set_value((0, 0, 2), np.ones(3) * 1) a.set_value((0, 2, 0), np.ones(3) * 2) a.set_value((2, 0, 0), np.ones(3) * 3) a.set_value((1, 1, 1), np.ones(3) * 4) a.set_value((2, 2, 2), np.ones(3) * -1) a.set_value((-1, -1, -1), np.ones(3) * -2) a.set_value((2, -1, -1), np.ones(3) * -3) a.set_value((-1, 2, -1), np.ones(3) * -4) a.set_value((-1, -1, 2), np.ones(3) * -5) a.set_value((2, 2, -1), np.ones(3) * -6) a.set_value((2, -1, 2), np.ones(3) * -7) a.set_value((-1, 2, 2), np.ones(3) * -8) pad = a.padding_at((0, 0, 0), 1, corners=False) expected = np.zeros((4, 4, 4, 3), dtype=int) expected[1, 1, 3] = 1 expected[1, 3, 1] = 2 expected[3, 1, 1] = 3 expected[2, 2, 2] = 4 self.assertEqual(expected.shape, pad.shape) self.assertEqual(str(expected), str(pad))
def test_padding_vec3_2(self): a = ChunkGrid(2, int, 0) expected = np.zeros((6, 6, 6), dtype=int) for n, i in enumerate(np.ndindex(6, 6, 6)): pos = np.array(i, dtype=int) - 2 value = (n + 1) a.set_value(pos, value) expected[i] = value data = a.padding_at((0, 0, 0), 2) self.assertEqual(expected.shape, data.shape) self.assertEqual(str(expected), str(data)) expected2 = expected[1:-1, 1:-1, 1:-1] data2 = a.padding_at((0, 0, 0), 1) self.assertEqual(expected2.shape, data2.shape) self.assertEqual(str(expected2), str(data2))
def test_padding_no_corners_edges(self): a = ChunkGrid(2, int, -1) expected = np.zeros((6, 6, 6), int) for n, i in enumerate(np.ndindex(6, 6, 6)): a.set_value(i, n) expected[i] = n self.assertEqual(expected.shape, tuple(a.size())) self.assertEqual(str(expected), str(a.to_dense())) padded1 = a.padding_at((1, 1, 1), 1, corners=True, edges=True) expected1 = expected[1:-1, 1:-1, 1:-1] self.assertEqual(str(expected1), str(padded1)) padded2 = a.padding_at((1, 1, 1), 1, corners=False, edges=False) expected2 = np.full((4, 4, 4), -1, int) expected2[1:-1, 1:-1, :] = expected[2:-2, 2:-2, 1:-1] expected2[1:-1, :, 1:-1] = expected[2:-2, 1:-1, 2:-2] expected2[:, 1:-1, 1:-1] = expected[1:-1, 2:-2, 2:-2] self.assertEqual(str(expected2), str(padded2))