tests.transforms.preprocessing.test_ensure_shape_multiple

Complexity

Total Complexity

5

Size/Duplication

Total Lines	32
Duplicated Lines	0 %

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	TestEnsureShapeMultiple.test_2d()	0	7	1
A	TestEnsureShapeMultiple.test_bad_method()	0	3	2
A	TestEnsureShapeMultiple.test_crop()	0	6	1
A	TestEnsureShapeMultiple.test_pad()	0	6	1

import torchio as tio
from ...utils import TorchioTestCase


class TestEnsureShapeMultiple(TorchioTestCase):

    def test_bad_method(self):
        with self.assertRaises(ValueError):
            tio.EnsureShapeMultiple(1, method='bad')

    def test_pad(self):
        sample_t1 = self.sample_subject.t1
        assert sample_t1.shape == (1, 10, 20, 30)
        transform = tio.EnsureShapeMultiple(4, method='pad')
        transformed = transform(sample_t1)
        assert transformed.shape == (1, 12, 20, 32)

    def test_crop(self):
        sample_t1 = self.sample_subject.t1
        assert sample_t1.shape == (1, 10, 20, 30)
        transform = tio.EnsureShapeMultiple(4, method='crop')
        transformed = transform(sample_t1)
        assert transformed.shape == (1, 8, 20, 28)

    def test_2d(self):
        sample_t1 = self.sample_subject.t1
        sample_2d = sample_t1.data[..., :1]
        assert sample_2d.shape == (1, 10, 20, 1)
        transform = tio.EnsureShapeMultiple(4, method='crop')
        transformed = transform(sample_2d)
        assert transformed.shape == (1, 8, 20, 1)