for testing and deploying your application
for finding and fixing issues
for empowering human code reviews
import torch
import torchio as tio
from ...utils import TorchioTestCase
class TestCopyAffine(TorchioTestCase):
"""Tests for `CopyAffine`."""
def test_missing_reference(self):
transform = tio.CopyAffine(target='missing')
with self.assertRaises(RuntimeError):
transform(self.sample_subject)
def test_wrong_target_type(self):
with self.assertRaises(ValueError):
tio.CopyAffine(target=[1])
def test_same_affine(self):
image = tio.ScalarImage(tensor=torch.rand(2, 2, 2, 2))
mask = tio.LabelMap(tensor=torch.rand(2, 2, 2, 2))
mask.affine *= 1.1
subject = tio.Subject(t1=image, mask=mask)
transform = tio.CopyAffine('t1')
transformed = transform(subject)
self.assertTensorEqual(
transformed['t1'].affine,
transformed['mask'].affine,
)
fepegar /
torchio
