code: def test_embed_dtype_check(self):
      emb = embed.Embed(
          embedding_matrix=_EMBEDDING_MATRIX, lookup_style="ARRAY_INDEX"
      )
      with self.assertRaisesRegex(
          ValueError,
          "hk.Embed's __call__ method must take an array of integer dtype but "
          "was called with an array of float32",
      ):
          emb([1.0, 2.0])
  
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 5 for calling the EmbedTest obj's underlying member method code with example usage: obj.test_embed_dtype_check() without return types