tests.transforms.augmentation.test_random_swap

Complexity

Total Complexity

7

Size/Duplication

Total Lines	26
Duplicated Lines	0 %

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	TestRandomSwap.test_with_swap()	0	5	2
A	TestRandomSwap.test_negative_num_iterations()	0	3	2
A	TestRandomSwap.test_wrong_num_iterations_type()	0	3	2
A	TestRandomSwap.test_no_swap()	0	4	1

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


class TestRandomSwap(TorchioTestCase):
    """Tests for `RandomSwap`."""
    def test_no_swap(self):
        transform = RandomSwap(patch_size=5, num_iterations=0)
        transformed = transform(self.sample)
        assert_array_equal(self.sample.t1.data, transformed.t1.data)

    def test_with_swap(self):
        transform = RandomSwap(patch_size=5)
        transformed = transform(self.sample)
        with self.assertRaises(AssertionError):
            assert_array_equal(self.sample.t1.data, transformed.t1.data)

    def test_wrong_num_iterations_type(self):
        with self.assertRaises(TypeError):
            RandomSwap(num_iterations='wrong')

    def test_negative_num_iterations(self):
        with self.assertRaises(ValueError):
            RandomSwap(num_iterations=-1)