Ejemplo test unitario

def cuadrado(lado=1):
    return lado**2

def rectangulo(lado1,lado2):
    return lado1*lado2

El test:

from unittest import TestCase

import areas


class Test(TestCase):
    def test_cuadrado(self):
        resultado = areas.cuadrado(1)
        self.assertEqual(resultado, 1)
        resultado = areas.cuadrado(3)
        self.assertEqual(resultado, 9)
        resultado = areas.cuadrado(10)
        self.assertEqual(resultado, 100)
    def test_rectangulo(self):
        resultado=areas.rectangulo(2,3)
        self.assertEqual(resultado,6)
        resultado = areas.rectangulo(0, 3)
        self.assertEqual(resultado, 0)
        resultado = areas.rectangulo(20, 3)
        self.assertEqual(resultado, 60)