python doctest sample

gistfile1.py

def square(value):
  """
  Multiple value 2 times.
  
  >>> square(2)
  4
  >>> square(3)
  9
  """
  return value * value
  
if __name__ == "__main__":
  import doctest
  doctest.testmod()