Python TDD Prime Numbers