fix and test find_maxima function
This commit is contained in:
parent
c5b36a07bf
commit
3a3a0438b4
|
@ -7,4 +7,14 @@ def find_maxima(x):
|
|||
Output:
|
||||
idx -- list of indices of the local maxima in x
|
||||
"""
|
||||
return []
|
||||
|
||||
local_maxima = []
|
||||
|
||||
for index, value in enumerate(x):
|
||||
if index <= len(x)-2:
|
||||
if x[index] > x[index+1] and x[index] > x[index-1]:
|
||||
local_maxima.append(index)
|
||||
|
||||
if len(local_maxima) > 0:
|
||||
return local_maxima
|
||||
|
||||
|
|
10
hands_on/local_maxima/test_local_maxima.py
Normal file
10
hands_on/local_maxima/test_local_maxima.py
Normal file
|
@ -0,0 +1,10 @@
|
|||
from local_maxima import find_maxima
|
||||
import pytest
|
||||
|
||||
def test_find_maxima():
|
||||
input_list = [0, 6, -2, 5, 1]
|
||||
expected = [1, 3]
|
||||
|
||||
result = find_maxima(input_list)
|
||||
|
||||
assert result == expected
|
Loading…
Reference in a new issue