tests.transforms.augmentation.test_random_flip

Complexity

Total Complexity

9

Size/Duplication

Total Lines	31
Duplicated Lines	0 %

Importance

Changes

0

5 Methods

Rating	Name	Duplication	Size	Complexity
A	TestRandomFlip.test_2d()	0	7	1
A	TestRandomFlip.test_wrong_flip_probability_type()	0	3	2
A	TestRandomFlip.test_out_of_range_axis()	0	3	2
A	TestRandomFlip.test_wrong_axes_type()	0	3	2
A	TestRandomFlip.test_out_of_range_axis_in_tuple()	0	3	2

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


class TestRandomFlip(TorchioTestCase):
    """Tests for `RandomFlip`."""
    def test_2d(self):
        sample = self.make_2d(self.sample)
        transform = RandomFlip(axes=(1, 2), flip_probability=1)
        transformed = transform(sample)
        assert_array_equal(
            sample.t1.data.numpy()[:, :, ::-1, ::-1],
            transformed.t1.data.numpy())

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

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

    def test_wrong_axes_type(self):
        with self.assertRaises(ValueError):
            RandomFlip(axes=None)

    def test_wrong_flip_probability_type(self):
        with self.assertRaises(ValueError):
            RandomFlip(flip_probability='wrong')