def test_transformchain_apply():
mocked_transform = Mock()
mocked_transform._apply.return_value = 3
mocked_transform2 = Mock()
mocked_transform2._apply.return_value = 4
transforms = [mocked_transform, mocked_transform2]
tr = TransformChain(transforms)
result = tr.apply(1)
assert result == 4
def transformchain_apply_test():
mocked_transform = Mock()
mocked_transform._apply.return_value = 3
mocked_transform2 = Mock()
mocked_transform2._apply.return_value = 4
transforms = [mocked_transform, mocked_transform2]
tr = TransformChain(transforms)
result = tr.apply(1)
assert (result == 4)
def chain_compose_after_inplace_chain_test():
a = PointCloud(np.random.random([10, 2]))
b = PointCloud(np.random.random([10, 2]))
t = Translation([3, 4])
s = Scale([4, 2])
chain_1 = TransformChain([t, s])
chain_2 = TransformChain([s.pseudoinverse(), t.pseudoinverse()])
chain_1.compose_before_inplace(chain_2)
points = PointCloud(np.random.random([10, 2]))
chain_res = chain_1.apply(points)
assert(np.allclose(points.points, chain_res.points))
def chain_compose_after_inplace_tps_test():
a = PointCloud(np.random.random([10, 2]))
b = PointCloud(np.random.random([10, 2]))
tps = ThinPlateSplines(a, b)
t = Translation([3, 4])
s = Scale([4, 2])
chain = TransformChain([t, s])
chain.compose_after_inplace(tps)
points = PointCloud(np.random.random([10, 2]))
manual_res = s.apply(t.apply(tps.apply(points)))
chain_res = chain.apply(points)
assert(np.all(manual_res.points == chain_res.points))
def manual_no_op_chain_test():
points = PointCloud(np.random.random([10, 2]))
t = Translation([3, 4])
chain = TransformChain([t, t.pseudoinverse()])
points_applied = chain.apply(points)
assert(np.allclose(points_applied.points, points.points))
