tests.transforms.augmentation.test_random_anisotropy

Complexity

Total Complexity

12

Size/Duplication

Total Lines	40
Duplicated Lines	0 %

Importance

Changes

0

7 Methods

Rating	Name	Duplication	Size	Complexity
A	TestRandomAnisotropy.test_out_of_range_axis_in_tuple()	0	3	2
A	TestRandomAnisotropy.test_2d_rgb()	0	3	1
A	TestRandomAnisotropy.test_wrong_axes_type()	0	3	2
A	TestRandomAnisotropy.test_wrong_downsampling_type()	0	3	2
A	TestRandomAnisotropy.test_out_of_range_axis()	0	3	2
A	TestRandomAnisotropy.test_downsample()	0	7	1
A	TestRandomAnisotropy.test_below_one_downsampling()	0	3	2

import torch
from torchio import ScalarImage, RandomAnisotropy
from ...utils import TorchioTestCase


class TestRandomAnisotropy(TorchioTestCase):
    """Tests for `RandomAnisotropy`."""

    def test_downsample(self):
        transform = RandomAnisotropy(
            axes=1,
            downsampling=(2., 2.)
        )
        transformed = transform(self.sample_subject)
        self.assertEqual(self.sample_subject.spacing[1], transformed.spacing[1])

    def test_out_of_range_axis(self):
        with self.assertRaises(ValueError):
            RandomAnisotropy(axes=3)

    def test_out_of_range_axis_in_tuple(self):
        with self.assertRaises(ValueError):
            RandomAnisotropy(axes=(0, -1, 2))

    def test_wrong_axes_type(self):
        with self.assertRaises(ValueError):
            RandomAnisotropy(axes='wrong')

    def test_wrong_downsampling_type(self):
        with self.assertRaises(ValueError):
            RandomAnisotropy(downsampling='wrong')

    def test_below_one_downsampling(self):
        with self.assertRaises(ValueError):
            RandomAnisotropy(downsampling=0.2)

    def test_2d_rgb(self):
        image = ScalarImage(tensor=torch.rand(3, 4, 5, 6))
        RandomAnisotropy()(image)