Código para calcular factores primos bajo filosofía de TDD. Creado por Alfredo Chavez (@alfredochv) y utilizado en el artículo "Test Driven Development: Guia para los no iniciados" publicado en SG #38.

Factores primos con TDD.py

from unittest import main, TestCase

class TestPrimeFactors(TestCase):
    def testPrimesOf0to1(self):
        self.assertEquals([], factorsOf(0))
        self.assertEquals([], factorsOf(1))
    def testPrimesOf2to4(self):
        self.assertEquals([2], factorsOf(2))
        self.assertEquals([3], factorsOf(3))
        self.assertEquals([2,2], factorsOf(4))

def factorsOf(n):
    result, factor = [], 2
    while n > 1:
        while n % factor == 0:
            result.append(factor)
            n /= factor
        factor += 1
    return result

if __name__ == '__main__':
    main()