tests.data.sampler.test_random_sampler

Complexity

Total Complexity

3

Size/Duplication

Total Lines	16
Duplicated Lines	0 %

Importance

Changes

0

1 Method

Rating	Name	Duplication	Size	Complexity
A	TestRandomSampler.test_not_implemented()	0	6	3

import torch
import torchio
import numpy as np
from torchio.data.sampler import RandomSampler
from ...utils import TorchioTestCase


class TestRandomSampler(TorchioTestCase):
    """Tests for `RandomSampler` class."""
    def test_not_implemented(self):
        sampler = RandomSampler(1)
        with self.assertRaises(NotImplementedError):
            sampler(self.sample, 5)
        with self.assertRaises(NotImplementedError):
            sampler.get_probability_map(self.sample)