import pytest

from src.modules.exercises.essentials import *

def test_subarray_sum():
	input = [2,3,-8,7,-1,2,3]
	result = maximum_subarray_sum(input)
	expected = [11, [7, -1, 2, 3]]
	assert result == expected
	input = [-2,-4]
	result = maximum_subarray_sum(input)
	expected = [-2, [-2]]
	assert result == expected
	input = [5,4,1,7,8]
	result = maximum_subarray_sum(input)
	expected = [25, [5,4,1,7,8]]
	assert result == expected

def test_find_missing():
	input = [8,2,4,5,3,7,1]
	result = find_missing(input)
	expected = 6
	assert result == expected
	input = [1,2,3,5]
	result = find_missing(input)
	expected = 4
	assert result == expected

def test_rain_watter():
	input = [3,0,1,0,4,0,2]
	expected = 10
	result = trap_rain_water(input)
	assert result == expected
	input = [3,0,2,0,4]
	expected = 7
	result = trap_rain_water(input)
	assert result == expected
	input = [1,2,3,4]
	expected = 0
	result = trap_rain_water(input)
	assert result == expected