code: def setUp(self):
      super(ModelsTest, self).setUp()
      config = mock.MagicMock(spec=config_definitions.ExperimentConfig)
      task = mock.MagicMock(spec=base_task.Task)
      self.run_experiment_kwargs = dict(
          task=task,
          mode="train_and_eval",
          params=config,
          model_dir="model_path",
          distribution_strategy="one_device",
      )
      self.load_params = mock.patch.object(
          models_entry_point,
          "load_params",
          autospec=True,
          return_value=self.run_experiment_kwargs,
      ).start()
  
      self.strategy = mock.patch.object(
          tf.distribute,
          "OneDeviceStrategy",
          autospec=True,
          return_value="one_device_strategy",
      ).start()
  
      self.model = mock.MagicMock()
      self.run_experiment = mock.patch.object(
          train_lib,
          "run_experiment",
          autospec=True,
          return_value=(self.model, {}),
      ).start()
  
file_path: cloud/src/python/tensorflow_cloud/core/experimental/tests/unit/models_entry_point_test.py
imports: from absl.testing import absltest
  import mock
  import tensorflow
  from tensorflow_cloud.core.experimental import models_entry_point
  from official.core import base_task
  from official.core import config_definitions
  from official.core import train_lib
label_desc: Use image node_id 1 for calling the ModelsTest obj's underlying member method code with example usage: obj.setUp() without return types