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 = TPS(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))