def chain_tps_after_pwa_test():
a_tm = TriMesh(np.random.random([10, 2]))
b = PointCloud(np.random.random([10, 2]))
pwa = PiecewiseAffineTransform(a_tm, b)
tps = TPS(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 = TPS(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 = TPS(a, b)
tps_two = TPS(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))
