Update sum_tests.py