Suggestions pour la fonction assert python

Question

J'utilise affirmer plusieurs fois à travers plusieurs scripts, je me demandais si quelqu'un a des suggestions sur une meilleure façon d'y parvenir au lieu des fonctions que j'ai créées ci-dessous.

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

Eh bien, c'est certainement plus court ... vous pouvez vraiment pas simplement utiliser assert expected == actual ou quoi que dans les scripts eux-mêmes?

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)