code: def testLinearOutputOneModule(self):
      batch_size = 4
      input_size = 5
      output_size = 3
      lin_a = linear.Linear(output_size)
      inputs = tf.random.uniform([batch_size, input_size])
      output = util.apply_linear(inputs, lin_a, activation=tf.nn.tanh)
  
      expected_output = np.tanh(
          np.matmul(inputs.numpy(), lin_a.w.numpy()) + lin_a.b.numpy()
      )
      self.assertAllClose(
          expected_output, output.numpy(), atol=self.get_atol()
      )
  
imports: from absl.testing import parameterized
  import numpy
  from sonnet.src import linear
  from sonnet.src import test_utils
  from sonnet.src.nets.dnc import util
  import tensorflow
  import tree
label_desc: Use image node_id 1 for calling the LinearTest obj's underlying member method code with example usage: obj.testLinearOutputOneModule() without return types