correct way to write doctest python code example

Example: how to write a python doctest

def a_function(*args):
  '''
  >>>an input
  expected output
  '''
  if this:
    return that
# A doctest may also include a docstring