tests.data.test_subject

Complexity

Total Complexity

14

Size/Duplication

Total Lines	59
Duplicated Lines	0 %

Importance

Changes

0

7 Methods

Rating	Name	Duplication	Size	Complexity
A	TestSubject.test_history()	0	3	1
A	TestSubject.test_input_dict()	0	5	2
A	TestSubject.test_no_sample()	0	6	3
A	TestSubject.test_positional_args()	0	4	3
A	TestSubject.test_inconsistent_spatial_shape()	0	7	2
A	TestSubject.test_inconsistent_shape()	0	8	2
A	TestSubject.test_plot()	0	7	1

#!/usr/bin/env python

"""Tests for Subject."""

import tempfile
import torch
from torchio import Subject, ScalarImage, RandomFlip
from ..utils import TorchioTestCase


class TestSubject(TorchioTestCase):
    """Tests for `Subject`."""
    def test_positional_args(self):
        with self.assertRaises(ValueError):
            with tempfile.NamedTemporaryFile() as f:
                Subject(ScalarImage(f.name))

    def test_input_dict(self):
        with tempfile.NamedTemporaryFile() as f:
            input_dict = {'image': ScalarImage(f.name)}
            Subject(input_dict)
            Subject(**input_dict)

    def test_no_sample(self):
        with tempfile.NamedTemporaryFile() as f:
            input_dict = {'image': ScalarImage(f.name)}
            subject = Subject(input_dict)
            with self.assertRaises(RuntimeError):
                RandomFlip()(subject)

    def test_history(self):
        transformed = RandomFlip()(self.sample)
        self.assertIs(len(transformed.history), 1)

    def test_inconsistent_shape(self):
        subject = Subject(
            a=ScalarImage(tensor=torch.rand(1, 2, 3, 4)),
            b=ScalarImage(tensor=torch.rand(2, 2, 3, 4)),
        )
        subject.spatial_shape
        with self.assertRaises(RuntimeError):
            subject.shape

    def test_inconsistent_spatial_shape(self):
        subject = Subject(
            a=ScalarImage(tensor=torch.rand(1, 3, 3, 4)),
            b=ScalarImage(tensor=torch.rand(2, 2, 3, 4)),
        )
        with self.assertRaises(RuntimeError):
            subject.spatial_shape

    def test_plot(self):
        self.sample.plot(
            show=False,
            output_path=self.dir / 'figure.png',
            cmap_dict=dict(
                t2='viridis',
                label={0: 'yellow', 1: 'blue'},
            ),
        )