Esempio n. 1
0
def chain_compose_before_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_before(tps)

    points = PointCloud(np.random.random([10, 2]))

    manual_res = tps.apply(s.apply(t.apply(points)))
    chain_res = chain_mod.apply(points)
    assert(np.all(manual_res.points == chain_res.points))
Esempio n. 2
0
def test_transformchain_compose_before_composes_with():
    tr = TransformChain([])
    new_tr = tr.compose_before(Mock(spec=Transform))
    assert new_tr is not tr
    assert len(new_tr.transforms) is 1
Esempio n. 3
0
def transformchain_compose_before_composes_with_test():
    tr = TransformChain([])
    new_tr = tr.compose_before(Mock(spec=Transform))
    assert (new_tr is not tr)
    assert (len(new_tr.transforms) is 1)