tests.transforms.preprocessing.test_transpose

Complexity

Total Complexity

2

Size/Duplication

Total Lines	22
Duplicated Lines	0 %

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	TestTranspose.test_transpose()	0	7	1
A	TestTranspose.test_orientation_reversed()	0	5	1

import SimpleITK as sitk

import torchio as tio

from ...utils import TorchioTestCase


class TestTranspose(TorchioTestCase):
    def test_transpose(self):
        transform = tio.Transpose()
        image = tio.ScalarImage(self.get_image_path('image'))
        transformed = transform(image)
        sitk_image = sitk.GetImageFromArray(image.numpy()[0])
        from_sitk = tio.ScalarImage.from_sitk(sitk_image)
        self.assert_tensor_equal(transformed.data, from_sitk.data)

    def test_orientation_reversed(self):
        transform = tio.Transpose()
        image = tio.ScalarImage(self.get_image_path('image'))
        transformed = transform(image)
        self.assertEqual(transformed.orientation_str, image.orientation_str[::-1])