Update test_sum.py