code: def test_default_creation(self, lookup_style):
      emb = embed.Embed(
          vocab_size=6, embed_dim=12, lookup_style=lookup_style
      )
      self.assertEqual(emb(_1D_IDS).shape, (2, 12))
  
file_path: dm-haiku/haiku/_src/embed_test.py
imports: import itertools
  from absl.testing import absltest
  from absl.testing import parameterized
  from haiku._src import embed
  from haiku._src import test_utils
  import jax.numpy
  import numpy
label_desc: Use image node_id 2 for calling the EmbedTest obj's underlying member method code with example usage: obj.test_default_creation(lookup_style) without return types