code: def test_mock_transformer_lm_eval_equals_predict(self):
      model_fn = functools.partial(
          test_utils.MockTransformerLM,
          sequence_fn=arithmetic_sequence,
          vocab_size=10,
      )
      test_utils.test_eval_equals_predict_discrete(
          model_fn, vocab_size=10
      )
  
file_path: trax/trax/layers/test_utils_test.py
imports: import functools
  from absl.testing import absltest
  import numpy
  from trax.layers import test_utils
  from trax.supervised import decoding
label_desc: Use image node_id 1 for calling the TestUtilsTest obj's underlying member method code with example usage: obj.test_mock_transformer_lm_eval_equals_predict() without return types