code: def test_simple(self, resnet_v2, bottleneck):
      image = jnp.ones([2, 64, 64, 3])
      model = resnet.ResNet(
          [1, 1, 1, 1], 10, resnet_v2=resnet_v2, bottleneck=bottleneck
      )
  
      for is_training in (True, False):
          logits = model(image, is_training=is_training)
          self.assertEqual(logits.shape, (2, 10))
  
file_path: dm-haiku/haiku/_src/nets/resnet_test.py
imports: from absl.testing import absltest
  from absl.testing import parameterized
  import haiku
  from haiku._src import test_utils
  from haiku._src import transform
  from haiku._src.nets import resnet
  import jax
  import jax.numpy
  import numpy
label_desc: Use image node_id 1 for calling the ResnetTest obj's underlying member method code with example usage: obj.test_simple(resnet_v2, bottleneck) without return types