Add unit tests for max_subarray