42 lines
958 B
Python
42 lines
958 B
Python
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
|