def METHOD_NAME(self, input_param, input_shape, expected_shape):
    net = SegResNet(**input_param).to(device)
    with eval_mode(net):
        result = net(torch.randn(input_shape).to(device))
        self.assertEqual(result.shape, expected_shape)