对Python assert函数的建议

Question

我在多个脚本中多次使用了assert这个功能，我在想有没有人能给我一些建议，告诉我有没有更好的方法来实现这个，而不是我下面创建的这些函数。

def assert_validation(expected, actual, type='', message=''):
    if type == '==':
        assert expected == actual, 'Expected: %s, Actual: %s, %s' %(expected, actual, message)
    elif type == '!=':
        assert expected != actual, 'Expected: %s, Actual: %s, %s' %(expected, actual, message)
    elif type == '<=':
        assert expected <= actual, 'Expected: %s, Actual: %s, %s' %(expected, actual, message)
    elif type == '>=':
        assert expected >= actual, 'Expected: %s, Actual: %s, %s' %(expected, actual, message)

def assert_str_validation(expected, actual, type='', message=''):
    if type == '==':
        assert str(expected) == str(actual), 'Expected: %s, Actual: %s, %s' %(expected, actual, message)
    elif type == '!=':
        assert str(expected) != str(actual), 'Expected: %s, Actual: %s, %s' %(expected, actual, message)
    elif type == '<=':
        assert str(expected) <= str(actual), 'Expected: %s, Actual: %s, %s' %(expected, actual, message)
    elif type == '>=':
        assert str(expected) >= str(actual), 'Expected: %s, Actual: %s, %s' %(expected, actual, message)

Answer 1

这段话的意思是，确实可以用更简短的方法来做这件事。你真的不能直接在脚本里用 assert expected == actual 这样的代码吗？

def assert_validation(expected, actual, type='', message='', trans=(lambda x: x)):
    m = { '==': (lambda e, a: e == a),
          '!=': (lambda e, a: e != a),
          '<=': (lambda e, a: e <= a),
          '>=': (lambda e, a: e >= a), }
    assert m[type](trans(expected), trans(actual)), 'Expected: %s, Actual: %s, %s' % (expected, actual, message)

def assert_str_validation(expected, actual, type='', message=''):
    assert_validation(expected, actual, type, message, trans=str)