| import lit | |
| import lit.formats | |
| CUSTOM_PASS = lit.Test.ResultCode("CUSTOM_PASS", "My Passed", False) | |
| CUSTOM_FAILURE = lit.Test.ResultCode("CUSTOM_FAILURE", "My Failed", True) | |
| class MyFormat(lit.formats.ShTest): | |
| def execute(self, test, lit_config): | |
| result = super(MyFormat, self).execute(test, lit_config) | |
| if result.code.isFailure: | |
| result.code = CUSTOM_FAILURE | |
| else: | |
| result.code = CUSTOM_PASS | |
| return result |