code: def test_inline_jit_add_module(self):
      mod = InlineJitAddModule()
      a = b = jnp.ones([])
      graph, args, c = dot.to_graph(mod)(a, b)
      self.assertEqual(args, (a, b))
      self.assertEqual(c, a + b)
      self.assertEmpty(graph.edges)
      (add_graph,) = graph.subgraphs
      self.assertEqual(add_graph.title, "inline_jit_add_module")
      self.assertEmpty(add_graph.subgraphs)
      add_edge_a, add_edge_b = add_graph.edges
      self.assertEqual(add_edge_a, (a, c))
      self.assertEqual(add_edge_b, (b, c))
      (add_node,) = add_graph.nodes
      self.assertEqual(add_node.title, "add")
      (add_out,) = add_node.outputs
      self.assertEqual(add_out, c)
  
imports: from absl.testing import absltest
  from absl.testing import parameterized
  from haiku._src import dot
  from haiku._src import module
  from haiku._src import test_utils
  import jax
  import jax.numpy
label_desc: Use image node_id 3 for calling the DotTest obj's underlying member method code with example usage: obj.test_inline_jit_add_module() without return types