code: def test_simple(self, use_bn):
      image = jnp.ones([2, 224, 224, 3])
      model = mobilenetv1.MobileNetV1(
          (1, 2, 2, 2, 2), (16, 32, 64, 128, 256), 100, use_bn=use_bn
      )
  
      logits = model(image, is_training=True)
      self.assertIsNotNone(logits)
      self.assertEqual(logits.shape, (2, 100))
  
file_path: dm-haiku/haiku/_src/nets/mobilenetv1_test.py
imports: from absl.testing import absltest
  from absl.testing import parameterized
  from haiku._src import test_utils
  from haiku._src.nets import mobilenetv1
  import jax.numpy
label_desc: Use image node_id 1 for calling the MobileNetV1Test obj's underlying member method code with example usage: obj.test_simple(use_bn) without return types