code: def handle_fault(fault):
      if fault == FaultType.Exception:
          raise FaultInjectionError("Fault Injection")
      elif fault == FaultType.UnhandledException:
          raise FaultInjectionUnhandledError(
              "Fault Injection Unhandled"
          )
      elif fault == FaultType.ProcessExit:
          # used to simulate process crash, no cleanup.
          os._exit(-1)
      assert (
          fault == FaultType.NoFault
      ), f"Got unexpected fault: {fault}"
  
file_path: mars/mars/services/tests/fault_injection_manager.py
imports: import enum
  import os
  import uuid
  from abc import ABC, abstractmethod
  from ...core.base import MarsError
  from ..session import SessionAPI
label_desc: Use image node_id 1 for calling a global function with example usage: handle_fault(fault) without return types