Update test_max_subarray.py