test.unit.test_layer - Code Metrics - Inspection of "640 refactor model layers" - DeepRegNet/DeepReg - Measure and Improve Code Quality continuously with Scrutinizer

Passed

Pull Request — main (#656)

by Yunguan

created 2021-03-04 22:47 UTC

test.unit.test_layer A

↳ Parent: Project

Complexity

Total Complexity

Size/Duplication

Total Lines	345
Duplicated Lines	0 %

Importance

Changes

Metric	Value
wmc	34
eloc	241
dl	0
loc	345
rs	9.68
c	0
b	0
f	0

16 Methods

Rating	Name	Size	Complexity
A	TestIntDVF.test_forward()	21	1
A	TestWarping.test_forward()	13	1
A	TestIntDVF.test_err()	3	2
A	TestWarping.test_get_config()	8	1
A	TestBSplines3DTransform.test_build()	14	1
A	TestResizeCPTransform.test_call()	15	1
A	TestResize3d.test_get_config()	9	1
A	TestResizeCPTransform.test_build()	12	1
A	TestResize3d.test_shape_err()	3	2
A	TestBSplines3DTransform.test_init()	11	2
A	TestBSplines3DTransform.test_coefficients()	12	1
A	TestBSplines3DTransform.test_interpolation()	19	1
D	TestBSplines3DTransform.generate_filter_coefficients()	44	12
A	TestResize3d.test_forward()	17	1
A	TestResize3d.test_image_shape_err()	6	2
A	TestResizeCPTransform.test_attributes()	10	2

2 Functions

Rating	Name	Duplication	Size	Complexity
A	test_res_block()	0	40	1
A	test_norm_block()	0	35	1

# coding=utf-8

"""
Tests for deepreg/model/layer
"""

import numpy as np
import pytest
import tensorflow as tf

import deepreg.model.layer as layer


@pytest.mark.parametrize("layer_name", ["conv3d", "deconv3d"])
@pytest.mark.parametrize("norm_name", ["batch", "layer"])
@pytest.mark.parametrize("activation", ["relu", "elu"])
def test_norm_block(layer_name: str, norm_name: str, activation: str):
    """
    Test output shapes and configs.

    :param layer_name: layer_name for layer definition
    :param norm_name: norm_name for layer definition
    :param activation: activation for layer definition
    """
    input_size = (2, 3, 4, 5, 6)  # (batch, *shape, ch)
    norm_block = layer.NormBlock(
        layer_name=layer_name,
        norm_name=norm_name,
        activation=activation,
        filters=3,
        kernel_size=1,
        padding="same",
    )
    inputs = tf.ones(shape=input_size)
    outputs = norm_block(inputs)
    assert outputs.shape == input_size[:-1] + (3,)

    config = norm_block.get_config()
    assert config == dict(
        layer_name=layer_name,
        norm_name=norm_name,
        activation=activation,
        filters=3,
        kernel_size=1,
        padding="same",
        name="norm_block",
        trainable=True,
        dtype="float32",
    )


class TestWarping:

    @pytest.mark.parametrize(
        ("moving_image_size", "fixed_image_size"),
        [
            ((1, 2, 3), (3, 4, 5)),
            ((1, 2, 3), (1, 2, 3)),
        ],

    )
    def test_forward(self, moving_image_size, fixed_image_size):
        batch_size = 2
        image = tf.ones(shape=(batch_size,) + moving_image_size)
        ddf = tf.ones(shape=(batch_size,) + fixed_image_size + (3,))
        outputs = layer.Warping(fixed_image_size=fixed_image_size)([ddf, image])
        assert outputs.shape == (batch_size, *fixed_image_size)

    def test_get_config(self):

        warping = layer.Warping(fixed_image_size=(2, 3, 4))
        config = warping.get_config()
        assert config == dict(
            fixed_image_size=(2, 3, 4),
            name="warping",
            trainable=True,
            dtype="float32",
        )


@pytest.mark.parametrize("layer_name", ["conv3d", "deconv3d"])
@pytest.mark.parametrize("norm_name", ["batch", "layer"])
@pytest.mark.parametrize("activation", ["relu", "elu"])
@pytest.mark.parametrize("num_layers", [2, 3])
def test_res_block(layer_name: str, norm_name: str, activation: str, num_layers: int):
    """
    Test output shapes and configs.

    :param layer_name: layer_name for layer definition
    :param norm_name: norm_name for layer definition
    :param activation: activation for layer definition
    :param num_layers: number of blocks in res block
    """
    ch = 3
    input_size = (2, 3, 4, 5, ch)  # (batch, *shape, ch)
    res_block = layer.ResidualBlock(
        layer_name=layer_name,
        num_layers=num_layers,
        norm_name=norm_name,
        activation=activation,
        filters=ch,
        kernel_size=3,
        padding="same",
    )
    inputs = tf.ones(shape=input_size)
    outputs = res_block(inputs)
    assert outputs.shape == input_size[:-1] + (3,)

    config = res_block.get_config()
    assert config == dict(
        layer_name=layer_name,
        num_layers=num_layers,
        norm_name=norm_name,
        activation=activation,
        filters=ch,
        kernel_size=3,
        padding="same",
        name="res_block",
        trainable=True,
        dtype="float32",
    )


class TestIntDVF:

    def test_forward(self):
        """
        Test output shape and config.
        """

        fixed_image_size = (8, 9, 10)
        input_shape = (2, *fixed_image_size, 3)

        int_layer = layer.IntDVF(fixed_image_size=fixed_image_size)

        inputs = tf.ones(shape=input_shape)
        outputs = int_layer(inputs)
        assert outputs.shape == input_shape

        config = int_layer.get_config()
        assert config == dict(
            fixed_image_size=fixed_image_size,
            num_steps=7,
            name="int_dvf",
            trainable=True,
            dtype="float32",
        )

    def test_err(self):

        with pytest.raises(AssertionError):
            layer.IntDVF(fixed_image_size=(2, 3))


class TestResizeCPTransform:

    @pytest.mark.parametrize(
        "parameter,cp_spacing", [((8, 8, 8), 8), ((8, 24, 16), (8, 24, 16))]
    )

    def test_attributes(self, parameter, cp_spacing):
        model = layer.ResizeCPTransform(cp_spacing)

        if isinstance(cp_spacing, int):
            cp_spacing = [cp_spacing] * 3
        assert list(model.cp_spacing) == list(parameter)
        assert model.kernel_sigma == [0.44 * cp for cp in cp_spacing]

    @pytest.mark.parametrize(
        "input_size,output_size,cp_spacing",
        [
            ((1, 8, 8, 8, 3), (12, 8, 12), (8, 16, 8)),
            ((1, 8, 8, 8, 3), (12, 12, 12), 8),
        ],

    )
    def test_build(self, input_size, output_size, cp_spacing):
        model = layer.ResizeCPTransform(cp_spacing)
        model.build(input_size)

        assert [a == b for a, b, in zip(model._output_shape, output_size)]
class MyParent:
    def __init__(self):
        self._x = 1;
        self.y = 2;

class MyChild(MyParent):
    def some_method(self):
        return self._x    # Ok, since accessed from a child class

class AnotherClass:
    def some_method(self, instance_of_my_child):
        return instance_of_my_child._x   # Would be flagged as AnotherClass is not
                                         # a child class of MyParent

    @pytest.mark.parametrize(
        "input_size,output_size,cp_spacing",
        [
            ((1, 68, 68, 68, 3), (1, 12, 8, 12, 3), (8, 16, 8)),
            ((1, 68, 68, 68, 3), (1, 12, 12, 12, 3), 8),
        ],

    )
    def test_call(self, input_size, output_size, cp_spacing):
        model = layer.ResizeCPTransform(cp_spacing)
        model.build(input_size)

        input = tf.random.normal(shape=input_size, dtype=tf.float32)

        output = model(input)

        assert output.shape == output_size


class TestBSplines3DTransform:
    """
    Test the layer.BSplines3DTransform class,
    its default attributes and its call() function.
    """

    @pytest.mark.parametrize(
        "input_size,cp",
        [((1, 8, 8, 8, 3), 8), ((1, 8, 8, 8, 3), (8, 16, 12))],
    )

    def test_init(self, input_size, cp):
        model = layer.BSplines3DTransform(cp, input_size[1:-1])

        if isinstance(cp, int):
            cp = (cp, cp, cp)

        assert model.cp_spacing == cp

    @pytest.mark.parametrize(
        "input_size,cp",
        [((1, 8, 8, 8, 3), (8, 8, 8)), ((1, 8, 8, 8, 3), (8, 16, 12))],
    )

    def generate_filter_coefficients(self, cp_spacing):

        b = {
            0: lambda u: np.float64((1 - u) ** 3 / 6),
            1: lambda u: np.float64((3 * (u ** 3) - 6 * (u ** 2) + 4) / 6),
            2: lambda u: np.float64((-3 * (u ** 3) + 3 * (u ** 2) + 3 * u + 1) / 6),
            3: lambda u: np.float64(u ** 3 / 6),
        }

        filters = np.zeros(
            (
                4 * cp_spacing[0],
                4 * cp_spacing[1],
                4 * cp_spacing[2],
                3,
                3,
            ),
            dtype=np.float32,
        )

        for u in range(cp_spacing[0]):

            for v in range(cp_spacing[1]):
                for w in range(cp_spacing[2]):
                    for x in range(4):
                        for y in range(4):
                            for z in range(4):
                                for it_dim in range(3):
                                    u_norm = 1 - (u + 0.5) / cp_spacing[0]
                                    v_norm = 1 - (v + 0.5) / cp_spacing[1]
                                    w_norm = 1 - (w + 0.5) / cp_spacing[2]
                                    filters[
                                        x * cp_spacing[0] + u,
                                        y * cp_spacing[1] + v,
                                        z * cp_spacing[2] + w,
                                        it_dim,
                                        it_dim,
                                    ] = (
                                        b[x](u_norm) * b[y](v_norm) * b[z](w_norm)
                                    )
        return filters

    @pytest.mark.parametrize(
        "input_size,cp",
        [((1, 8, 8, 8, 3), (8, 8, 8)), ((1, 8, 8, 8, 3), (8, 16, 12))],
    )

    def test_build(self, input_size, cp):
        model = layer.BSplines3DTransform(cp, input_size[1:-1])

        model.build(input_size)
        assert model.filter.shape == (
            4 * cp[0],
            4 * cp[1],
            4 * cp[2],
            3,
            3,
        )

    @pytest.mark.parametrize(
        "input_size,cp",
        [((1, 8, 8, 8, 3), (8, 8, 8)), ((1, 8, 8, 8, 3), (8, 16, 12))],
    )

    def test_coefficients(self, input_size, cp):

        filters = self.generate_filter_coefficients(cp_spacing=cp)

        model = layer.BSplines3DTransform(cp, input_size[1:-1])
        model.build(input_size)

        assert np.allclose(filters, model.filter.numpy(), atol=1e-8)

    @pytest.mark.parametrize(
        "input_size,cp",
        [((1, 8, 8, 8, 3), (8, 8, 8)), ((1, 8, 8, 8, 3), (8, 16, 12))],
    )

    def test_interpolation(self, input_size, cp):
        model = layer.BSplines3DTransform(cp, input_size[1:-1])
        model.build(input_size)

        vol_shape = input_size[1:-1]
        num_cp = (
            [input_size[0]]
            + [int(np.ceil(isize / cpsize) + 3) for isize, cpsize in zip(vol_shape, cp)]
            + [input_size[-1]]
        )

        field = tf.random.normal(shape=num_cp, dtype=tf.float32)

        ddf = model.call(field)
        assert ddf.shape == input_size


class TestResize3d:

    @pytest.mark.parametrize(
        ("input_shape", "resize_shape", "output_shape"),
        [
            ((1, 2, 3), (3, 4, 5), (3, 4, 5)),
            ((2, 1, 2, 3), (3, 4, 5), (2, 3, 4, 5)),
            ((2, 1, 2, 3, 1), (3, 4, 5), (2, 3, 4, 5, 1)),
            ((2, 1, 2, 3, 6), (3, 4, 5), (2, 3, 4, 5, 6)),
            ((1, 2, 3), (1, 2, 3), (1, 2, 3)),
            ((2, 1, 2, 3), (1, 2, 3), (2, 1, 2, 3)),
            ((2, 1, 2, 3, 1), (1, 2, 3), (2, 1, 2, 3, 1)),
            ((2, 1, 2, 3, 6), (1, 2, 3), (2, 1, 2, 3, 6)),
        ],

    )
    def test_forward(self, input_shape, resize_shape, output_shape):
        inputs = tf.ones(shape=input_shape)
        outputs = layer.Resize3d(shape=resize_shape)(inputs)
        assert outputs.shape == output_shape

    def test_get_config(self):

        resize = layer.Resize3d(shape=(2, 3, 4))
        config = resize.get_config()
        assert config == dict(
            shape=(2, 3, 4),
            method=tf.image.ResizeMethod.BILINEAR,
            name="resize3d",
            trainable=True,
            dtype="float32",
        )

    def test_shape_err(self):

        with pytest.raises(AssertionError):
            layer.Resize3d(shape=(2, 3))

    def test_image_shape_err(self):

        with pytest.raises(ValueError) as err_info:
            resize = layer.Resize3d(shape=(2, 3, 4))
            resize(tf.ones(1, 1))
        assert "Resize3d takes input image of dimension 3 or 4 or 5" in str(
            err_info.value
        )


1			# coding=utf-8
2
3			"""
4			Tests for deepreg/model/layer
5			"""
6
7			import numpy as np
8			import pytest
9			import tensorflow as tf
10
11			import deepreg.model.layer as layer
12
13
14			@pytest.mark.parametrize("layer_name", ["conv3d", "deconv3d"])
15			@pytest.mark.parametrize("norm_name", ["batch", "layer"])
16			@pytest.mark.parametrize("activation", ["relu", "elu"])
17			def test_norm_block(layer_name: str, norm_name: str, activation: str):
18			"""
19			Test output shapes and configs.
20
21			:param layer_name: layer_name for layer definition
22			:param norm_name: norm_name for layer definition
23			:param activation: activation for layer definition
24			"""
25			input_size = (2, 3, 4, 5, 6) # (batch, *shape, ch)
26			norm_block = layer.NormBlock(
27			layer_name=layer_name,
28			norm_name=norm_name,
29			activation=activation,
30			filters=3,
31			kernel_size=1,
32			padding="same",
33			)
34			inputs = tf.ones(shape=input_size)
35			outputs = norm_block(inputs)
36			assert outputs.shape == input_size[:-1] + (3,)
37
38			config = norm_block.get_config()
39			assert config == dict(
40			layer_name=layer_name,
41			norm_name=norm_name,
42			activation=activation,
43			filters=3,
44			kernel_size=1,
45			padding="same",
46			name="norm_block",
47			trainable=True,
48			dtype="float32",
49			)
50
51
52			class TestWarping:
			0 ignored issues – show introduced 2021-01-26 01:41 UTC by Report Bug Copy Issue Report Missing class docstring Loading history...
53			@pytest.mark.parametrize(
54			("moving_image_size", "fixed_image_size"),
55			[
56			((1, 2, 3), (3, 4, 5)),
57			((1, 2, 3), (1, 2, 3)),
58			],
			0 ignored issues – show introduced 2021-01-26 01:41 UTC by Report Bug Copy Issue Report Missing function or method docstring Loading history...
59			)
60			def test_forward(self, moving_image_size, fixed_image_size):
61			batch_size = 2
62			image = tf.ones(shape=(batch_size,) + moving_image_size)
63			ddf = tf.ones(shape=(batch_size,) + fixed_image_size + (3,))
64			outputs = layer.Warping(fixed_image_size=fixed_image_size)([ddf, image])
65			assert outputs.shape == (batch_size, *fixed_image_size)
66
67			def test_get_config(self):
			0 ignored issues – show introduced 2021-01-26 01:41 UTC by Report Bug Copy Issue Report Missing function or method docstring Loading history...
68			warping = layer.Warping(fixed_image_size=(2, 3, 4))
69			config = warping.get_config()
70			assert config == dict(
71			fixed_image_size=(2, 3, 4),
72			name="warping",
73			trainable=True,
74			dtype="float32",
75			)
76
77
78			@pytest.mark.parametrize("layer_name", ["conv3d", "deconv3d"])
79			@pytest.mark.parametrize("norm_name", ["batch", "layer"])
80			@pytest.mark.parametrize("activation", ["relu", "elu"])
81			@pytest.mark.parametrize("num_layers", [2, 3])
82			def test_res_block(layer_name: str, norm_name: str, activation: str, num_layers: int):
83			"""
84			Test output shapes and configs.
85
86			:param layer_name: layer_name for layer definition
87			:param norm_name: norm_name for layer definition
88			:param activation: activation for layer definition
89			:param num_layers: number of blocks in res block
90			"""
91			ch = 3
92			input_size = (2, 3, 4, 5, ch) # (batch, *shape, ch)
93			res_block = layer.ResidualBlock(
94			layer_name=layer_name,
95			num_layers=num_layers,
96			norm_name=norm_name,
97			activation=activation,
98			filters=ch,
99			kernel_size=3,
100			padding="same",
101			)
102			inputs = tf.ones(shape=input_size)
103			outputs = res_block(inputs)
104			assert outputs.shape == input_size[:-1] + (3,)
105
106			config = res_block.get_config()
107			assert config == dict(
108			layer_name=layer_name,
109			num_layers=num_layers,
110			norm_name=norm_name,
111			activation=activation,
112			filters=ch,
113			kernel_size=3,
114			padding="same",
115			name="res_block",
116			trainable=True,
117			dtype="float32",
118			)
119
120
121			class TestIntDVF:
			0 ignored issues – show introduced 2021-01-26 01:41 UTC by Report Bug Copy Issue Report Missing class docstring Loading history...
122			def test_forward(self):
123			"""
124			Test output shape and config.
125			"""
126
127			fixed_image_size = (8, 9, 10)
128			input_shape = (2, *fixed_image_size, 3)
129
130			int_layer = layer.IntDVF(fixed_image_size=fixed_image_size)
131
132			inputs = tf.ones(shape=input_shape)
133			outputs = int_layer(inputs)
134			assert outputs.shape == input_shape
135
136			config = int_layer.get_config()
137			assert config == dict(
138			fixed_image_size=fixed_image_size,
139			num_steps=7,
140			name="int_dvf",
141			trainable=True,
142			dtype="float32",
143			)
144
145			def test_err(self):
			0 ignored issues – show introduced 2021-01-26 01:41 UTC by Report Bug Copy Issue Report Missing function or method docstring Loading history...
146			with pytest.raises(AssertionError):
147			layer.IntDVF(fixed_image_size=(2, 3))
148
149
150			class TestResizeCPTransform:
			0 ignored issues – show introduced 2021-02-02 00:20 UTC by Report Bug Copy Issue Report Missing class docstring Loading history...
151			@pytest.mark.parametrize(
152			"parameter,cp_spacing", [((8, 8, 8), 8), ((8, 24, 16), (8, 24, 16))]
153			)
			0 ignored issues – show introduced 2021-01-26 01:41 UTC by Report Bug Copy Issue Report Missing function or method docstring Loading history...
154			def test_attributes(self, parameter, cp_spacing):
155			model = layer.ResizeCPTransform(cp_spacing)
156
157			if isinstance(cp_spacing, int):
158			cp_spacing = [cp_spacing] * 3
159			assert list(model.cp_spacing) == list(parameter)
160			assert model.kernel_sigma == [0.44 * cp for cp in cp_spacing]
161
162			@pytest.mark.parametrize(
163			"input_size,output_size,cp_spacing",
164			[
165			((1, 8, 8, 8, 3), (12, 8, 12), (8, 16, 8)),
166			((1, 8, 8, 8, 3), (12, 12, 12), 8),
167			],
			0 ignored issues – show introduced 2021-01-26 01:41 UTC by Report Bug Copy Issue Report Missing function or method docstring Loading history...
168			)
169			def test_build(self, input_size, output_size, cp_spacing):
170			model = layer.ResizeCPTransform(cp_spacing)
171			model.build(input_size)
172
173			assert [a == b for a, b, in zip(model._output_shape, output_size)]
			0 ignored issues – show Coding Style Best Practice introduced 2021-01-26 01:41 UTC by Report Bug Copy Issue Report It seems like `_output_shape` was declared protected and should not be accessed from this context. Prefixing a member variable `_` is usually regarded as the equivalent of declaring it with protected visibility that exists in other languages. Consequentially, such a member should only be accessed from the same class or a child class: class MyParent: def __init__(self): self._x = 1; self.y = 2; class MyChild(MyParent): def some_method(self): return self._x # Ok, since accessed from a child class class AnotherClass: def some_method(self, instance_of_my_child): return instance_of_my_child._x # Would be flagged as AnotherClass is not # a child class of MyParent Loading history...
174
175			@pytest.mark.parametrize(
176			"input_size,output_size,cp_spacing",
177			[
178			((1, 68, 68, 68, 3), (1, 12, 8, 12, 3), (8, 16, 8)),
179			((1, 68, 68, 68, 3), (1, 12, 12, 12, 3), 8),
180			],
			0 ignored issues – show introduced 2021-01-26 01:41 UTC by Report Bug Copy Issue Report Missing function or method docstring Loading history...
181			)
182			def test_call(self, input_size, output_size, cp_spacing):
183			model = layer.ResizeCPTransform(cp_spacing)
184			model.build(input_size)
185
186			input = tf.random.normal(shape=input_size, dtype=tf.float32)
			0 ignored issues – show Bug Best Practice introduced 2021-01-26 01:41 UTC by Report Bug Copy Issue Report This seems to re-define the built-in `input`. It is generally discouraged to redefine built-ins as this makes code very hard to read. Loading history...
187			output = model(input)
188
189			assert output.shape == output_size
190
191
192			class TestBSplines3DTransform:
193			"""
194			Test the layer.BSplines3DTransform class,
195			its default attributes and its call() function.
196			"""
197
198			@pytest.mark.parametrize(
199			"input_size,cp",
200			[((1, 8, 8, 8, 3), 8), ((1, 8, 8, 8, 3), (8, 16, 12))],
201			)
			0 ignored issues – show introduced 2021-01-26 01:41 UTC by Report Bug Copy Issue Report Missing function or method docstring Loading history...
202			def test_init(self, input_size, cp):
203			model = layer.BSplines3DTransform(cp, input_size[1:-1])
204
205			if isinstance(cp, int):
206			cp = (cp, cp, cp)
207
208			assert model.cp_spacing == cp
209
210			@pytest.mark.parametrize(
211			"input_size,cp",
212			[((1, 8, 8, 8, 3), (8, 8, 8)), ((1, 8, 8, 8, 3), (8, 16, 12))],
213			)
			0 ignored issues – show introduced 2021-01-26 01:41 UTC by Report Bug Copy Issue Report Missing function or method docstring Loading history...
214			def generate_filter_coefficients(self, cp_spacing):
215
216			b = {
217			0: lambda u: np.float64((1 - u) ** 3 / 6),
218			1: lambda u: np.float64((3 * (u ** 3) - 6 * (u ** 2) + 4) / 6),
219			2: lambda u: np.float64((-3 * (u ** 3) + 3 * (u ** 2) + 3 * u + 1) / 6),
220			3: lambda u: np.float64(u ** 3 / 6),
221			}
222
223			filters = np.zeros(
224			(
225			4 * cp_spacing[0],
226			4 * cp_spacing[1],
227			4 * cp_spacing[2],
228			3,
229			3,
230			),
231			dtype=np.float32,
232			)
233
234			for u in range(cp_spacing[0]):
			0 ignored issues – show unused-code introduced 2021-01-26 01:41 UTC by Report Bug Copy Issue Report Too many nested blocks (7/5) Loading history...
235			for v in range(cp_spacing[1]):
236			for w in range(cp_spacing[2]):
237			for x in range(4):
238			for y in range(4):
239			for z in range(4):
240			for it_dim in range(3):
241			u_norm = 1 - (u + 0.5) / cp_spacing[0]
242			v_norm = 1 - (v + 0.5) / cp_spacing[1]
243			w_norm = 1 - (w + 0.5) / cp_spacing[2]
244			filters[
245			x * cp_spacing[0] + u,
246			y * cp_spacing[1] + v,
247			z * cp_spacing[2] + w,
248			it_dim,
249			it_dim,
250			] = (
251			b[x](u_norm) * b[y](v_norm) * b[z](w_norm)
252			)
253			return filters
254
255			@pytest.mark.parametrize(
256			"input_size,cp",
257			[((1, 8, 8, 8, 3), (8, 8, 8)), ((1, 8, 8, 8, 3), (8, 16, 12))],
258			)
			0 ignored issues – show introduced 2021-01-26 01:41 UTC by Report Bug Copy Issue Report Missing function or method docstring Loading history...
259			def test_build(self, input_size, cp):
260			model = layer.BSplines3DTransform(cp, input_size[1:-1])
261
262			model.build(input_size)
263			assert model.filter.shape == (
264			4 * cp[0],
265			4 * cp[1],
266			4 * cp[2],
267			3,
268			3,
269			)
270
271			@pytest.mark.parametrize(
272			"input_size,cp",
273			[((1, 8, 8, 8, 3), (8, 8, 8)), ((1, 8, 8, 8, 3), (8, 16, 12))],
274			)
			0 ignored issues – show introduced 2021-01-30 23:31 UTC by Report Bug Copy Issue Report Missing function or method docstring Loading history...
275			def test_coefficients(self, input_size, cp):
276
277			filters = self.generate_filter_coefficients(cp_spacing=cp)
278
279			model = layer.BSplines3DTransform(cp, input_size[1:-1])
280			model.build(input_size)
281
282			assert np.allclose(filters, model.filter.numpy(), atol=1e-8)
283
284			@pytest.mark.parametrize(
285			"input_size,cp",
286			[((1, 8, 8, 8, 3), (8, 8, 8)), ((1, 8, 8, 8, 3), (8, 16, 12))],
287			)
			0 ignored issues – show introduced 2021-02-24 21:19 UTC by Report Bug Copy Issue Report Missing function or method docstring Loading history...
288			def test_interpolation(self, input_size, cp):
289			model = layer.BSplines3DTransform(cp, input_size[1:-1])
290			model.build(input_size)
291
292			vol_shape = input_size[1:-1]
293			num_cp = (
294			[input_size[0]]
295			+ [int(np.ceil(isize / cpsize) + 3) for isize, cpsize in zip(vol_shape, cp)]
296			+ [input_size[-1]]
297			)
298
299			field = tf.random.normal(shape=num_cp, dtype=tf.float32)
300
301			ddf = model.call(field)
302			assert ddf.shape == input_size
303
304
305			class TestResize3d:
			0 ignored issues – show introduced 2021-01-30 23:31 UTC by Report Bug Copy Issue Report Missing class docstring Loading history...
306			@pytest.mark.parametrize(
307			("input_shape", "resize_shape", "output_shape"),
308			[
309			((1, 2, 3), (3, 4, 5), (3, 4, 5)),
310			((2, 1, 2, 3), (3, 4, 5), (2, 3, 4, 5)),
311			((2, 1, 2, 3, 1), (3, 4, 5), (2, 3, 4, 5, 1)),
312			((2, 1, 2, 3, 6), (3, 4, 5), (2, 3, 4, 5, 6)),
313			((1, 2, 3), (1, 2, 3), (1, 2, 3)),
314			((2, 1, 2, 3), (1, 2, 3), (2, 1, 2, 3)),
315			((2, 1, 2, 3, 1), (1, 2, 3), (2, 1, 2, 3, 1)),
316			((2, 1, 2, 3, 6), (1, 2, 3), (2, 1, 2, 3, 6)),
317			],
			0 ignored issues – show introduced 2021-01-30 23:31 UTC by Report Bug Copy Issue Report Missing function or method docstring Loading history...
318			)
319			def test_forward(self, input_shape, resize_shape, output_shape):
320			inputs = tf.ones(shape=input_shape)
321			outputs = layer.Resize3d(shape=resize_shape)(inputs)
322			assert outputs.shape == output_shape
323
324			def test_get_config(self):
			0 ignored issues – show introduced 2021-01-30 23:31 UTC by Report Bug Copy Issue Report Missing function or method docstring Loading history...
325			resize = layer.Resize3d(shape=(2, 3, 4))
326			config = resize.get_config()
327			assert config == dict(
328			shape=(2, 3, 4),
329			method=tf.image.ResizeMethod.BILINEAR,
330			name="resize3d",
331			trainable=True,
332			dtype="float32",
333			)
334
335			def test_shape_err(self):
			0 ignored issues – show introduced 2021-01-31 19:38 UTC by Report Bug Copy Issue Report Missing function or method docstring Loading history...
336			with pytest.raises(AssertionError):
337			layer.Resize3d(shape=(2, 3))
338
339			def test_image_shape_err(self):
			0 ignored issues – show introduced 2021-01-30 23:31 UTC by Report Bug Copy Issue Report Missing function or method docstring Loading history...
340			with pytest.raises(ValueError) as err_info:
341			resize = layer.Resize3d(shape=(2, 3, 4))
342			resize(tf.ones(1, 1))
343			assert "Resize3d takes input image of dimension 3 or 4 or 5" in str(
344			err_info.value
345			)
346

DeepRegNet / DeepReg

Pull Request — main (#656)

test.unit.test_layer A

Complexity

Size/Duplication

Importance

16 Methods

2 Functions

Duplication Side-by-Side

Filter issues like