tests.transforms.augmentation.test_oneof

Complexity

Total Complexity

9

Size/Duplication

Total Lines	28
Duplicated Lines	0 %

Importance

Changes

0

5 Methods

Rating	Name	Duplication	Size	Complexity
A	TestOneOf.test_wrong_input_type()	0	3	2
A	TestOneOf.test_negative_probabilities()	0	3	2
A	TestOneOf.test_zero_probabilities()	0	3	2
A	TestOneOf.test_not_transform()	0	3	2
A	TestOneOf.test_one_of()	0	4	1

import torchio
from torchio.transforms import OneOf, RandomAffine, RandomElasticDeformation
from ...utils import TorchioTestCase


class TestOneOf(TorchioTestCase):
    """Tests for `OneOf`."""
    def test_wrong_input_type(self):
        with self.assertRaises(ValueError):
            OneOf(1)

    def test_negative_probabilities(self):
        with self.assertRaises(ValueError):
            OneOf({RandomAffine(): -1, RandomElasticDeformation(): 1})

    def test_zero_probabilities(self):
        with self.assertRaises(ValueError):
            OneOf({RandomAffine(): 0, RandomElasticDeformation(): 0})

    def test_not_transform(self):
        with self.assertRaises(ValueError):
            OneOf({RandomAffine: 1, RandomElasticDeformation: 2})

    def test_one_of(self):
        transform = OneOf(
            {RandomAffine(): 0.2, RandomElasticDeformation(): 0.8})
        transform(self.sample)