Function test_downcast_returned

Wrapping/Python/tests/sitkTransformTests.py:353–376

Test python specific methods where returned type is downcasted to derived class.

()

def test_downcast_returned():
    """Test python specific methods where returned type is downcasted to derived class."""
    tx = sitk.AffineTransform(3)
    itx = tx.GetInverse()
    assert tx.__class__ == itx.__class__
    assert tx.GetDimension() == itx.GetDimension()

    tx = sitk.Euler2DTransform()
    itx = tx.GetInverse()
    assert tx.__class__ == itx.__class__
    assert tx.GetDimension() == itx.GetDimension()

    tx_list = [
        sitk.AffineTransform(2),
        sitk.Euler2DTransform(),
        sitk.TranslationTransform(2),
        sitk.AffineTransform(2),
    ]
    ctx = sitk.CompositeTransform(tx_list)

    assert ctx.GetBackTransform().__class__ == sitk.AffineTransform

    for i, itx in enumerate(tx_list):
        assert ctx.GetNthTransform(i).__class__ == itx.__class__