I believe that doctest is the best of both worlds, https://docs.python.org/3/library/doctest.html