code: def test_main():
      """Main function to be called within `__main__` of a test file."""
      multi_process_lib.test_main()
  
imports: import tempfile
  from tensorflow.core.protobuf import data_service_pb2
  from tensorflow.core.protobuf import service_config_pb2
  from tensorflow.python.data.experimental.kernel_tests.service import test_base
  from tensorflow.python.data.experimental.service import server_lib
  from tensorflow.python.distribute import multi_process_lib
  from tensorflow.python.framework import test_util
  from tensorflow.python.platform import googletest
label_desc: Use image node_id 1 for calling a global function with example usage: test_main() without return types