tests in python code example

Example: python testing

>>> assert sum([1, 2, 3]) == 6, "Should be 6"