code: def test_output_size_valid(self):
      mod = bias.Bias(output_size=(2 * 2,))
      mod(jnp.ones([2, 2 * 2]))
  
file_path: dm-haiku/haiku/_src/bias_test.py
imports: from absl.testing import absltest
  from haiku._src import bias
  from haiku._src import test_utils
  from haiku._src import transform
  import jax.numpy
  import numpy
label_desc: Use image node_id 2 for calling the BiasTest obj's underlying member method code with example usage: obj.test_output_size_valid() without return types