tests.data.inference.test_grid_sampler

Complexity

Total Complexity

4

Size/Duplication

Total Lines	33
Duplicated Lines	0 %

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	TestGridSampler.test_large_window()	0	3	2
A	TestGridSampler.test_single_location()	0	4	1
A	TestGridSampler.test_locations()	0	14	1

#!/usr/bin/env python

from torchio.data import GridSampler
from ...utils import TorchioTestCase


class TestGridSampler(TorchioTestCase):
    """Tests for `GridSampler`."""

    def test_locations(self):
        sampler = GridSampler(self.sample, (5, 20, 20), (1, 2, 3))
        fixture = [
            [0, 0, 0, 5, 20, 20],
            [0, 0, 10, 5, 20, 30],
            [0, 0, 5, 5, 20, 25],
            [3, 0, 0, 8, 20, 20],
            [3, 0, 10, 8, 20, 30],
            [3, 0, 5, 8, 20, 25],
            [5, 0, 0, 10, 20, 20],
            [5, 0, 10, 10, 20, 30],
            [5, 0, 5, 10, 20, 25],
        ]
        self.assertEqual(sampler.locations.tolist(), fixture)

    def test_large_window(self):
        with self.assertRaises(ValueError):
            GridSampler(self.sample, (5, 21, 5), (1, 2, 3))

    def test_single_location(self):
        sampler = GridSampler(self.sample, (10, 20, 30), 0)
        fixture = [[0, 0, 0, 10, 20, 30]]
        self.assertEqual(sampler.locations.tolist(), fixture)