for testing and deploying your application
for finding and fixing issues
for empowering human code reviews
from torchio import RandomSwap
from ...utils import TorchioTestCase
from numpy.testing import assert_array_equal
class TestRandomSwap(TorchioTestCase):
"""Tests for `RandomSwap`."""
def test_no_swap(self):
transform = RandomSwap(patch_size=5, num_iterations=0)
transformed = transform(self.sample)
assert_array_equal(self.sample.t1.data, transformed.t1.data)
def test_with_swap(self):
transform = RandomSwap(patch_size=5)
with self.assertRaises(AssertionError):
def test_wrong_num_iterations_type(self):
with self.assertRaises(TypeError):
RandomSwap(num_iterations='wrong')
def test_negative_num_iterations(self):
with self.assertRaises(ValueError):
RandomSwap(num_iterations=-1)
fepegar /
torchio
