code: def test_output_size_correct(self):
      layer = parallel_linear.ParallelLinears(3)
  
      outputs = layer(tf.ones([4, 2, 6]))
      self.assertEqual(outputs.shape, [4, 2, 3])
  
imports: from sonnet.src import linear
  from sonnet.src import parallel_linear
  from sonnet.src import test_utils
  import tensorflow
label_desc: Use image node_id 1 for calling the ParallelLinearTest obj's underlying member method code with example usage: obj.test_output_size_correct() without return types