code: def setUp(self):
      super(FactorizedVocabEmbeddingTest, self).setUp()
      self.graph = mtf.Graph()
      self.mesh = mtf.Mesh(self.graph, "mtf_mesh")
      self.variable_dtype = mtf.VariableDType(
          activation_dtype=tf.float32
      )
  
      self.addCleanup(mock.patch.stopall)
      self.initializer_mock = mock.MagicMock()
      random_normal_initializer_mock = mock.patch.object(
          tf, "random_normal_initializer"
      ).start()
      random_normal_initializer_mock.return_value = (
          self.initializer_mock
      )
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import mesh_tensorflow
  from mesh_tensorflow.transformer import vocab_embeddings
  import mock
  import numpy
  import scipy.special
  import tensorflow.compat.v1
label_desc: Use image node_id 1 for calling the FactorizedVocabEmbeddingTest obj's underlying member method code with example usage: obj.setUp() without return types