test-leet-code.py

def testcase(*testcases):
    def build_format():
        max_param_len = 0
        max_result_len = 0
        num_params = 0
        for case in testcases:
            num_params = len(case) - 1
            for param in case[1:]:
                str_param = str(param)
                max_param_len = max(max_param_len, len(str_param))
            max_result_len = max(max_result_len, len(str(case[0])))
        
        param_format = ("%" + "%ds" % (max_param_len + 2)) * num_params
        result_format = "%s  -->  %-" + "%ds" % (max_result_len + 2) + "  |  %s"
        return param_format, result_format
    
    def deco(fun):
        def wrapper():
            param_format, result_format = build_format()
            
            for case in testcases:
                expected = case[0]
                params = case[1:]
                result = fun(*params)
                
                param_text = param_format % params
                
                test_result = 'UNCHECKED' if expected is None else 'PASSED' if result == expected else 'FAILED'
                
                print result_format % (param_text, result, test_result)
        
        return wrapper
    
    return deco