#Función que calcula el área de un triángulo
def calcula_area(base,altura):

    '''
    Comprobando el funcionamiento del script
    
    >>> calcula_area(10,5)
    25
    '''
    
    return (base*altura)/2

import doctest #importar módulo
doctest.testmod()

Para realizar las pruebas unitarias en Python se puede hacer uso de la librería doctest, la cual viene pre-instalada al instalar Python en nuestras computadoras.

La sintaxis para colocar pruebas unitarias usando doctest es la siguiente:

    '''
    Comprobando el funcionamiento del script
    
    >>> calcula_area(10,5)
    25
    '''

Se hace uso de >>> y el llamado al bloque de código a testear.

El resultado:

El resultado esperado es 25, aunque esta función retornó 25.0

Este es un error «bueno» que interpreta Python, no es en sí que no haya sido exitosa la prueba, sino que el formato de salida del número en la función difiere del esperado. Queda a criterio de quién hace la prueba unitaria el dar por buena o no esta validación.