code: def test_before_generating_trial_model(self):
      connection_config = metadata_store_pb2.ConnectionConfig()
      connection_config.sqlite.filename_uri = os.path.join(
          FLAGS.test_tmpdir, "1"
      )
      connection_config.sqlite.connection_mode = 3
      handler = ml_metadata_db.MLMetaData(
          None, None, None, connection_config=connection_config
      )
      handler.before_generating_trial_model(
          trial_id=1, model_dir="/tmp/1"
      )
      output = handler._store.get_executions_by_type("Trial")
      self.assertLen(output, 1)
      output = output[0]
      self.assertEqual(output.properties["id"].int_value, 1)
      self.assertEqual(
          output.properties["state"].string_value, "RUNNING"
      )
      self.assertEqual(
          output.properties["serialized_data"].string_value, ""
      )
      self.assertEqual(
          output.properties["model_dir"].string_value, "/tmp/1"
      )
      self.assertEqual(output.properties["evaluation"].string_value, "")
  
file_path: model_search/model_search/metadata/ml_metadata_db_test.py
imports: import os
  from absl import flags
  from absl.testing import absltest
  from model_search.metadata import ml_metadata_db
  from ml_metadata.proto import metadata_store_pb2
label_desc: Use image node_id 1 for calling the MlMetadataTest obj's underlying member method code with example usage: obj.test_before_generating_trial_model() without return types