code: def shape_test(self, input_shape, expected_shape):
      x = tf.ones(input_shape)
  
      highres_layer = HighRes3DNet(num_classes=5)
      highres_layer_small = HighRes3DNetSmall(num_classes=5)
      highres_layer_large = HighRes3DNetLarge(num_classes=5)
  
      out = highres_layer(x, is_training=True)
      out_small = highres_layer_small(x, is_training=True)
      out_large = highres_layer_large(x, is_training=True)
  
      with self.cached_session() as sess:
          sess.run(tf.global_variables_initializer())
          out, out_large, out_small = sess.run(
              [out, out_large, out_small]
          )
          self.assertAllClose(expected_shape, out.shape)
          self.assertAllClose(expected_shape, out_large.shape)
          self.assertAllClose(expected_shape, out_small.shape)
  
file_path: NiftyNet/tests/highres3dnet_test.py
imports: from __future__ import absolute_import, print_function
  import os
  import unittest
  import tensorflow
  from tensorflow.contrib.layers.python.layers import regularizers
  from niftynet.network.highres3dnet import HighRes3DNet
  from niftynet.network.highres3dnet_large import HighRes3DNetLarge
  from niftynet.network.highres3dnet_small import HighRes3DNetSmall
  from tests.niftynet_testcase import NiftyNetTestCase
label_desc: Use image node_id 1 for calling the HighRes3DNetTest obj's underlying member method code with example usage: obj.shape_test(input_shape, expected_shape) without return types