doctest_simple.py

def my_function(a , b):
    """
    >>> my_function (2, 3)
    6
    >>> my_function('a', 3)
    'aaa'
    """
    return a * b