fepegar /
torchio
| Total Complexity | 1 |
| Total Lines | 20 |
| Duplicated Lines | 0 % |
| Changes | 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 |
