def test_transformchain_compose_after_inplace_composes_with():
tr = TransformChain([])
ref = tr
no_return = tr.compose_after_inplace(Mock(spec=Transform))
assert no_return is None
assert ref is tr
assert len(tr.transforms) is 1
def transformchain_compose_after_inplace_composes_with_test():
tr = TransformChain([])
ref = tr
no_return = tr.compose_after_inplace(Mock(spec=Transform))
assert (no_return is None)
assert (ref is tr)
assert (len(tr.transforms) is 1)
def test_transformchain_apply():
mocked_transform = Mock()
mocked_transform._apply.return_value = 3
mocked_transform2 = Mock()
mocked_transform2._apply.return_value = 4
transforms = [mocked_transform, mocked_transform2]
tr = TransformChain(transforms)
result = tr.apply(1)
assert result == 4
def transformchain_apply_test():
mocked_transform = Mock()
mocked_transform._apply.return_value = 3
mocked_transform2 = Mock()
mocked_transform2._apply.return_value = 4
transforms = [mocked_transform, mocked_transform2]
tr = TransformChain(transforms)
result = tr.apply(1)
assert (result == 4)
def test_chain_compose_after_inplace_tps():
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.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 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))
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 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 test_transformchain_compose_after_inplace_order():
m1 = Mock(spec=Transform)
m2 = Mock(spec=Transform)
tr = TransformChain([m1])
tr.compose_after_inplace(m2)
assert tr.transforms[0] is m2
def test_transformchain_compose_before_inplace_order():
m1 = Mock(spec=Transform)
m2 = Mock(spec=Transform)
tr = TransformChain([m1])
tr.compose_before_inplace(m2)
assert (tr.transforms[1] is m2)
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)
def transformchain_compose_after_inplace_order_test():
m1 = Mock(spec=Transform)
m2 = Mock(spec=Transform)
tr = TransformChain([m1])
tr.compose_after_inplace(m2)
assert (tr.transforms[0] is m2)
def transformchain_compose_before_inplace_order_test():
m1 = Mock(spec=Transform)
m2 = Mock(spec=Transform)
tr = TransformChain([m1])
tr.compose_before_inplace(m2)
assert (tr.transforms[1] is m2)
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)
def transformchain_composes_inplace_with_test():
tr = TransformChain([])
assert (tr.composes_inplace_with == Transform)
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))
def test_transformchain_composes_inplace_with():
tr = TransformChain([])
assert tr.composes_inplace_with == Transform
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)
