code: def testReshape(self, preserve_dims, expected_output_shape):
      mod = reshape.Reshape(
          output_shape=(-1, D), preserve_dims=preserve_dims
      )
      outputs = mod(tf.ones([B, H, W, C, D]))
      self.assertEqual(outputs.shape, expected_output_shape)
  
imports: from absl.testing import parameterized
  import numpy
  from sonnet.src import reshape
  from sonnet.src import test_utils
  import tensorflow
label_desc: Use image node_id 1 for calling the ReshapeTest obj's underlying member method code with example usage: obj.testReshape(preserve_dims, expected_output_shape) without return types