code: def test_embed_dim(self, embed_dim):
      e = embed.Embed(vocab_size=100, embed_dim=embed_dim)
      self.assertEqual(e.embed_dim, embed_dim)
      self.assertEqual(e.embeddings.shape[1], embed_dim)
  
file_path: sonnet/sonnet/src/embed_test.py
imports: from absl.testing import parameterized
  from sonnet.src import embed
  from sonnet.src import initializers
  from sonnet.src import test_utils
  import tensorflow
label_desc: Use image node_id 2 for calling the EmbedTest obj's underlying member method code with example usage: obj.test_embed_dim(embed_dim) without return types