Passed
Push — master ( b9ac52...6aebda )
by Fernando
10:37 queued 20s
created

TestSequentialLabels.test_sequential()   A

Complexity

Conditions 1

Size

Total Lines 13
Code Lines 10

Duplication

Lines 0
Ratio 0 %

Importance

Changes 0
Metric Value
cc 1
eloc 10
nop 1
dl 0
loc 13
rs 9.9
c 0
b 0
f 0
1
from torchio.transforms import SequentialLabels
2
from ...utils import TorchioTestCase
3
4
5
class TestSequentialLabels(TorchioTestCase):
6
    """Tests for `SequentialLabels`."""
7
    def test_sequential(self):
8
        initial_labels = (2, 8, 9, 10, 15, 20, 100)
9
        transformed_labels = (1, 2, 3, 4, 5, 6, 7)
10
11
        sequential_labels = SequentialLabels()
12
13
        subject = self.get_subject_with_labels(labels=initial_labels)
14
        transformed = sequential_labels(subject)
15
        inverse_transformed = transformed.apply_inverse_transform()
16
17
        self.assertEqual(self.get_unique_labels(subject.label), set(initial_labels))
18
        self.assertEqual(self.get_unique_labels(transformed.label), set(transformed_labels))
19
        self.assertEqual(self.get_unique_labels(inverse_transformed.label), set(initial_labels))
20