code: def test_embedding(term_index):
      embed = mz.embedding.load_from_file(
          mz.datasets.embeddings.EMBED_RANK
      )
      matrix = embed.build_matrix(term_index)
      assert matrix.shape == (len(term_index), 50)
      embed = mz.embedding.load_from_file(
          mz.datasets.embeddings.EMBED_10_GLOVE, mode="glove"
      )
      matrix = embed.build_matrix(term_index)
      assert matrix.shape == (len(term_index), 10)
      assert embed.input_dim == 5
  
label_desc: Use image node_id 2 for calling a global function with example usage: test_embedding(term_index) without return types