code: def test_vocab_size(self, vocab_size):
      e = embed.Embed(vocab_size=vocab_size)
      self.assertEqual(e.vocab_size, vocab_size)
      self.assertEqual(e.embeddings.shape[0], vocab_size)
  
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 1 for calling the EmbedTest obj's underlying member method code with example usage: obj.test_vocab_size(vocab_size) without return types