proper testing in python 3 code example

Example: python testing

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