Doctests de Python: prueba de Ninguno

Question

Usando Python 2.7, estoy tratando de probar que el resultado de una llamada a una función en particular es Ninguna

Espero que estas pruebas pasen (disculpe el ejemplo bastante tonto)

def six_or_none(val):
    """
    >>> six_or_none(6)
    6
    >>> six_or_none(4)
    None
    """
    if val == 6:
        return 6
    return None

Sin embargo dan el siguiente resultado.

Failed example:
    six_or_none(4)
Expected:
    None
Got nothing

¿Cuál es la forma correcta de probar Ninguna en doctests?

Answer 1

El intérprete de Python ignora None devuelve valores, por lo que doctests hace lo mismo.

Prueba para is None lugar:

>>> six_or_none(4) is None
True

Answer 2

Otra opción sería una verificación directa de None :

def six_or_none(val):
    """
    >>> six_or_none(6)
    6
    >>> six_or_none(4)
    """
    if val == 6:
        return 6
    return None