tests.data.test_images_dataset

Complexity

Total Complexity

28

Size/Duplication

Total Lines	88
Duplicated Lines	14.77 %

Importance

Changes

0

14 Methods

Rating	Name	Duplication	Size	Complexity
A	TestSubjectsDataset.test_wrong_transform_arg()	0	3	2
A	TestSubjectsDataset.test_data_loader()	0	8	2
A	TestSubjectsDataset.iterate_dataset()	0	5	2
A	TestSubjectsDataset.test_save_sample()	13	13	3
A	TestSubjectsDataset.test_wrong_transform_init()	0	5	2
A	TestSubjectsDataset.test_save_deprecated()	0	3	2
A	TestSubjectsDataset.test_wrong_subjects_type()	0	3	2
A	TestSubjectsDataset.test_wrong_subject_type_dict()	0	3	2
A	TestSubjectsDataset.test_wrong_subject_type_int()	0	3	2
A	TestSubjectsDataset.test_empty_subjects_tuple()	0	3	2
A	TestSubjectsDataset.test_images()	0	2	1
A	TestSubjectsDataset.test_wrong_index()	0	3	2
A	TestSubjectsDataset.test_empty_subjects_list()	0	3	2
A	TestSubjectsDataset.test_images_dataset_deprecated()	0	3	2

#!/usr/bin/env python

"""Tests for SubjectsDataset."""

import nibabel as nib
import torchio
from torchio import DATA, SubjectsDataset, ImagesDataset
from ..utils import TorchioTestCase


class TestSubjectsDataset(TorchioTestCase):
    """Tests for `SubjectsDataset`."""

    def test_images(self):
        self.iterate_dataset(self.subjects_list)

    def test_empty_subjects_list(self):
        with self.assertRaises(ValueError):
            self.iterate_dataset([])

    def test_empty_subjects_tuple(self):
        with self.assertRaises(ValueError):
            self.iterate_dataset(())

    def test_wrong_subjects_type(self):
        with self.assertRaises(TypeError):
            self.iterate_dataset(0)

    def test_wrong_subject_type_int(self):
        with self.assertRaises(TypeError):
            self.iterate_dataset([0])

    def test_wrong_subject_type_dict(self):
        with self.assertRaises(TypeError):
            self.iterate_dataset([{}])

    def test_wrong_index(self):
        with self.assertRaises(ValueError):
            self.dataset[:3]

    def test_save_sample(self):

This code seems to be duplicated in your project.

        dataset = SubjectsDataset(
            self.subjects_list, transform=lambda x: x)
        _ = len(dataset)  # for coverage
        sample = dataset[0]
        output_path = self.dir / 'test.nii.gz'
        paths_dict = {'t1': output_path}
        with self.assertWarns(DeprecationWarning):
            dataset.save_sample(sample, paths_dict)
        nii = nib.load(str(output_path))
        ndims_output = len(nii.shape)
        ndims_sample = len(sample['t1'].shape)
        assert ndims_sample == ndims_output + 1

    def test_wrong_transform_init(self):
        with self.assertRaises(ValueError):
            SubjectsDataset(
                self.subjects_list,
                transform=dict(),
            )

    def test_wrong_transform_arg(self):
        with self.assertRaises(ValueError):
            self.dataset.set_transform(1)

    @staticmethod
    def iterate_dataset(subjects_list):
        dataset = SubjectsDataset(subjects_list)
        for _ in dataset:
            pass

    def test_data_loader(self):
        from torch.utils.data import DataLoader
        subj_list = [torchio.datasets.Colin27()]
        dataset = SubjectsDataset(subj_list)
        loader = DataLoader(dataset, batch_size=1, shuffle=True)
        for batch in loader:
            batch['t1'][DATA]
            batch['brain'][DATA]

    def test_save_deprecated(self):
        with self.assertWarns(DeprecationWarning):
            self.dataset.save_sample(self.sample, {})

    def test_images_dataset_deprecated(self):
        with self.assertWarns(DeprecationWarning):
            ImagesDataset(self.subjects_list)