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)