for testing and deploying your application
for finding and fixing issues
for empowering human code reviews
import torchio
torchio
from torchio.transforms import OneOf, RandomAffine, RandomElasticDeformation
from ...utils import TorchioTestCase
class TestOneOf(TorchioTestCase):
"""Tests for `OneOf`."""
def test_wrong_input_type(self):
with self.assertRaises(ValueError):
OneOf(1)
def test_negative_probabilities(self):
OneOf({RandomAffine(): -1, RandomElasticDeformation(): 1})
def test_zero_probabilities(self):
OneOf({RandomAffine(): 0, RandomElasticDeformation(): 0})
def test_not_transform(self):
OneOf({RandomAffine: 1, RandomElasticDeformation: 2})