tests.transforms.augmentation.test_random_motion

Complexity

Total Complexity

17

Size/Duplication

Total Lines	52
Duplicated Lines	0 %

Importance

Changes

0

9 Methods

Rating	Name	Duplication	Size	Complexity
A	TestRandomMotion.test_negative_translation()	0	3	2
A	TestRandomMotion.test_wrong_image_interpolation_type()	0	3	2
A	TestRandomMotion.test_wrong_translation_type()	0	3	2
A	TestRandomMotion.test_bad_num_transforms_value()	0	3	2
A	TestRandomMotion.test_wrong_degrees_type()	0	3	2
A	TestRandomMotion.test_with_movement()	0	7	2
A	TestRandomMotion.test_wrong_image_interpolation_value()	0	3	2
A	TestRandomMotion.test_no_movement()	0	8	1
A	TestRandomMotion.test_negative_degrees()	0	3	2

from torchio import RandomMotion
from ...utils import TorchioTestCase
from numpy.testing import assert_array_equal


class TestRandomMotion(TorchioTestCase):
    """Tests for `RandomMotion`."""
    def test_bad_num_transforms_value(self):
        with self.assertRaises(ValueError):
            RandomMotion(num_transforms=0)

    def test_no_movement(self):
        transform = RandomMotion(
            degrees=0,
            translation=0,
            num_transforms=1
        )
        transformed = transform(self.sample)
        assert_array_equal(self.sample.t1.data, transformed.t1.data)

    def test_with_movement(self):
        transform = RandomMotion(
            num_transforms=1
        )
        transformed = transform(self.sample)
        with self.assertRaises(AssertionError):
            assert_array_equal(self.sample.t1.data, transformed.t1.data)

    def test_negative_degrees(self):
        with self.assertRaises(ValueError):
            RandomMotion(degrees=-10)

    def test_wrong_degrees_type(self):
        with self.assertRaises(ValueError):
            RandomMotion(degrees='wrong')

    def test_negative_translation(self):
        with self.assertRaises(ValueError):
            RandomMotion(translation=-10)

    def test_wrong_translation_type(self):
        with self.assertRaises(ValueError):
            RandomMotion(translation='wrong')

    def test_wrong_image_interpolation_type(self):
        with self.assertRaises(TypeError):
            RandomMotion(image_interpolation=0)

    def test_wrong_image_interpolation_value(self):
        with self.assertRaises(AttributeError):
            RandomMotion(image_interpolation='wrong')