code: def test_format_array(self):
      self.assertEqual(
          utils.format_array(np.ones([], np.float32)), "f32[]"
      )
      self.assertEqual(
          utils.format_array(np.ones([1, 2], np.int8)), "s8[1,2]"
      )
      self.assertEqual(
          utils.format_array(np.ones([], jnp.bfloat16)), "bf16[]"
      )
  
imports: from absl.testing import absltest
  from absl.testing import parameterized
  from haiku._src import test_utils
  from haiku._src import utils
  import jax.numpy
  import numpy
label_desc: Use image node_id 5 for calling the UtilsTest obj's underlying member method code with example usage: obj.test_format_array() without return types