code: def test_parse_shape_str(self):
      self.assertParsedShape("f32[]", [], jnp.float32)
      self.assertParsedShape("f32[1,2,3]", [1, 2, 3], jnp.float32)
      self.assertParsedShape("pred[1]", [1], jnp.bool_)
      self.assertParsedShape("s8[1]", [1], jnp.int8)
      self.assertParsedShape("s16[1]", [1], jnp.int16)
      self.assertParsedShape("s32[1]", [1], jnp.int32)
      self.assertParsedShape("s64[1]", [1], jnp.int64)
      self.assertParsedShape("u8[1]", [1], jnp.uint8)
      self.assertParsedShape("u16[1]", [1], jnp.uint16)
      self.assertParsedShape("u32[1]", [1], jnp.uint32)
      self.assertParsedShape("u64[1]", [1], jnp.uint64)
      self.assertParsedShape("f16[1]", [1], jnp.float16)
      self.assertParsedShape("f32[1]", [1], jnp.float32)
      self.assertParsedShape("f64[1]", [1], jnp.float64)
      self.assertParsedShape("bf16[1]", [1], jnp.bfloat16)
      self.assertParsedShape("c64[1]", [1], jnp.complex64)
      self.assertParsedShape("c128[1]", [1], jnp.complex128)
  
file_path: jax/tests/jax_to_ir_test.py
imports: import unittest
  from absl.testing import absltest
  import jax.numpy
  from jax.tools import jax_to_ir
  from jax._src import test_util
label_desc: Use image node_id 6 for calling the JaxToIRTest obj's underlying member method code with example usage: obj.test_parse_shape_str() without return types