def square(x): """ Squares x . >>> square (2) 4 >>> square (-2) 4 >>> square (square(-3)) 81 """ return x*x if __name__ == '__main__': import doctest print "test" doctest.testmod()