xref: /llvm-project/llvm/utils/lit/tests/Inputs/custom-result-category/format.py (revision b71edfaa4ec3c998aadb35255ce2f60bba2940b0) 
1import lit
2import lit.formats
3
4CUSTOM_PASS = lit.Test.ResultCode("CUSTOM_PASS", "My Passed", False)
5CUSTOM_FAILURE = lit.Test.ResultCode("CUSTOM_FAILURE", "My Failed", True)
6
7
8class 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
16