tests.transforms.preprocessing.test_pad

Complexity

Total Complexity

2

Size/Duplication

Total Lines	26
Duplicated Lines	0 %

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	TestPad.test_pad()	0	11	1
A	TestPad.test_nans_history()	0	4	1

import torch
import SimpleITK as sitk
import torchio as tio
from torchio.utils import sitk_to_nib
from ...utils import TorchioTestCase


class TestPad(TorchioTestCase):
    """Tests for `Pad`."""
    def test_pad(self):
        image = self.sample_subject.t1
        padding = 1, 2, 3, 4, 5, 6
        sitk_image = image.as_sitk()
        low, high = padding[::2], padding[1::2]
        sitk_padded = sitk.ConstantPad(sitk_image, low, high, 0)
        tio_padded = tio.Pad(padding, padding_mode=0)(image)
        sitk_tensor, sitk_affine = sitk_to_nib(sitk_padded)
        tio_tensor, tio_affine = sitk_to_nib(tio_padded.as_sitk())
        self.assertTensorEqual(sitk_tensor, tio_tensor)
        self.assertTensorEqual(sitk_affine, tio_affine)

    def test_nans_history(self):
        padded = tio.Pad(1, padding_mode=2)(self.sample_subject)
        again = padded.history[0](self.sample_subject)
        assert not torch.isnan(again.t1.data).any()