code: def test_main(self):
      models_entry_point.main()
  
      self.load_params.assert_called_with(
          models_entry_point.PARAMS_FILE_NAME
      )
      self.run_experiment_kwargs.update(
          dict(distribution_strategy="one_device_strategy")
      )
      self.run_experiment.assert_called_with(
          **self.run_experiment_kwargs
      )
      self.model.save.assert_called_with(
          self.run_experiment_kwargs["model_dir"]
      )
  
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 3 for calling the ModelsTest obj's underlying member method code with example usage: obj.test_main() without return types