code: def test_embed_invalid_lookup(self):
      lookup_style = "FOO"
      emb = embed.Embed(
          embedding_matrix=_EMBEDDING_MATRIX, lookup_style="FOO"
      )
      with self.assertRaisesRegex(AttributeError, lookup_style):
          emb(_1D_IDS)
  
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 6 for calling the EmbedTest obj's underlying member method code with example usage: obj.test_embed_invalid_lookup() without return types