tests.data.test_image

Complexity

Total Complexity

8

Size/Duplication

Total Lines	33
Duplicated Lines	0 %

Importance

Changes

0

5 Methods

Rating	Name	Duplication	Size	Complexity
A	TestImage.test_image_not_found()	0	3	2
A	TestImage.test_wrong_path_type()	0	3	2
A	TestImage.test_tensor_flip()	0	3	1
A	TestImage.test_incompatible_arguments()	0	3	2
A	TestImage.test_tensor_affine()	0	3	1

#!/usr/bin/env python

"""Tests for Image."""

import torch

Unable to import 'torch'

from torchio import INTENSITY, Image
from ..utils import TorchioTestCase
from torchio import RandomFlip, RandomAffine

first party import "from torchio import RandomFlip, RandomAffine" should be placed before "from ..utils import TorchioTestCase"

class TestImage(TorchioTestCase):
    """Tests for `Image`."""

    def test_image_not_found(self):
        with self.assertRaises(FileNotFoundError):
            Image('nopath', INTENSITY)

    def test_wrong_path_type(self):
        with self.assertRaises(TypeError):
            Image(5, INTENSITY)

    def test_incompatible_arguments(self):
        with self.assertRaises(ValueError):
            Image(5, INTENSITY, affine=1)

    def test_tensor_flip(self):

This method could be written as a function/class method.

        sample_input = torch.ones((4, 30, 30, 30))
        RandomFlip()(sample_input)

    def test_tensor_affine(self):

This method could be written as a function/class method.

        sample_input = torch.ones((4, 10, 10, 10))
        RandomAffine()(sample_input)

Final newline missing