code: def test_one_row_per_method_call(self, num_calls):
      def f():
          m = IdentityModule()
          for _ in range(num_calls):
              m(x)
  
      x = jnp.ones([])
      invocations = get_summary(f)
      self.assertLen(invocations, num_calls)
      for invocation in invocations[1:]:
          self.assertEqual(
              invocations[0].context.method_name,
              invocation.context.method_name,
          )
  
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 3 for calling the SummariseTest obj's underlying member method code with example usage: obj.test_one_row_per_method_call(num_calls) without return types