code: def test_output_column_structured(self):
      f = lambda: IdentityModule()(  # pylint: disable=g-long-lambda
          {
              "a": jnp.ones([32, 32]),
              "b": [jnp.zeros([1]), jnp.zeros([], jnp.int16)],
          }
      )
      rows = tabulate_to_list(f, columns=("output",))
      expected = [["{'a': f32[32,32], 'b': [f32[1], s16[]]}"]]
      self.assertEqual(rows, expected)
  
imports: from collections.abc import Sequence
  import typing
  from absl.testing import absltest
  from absl.testing import parameterized
  from haiku._src import base
  from haiku._src import basic
  from haiku._src import module
  from haiku._src import summarise
  from haiku._src import test_utils
  from haiku._src import transform
  import jax
  import jax.numpy
label_desc: Use image node_id 7 for calling the TabulateTest obj's underlying member method code with example usage: obj.test_output_column_structured() without return types