I like using doctest in these simple cases - the tests go in the docstrings!

(Also, it’s part of the standard library.)

https://docs.python.org/3/library/doctest.html