def chain_tps_before_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_before(pwa) assert(isinstance(chain, TransformChain))
def chain_tps_before_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_before(tps_two) assert(isinstance(chain, TransformChain)) points = PointCloud(np.random.random([10, 2])) chain_res = chain.apply(points) manual_res = tps_two.apply(tps_one.apply(points)) assert (np.all(chain_res.points == manual_res.points))