code: def test_embed_property_check(self):
      lookup_style = "ONE_HOT"
      emb = embed.Embed(
          embedding_matrix=_EMBEDDING_MATRIX, lookup_style=lookup_style
      )
  
      self.assertEqual(emb.vocab_size, 3)
      self.assertEqual(emb.embed_dim, 4)
      np.testing.assert_allclose(
          emb.embeddings,
          jnp.asarray(
              [
                  [0.0, 0.0, 0.0, 0.0],
                  [0.5, 0.5, 0.5, 0.5],
                  [0.1, 0.2, 0.3, 0.4],
              ]
          ),
      )
  
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 7 for calling the EmbedTest obj's underlying member method code with example usage: obj.test_embed_property_check() without return types