def chain_compose_after_tps_test(): a = PointCloud(np.random.random([10, 2])) b = PointCloud(np.random.random([10, 2])) tps = ThinPlateSplines(a, b) t = Translation([3, 4]) s = Scale([4, 2]) chain = TransformChain([t, s]) chain_mod = chain.compose_after(tps) points = PointCloud(np.random.random([10, 2])) manual_res = s.apply(t.apply(tps.apply(points))) chain_res = chain_mod.apply(points) assert(np.all(manual_res.points == chain_res.points))
def test_transformchain_compose_after_composes_with(): tr = TransformChain([]) new_tr = tr.compose_after(Mock(spec=Transform)) assert new_tr is not tr assert len(new_tr.transforms) is 1
def transformchain_compose_after_composes_with_test(): tr = TransformChain([]) new_tr = tr.compose_after(Mock(spec=Transform)) assert (new_tr is not tr) assert (len(new_tr.transforms) is 1)