gistfile1.pyw

def foo(val):
    """
    >>> foo(0)
    2
    >>> foo(1)
    Traceback (most recent call last):
        ...
    ValueError: 1
    """
    if val == 1:
        raise ValueError(val)
    return 2

import doctest
doctest.testmod()