code: def test_translation_matrix(self):
      model = translation_matrix.TranslationMatrix(
          self.source_word_vec, self.target_word_vec, self.word_pairs
      )
      model.train(self.word_pairs)
      self.assertEqual(model.translation_matrix.shape, (300, 300))
  
imports: from collections import namedtuple
  import unittest
  import logging
  import numpy
  import pytest
  from scipy.spatial.distance import cosine
  from gensim.models.doc2vec import Doc2Vec
  from gensim import utils
  from gensim.models import translation_matrix
  from gensim.models import KeyedVectors
  from gensim.test.utils import datapath, get_tmpfile
label_desc: Use image node_id 2 for calling the TestTranslationMatrix obj's underlying member method code with example usage: obj.test_translation_matrix() without return types