Method test_shape

tests/networks/nets/test_restormer.py:114–118

(self, spatial_dims, in_channels, embed_dim, input_shape, expected_shape)

    @parameterized.expand(TEST_CASES_PATCHEMBED)
    def test_shape(self, spatial_dims, in_channels, embed_dim, input_shape, expected_shape):
        net = OverlapPatchEmbed(spatial_dims=spatial_dims, in_channels=in_channels, embed_dim=embed_dim)
        with eval_mode(net):
            result = net(torch.randn(input_shape))
            self.assertEqual(result.shape, expected_shape)


class TestRestormer(unittest.TestCase):