Una función en python para mostrar cómo testear una función de forma sencilla

probar_funcion.py

def funcion_que_prueba_la_suma():
    suma = 1 + 2
    assert suma == 3, 'La suma no funciona correctamente, su resultado fue ' + str(suma) + ', distinto de 3'
    # si el assert no corta la ejecución, entonces funcionó bien
    print 'La suma funciona correctamente'