llvm/utils/lit/tests/Inputs/custom-result-category/format.py

   1 import lit
   2 import lit.formats
   3
   4 CUSTOM_PASS = lit.Test.ResultCode("CUSTOM_PASS", "My Passed", False)
   5 CUSTOM_FAILURE = lit.Test.ResultCode("CUSTOM_FAILURE", "My Failed", True)
   6
   7
   8 class MyFormat(lit.formats.ShTest):
   9     def execute(self, test, lit_config):
  10         result = super(MyFormat, self).execute(test, lit_config)
  11         if result.code.isFailure:
  12             result.code = CUSTOM_FAILURE
  13         else:
  14             result.code = CUSTOM_PASS
  15         return result