tests.data.sampler.test_label_sampler

Complexity

Total Complexity

2

Size/Duplication

Total Lines	13
Duplicated Lines	0 %

Importance

Changes

0

1 Method

Rating	Name	Duplication	Size	Complexity
A	TestLabelSampler.test_label_sampler()	0	4	2

from torchio import DATA
from torchio.data import LabelSampler
from ...utils import TorchioTestCase


class TestLabelSampler(TorchioTestCase):
    """Tests for `LabelSampler` class."""

    def test_label_sampler(self):
        sampler = LabelSampler(5, 'label')
        for patch in sampler(self.sample, num_patches=10):
            self.assertEqual(patch['label'][DATA][0, 2, 2, 2], 1)