tests.transforms.augmentation.test_random_affine.TestRandomAffine.test_rotation_image() - Code Metrics - Inspection of "Fix random affine not rotating around center (#173..." - fepegar/torchio - Measure and Improve Code Quality continuously with Scrutinizer

Passed

Push — master ( a5fd0f...582603 )

by Fernando

created 2020-05-29 12:24 UTC

TestRandomAffine.test_rotation_image() A

↳ Parent: tests.transforms.augmentation.test_random_affine

Complexity

Conditions

Size

Total Lines	10
Code Lines	8

Duplication

Lines	0
Ratio	0 %

Importance

Changes

Metric	Value
cc	1
eloc	8
nop	1
dl	0
loc	10
rs	10
c	0
b	0
f	0

from torchio.transforms import RandomAffine
from ...utils import TorchioTestCase


class TestRandomAffine(TorchioTestCase):
    """Tests for `RandomAffine`."""
    def setUp(self):
        # Set image origin far from center
        super().setUp()
        affine = self.sample.t1.affine
        affine[:3, 3] = 1e5

    def test_rotation_image(self):
        # Rotation around image center
        transform = RandomAffine(
            degrees=(90, 90),
            default_pad_value=0,
            center='image',
        )
        transformed = transform(self.sample)
        total = transformed.t1.data.sum()
        self.assertNotEqual(total, 0)

    def test_rotation_origin(self):
        # Rotation around far away point, image should be empty
        transform = RandomAffine(
            degrees=(90, 90),
            default_pad_value=0,
            center='origin',
        )
        transformed = transform(self.sample)
        total = transformed.t1.data.sum()
        self.assertEqual(total, 0)


1			from torchio.transforms import RandomAffine
2			from ...utils import TorchioTestCase
3
4
5			class TestRandomAffine(TorchioTestCase):
6			"""Tests for `RandomAffine`."""
7			def setUp(self):
8			# Set image origin far from center
9			super().setUp()
10			affine = self.sample.t1.affine
11			affine[:3, 3] = 1e5
12
13			def test_rotation_image(self):
14			# Rotation around image center
15			transform = RandomAffine(
16			degrees=(90, 90),
17			default_pad_value=0,
18			center='image',
19			)
20			transformed = transform(self.sample)
21			total = transformed.t1.data.sum()
22			self.assertNotEqual(total, 0)
23
24			def test_rotation_origin(self):
25			# Rotation around far away point, image should be empty
26			transform = RandomAffine(
27			degrees=(90, 90),
28			default_pad_value=0,
29			center='origin',
30			)
31			transformed = transform(self.sample)
32			total = transformed.t1.data.sum()
33			self.assertEqual(total, 0)
34

fepegar / torchio

Push — master ( a5fd0f...582603 )

TestRandomAffine.test_rotation_image() A

Complexity

Size

Duplication

Importance

Duplication Side-by-Side

Filter issues like