tests.data.sampler.test_patch_sampler

Complexity

Total Complexity

5

Size/Duplication

Total Lines	20
Duplicated Lines	0 %

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	TestPatchSampler.test_bad_patch_size()	0	7	4
A	TestPatchSampler.test_extract_patch()	0	2	1

import torch
import torchio
import numpy as np
from torchio.data import PatchSampler
from ...utils import TorchioTestCase


class TestPatchSampler(TorchioTestCase):
    """Tests for `PatchSampler` class."""
    def test_bad_patch_size(self):
        with self.assertRaises(ValueError):
            PatchSampler(0)
        with self.assertRaises(ValueError):
            PatchSampler(-1)
        with self.assertRaises(ValueError):
            PatchSampler(1.5)

    def test_extract_patch(self):
        PatchSampler(1).extract_patch(self.sample, (3, 4, 5))