for testing and deploying your application
for finding and fixing issues
for empowering human code reviews
from torchio.transforms import RandomAnisotropy
from ...utils import TorchioTestCase
class TestRandomAnisotropy(TorchioTestCase):
"""Tests for `RandomAnisotropy`."""
def test_downsample(self):
transform = RandomAnisotropy(
axes=1,
downsampling=(2., 2.)
)
transformed = transform(self.sample_subject)
self.assertEqual(self.sample_subject.spacing[1], transformed.spacing[1])
def test_out_of_range_axis(self):
with self.assertRaises(ValueError):
RandomAnisotropy(axes=3)
def test_out_of_range_axis_in_tuple(self):
RandomAnisotropy(axes=(0, -1, 2))
def test_wrong_axes_type(self):
RandomAnisotropy(axes='wrong')
def test_wrong_downsampling_type(self):
RandomAnisotropy(downsampling='wrong')
def test_below_one_downsampling(self):
RandomAnisotropy(downsampling=0.2)
fepegar /
torchio
