tests.transforms.augmentation.test_oneof

Complexity

Total Complexity

8

Size/Duplication

Total Lines	23
Duplicated Lines	0 %

Importance

Changes

0

4 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

import torchio

The import torchio seems to be unused.

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})