how to write python code example

Example 1: python write

with open(filename,"w") as f:
  f.write('Hello World')

Example 2: 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