def chain_tps_after_pwa_test():
a_tm = TriMesh(np.random.random([10, 2]))
b = PointCloud(np.random.random([10, 2]))
pwa = PiecewiseAffine(a_tm, b)
tps = ThinPlateSplines(b, a_tm)
chain = tps.compose_after(pwa)
assert(isinstance(chain, TransformChain))
def compose_tps_after_translation_test():
a = PointCloud(np.random.random([10, 2]))
b = PointCloud(np.random.random([10, 2]))
t = Translation([3, 4])
tps = ThinPlateSplines(a, b)
chain = tps.compose_after(t)
assert(isinstance(chain, TransformChain))
def chain_tps_after_tps_test():
a = PointCloud(np.random.random([10, 2]))
b = PointCloud(np.random.random([10, 2]))
tps_one = ThinPlateSplines(a, b)
tps_two = ThinPlateSplines(b, a)
chain = tps_one.compose_after(tps_two)
assert(isinstance(chain, TransformChain))
points = PointCloud(np.random.random([10, 2]))
chain_res = chain.apply(points)
manual_res = tps_one.apply(tps_two.apply(points))
assert (np.all(chain_res.points == manual_res.points))
