tests.data.test_image

Complexity

Total Complexity

6

Size/Duplication

Total Lines	23
Duplicated Lines	0 %

Importance

Changes

0

3 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_bad_key()	0	3	2

#!/usr/bin/env python

"""Tests for Image."""

from torchio import INTENSITY, Image
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_bad_key(self):
        with self.assertRaises(ValueError):
            Image(5, INTENSITY, affine=1)