doctest_ex1.py

#! /usr/bin/env python
"""
"""
def square(x):
    """
    Squares x.
    > square(2)
    4
    > square(-2)
    4
    """

    return x * x

if __name__ == '__main__':
    import doctest
    doctest.testmod()