Add sum test.py