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))