import doctest def add(a, b): """ >>> add(2, 3) 5 """ return a + b r = doctest.testmod() print(r.failed == 0)