def test_points_equal(self):
points1 = {1: gp.Point([0, 1])}
points2 = {1: gp.Point([0, 1])}
self.assertTrue(self.points_equal(points1, points2))
points1[2] = gp.Point([1, 2])
points2[2] = gp.Point([2, 1])
self.assertFalse(self.points_equal(points1, points2))
def test_shift_points5(self):
data = {
0: gp.Point([3, 0]),
1: gp.Point([3, 2]),
2: gp.Point([3, 4]),
3: gp.Point([3, 6]),
4: gp.Point([3, 8])
}
spec = gp.PointsSpec(gp.Roi(offset=(0, 0), shape=(15, 10)))
points = gp.Points(data, spec)
request_roi = gp.Roi(offset=(3, 0), shape=(9, 10))
shift_array = np.array([[3, 0], [-3, 0], [0, 0], [-3, 0], [3, 0]],
dtype=int)
lcm_voxel_size = gp.Coordinate((3, 2))
shifted_data = {
0: gp.Point([6, 0]),
2: gp.Point([3, 4]),
4: gp.Point([6, 8])
}
result = gp.ShiftAugment.shift_points(points,
request_roi,
shift_array,
shift_axis=1,
lcm_voxel_size=lcm_voxel_size)
# print("test 4", result.data, shifted_data)
self.assertTrue(self.points_equal(result.data, shifted_data))
self.assertTrue(result.spec == gp.PointsSpec(request_roi))
def test_shift_points3(self):
data = {1: gp.Point([0, 1])}
spec = gp.PointsSpec(gp.Roi(offset=(0, 0), shape=(5, 5)))
points = gp.Points(data, spec)
request_roi = gp.Roi(offset=(0, 1), shape=(5, 3))
shift_array = np.array([[0, 1], [0, -1], [0, 0], [0, 0], [0, 1]],
dtype=int)
lcm_voxel_size = gp.Coordinate((1, 1))
shifted_points = gp.Points({1: gp.Point([0, 2])},
gp.PointsSpec(request_roi))
result = gp.ShiftAugment.shift_points(points,
request_roi,
shift_array,
shift_axis=0,
lcm_voxel_size=lcm_voxel_size)
# print("test 3", result.data, shifted_points.data)
self.assertTrue(self.points_equal(result.data, shifted_points.data))
self.assertTrue(result.spec == gp.PointsSpec(request_roi))