108 lines
2.8 KiB
Python
108 lines
2.8 KiB
Python
|
import numpy as np
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from local_maxima import find_maxima
|
||
|
|
|
||
|
|
|
||
|
|
def test_find_maxima():
|
||
|
|
values = [1, 3, -2, 0, 2, 1]
|
||
|
|
expected = [1, 4]
|
||
|
|
maxima = find_maxima(values)
|
||
|
|
assert maxima == expected
|
||
|
|
|
||
|
|
|
||
|
|
def test_find_maxima_edges():
|
||
|
|
values = [4, 2, 1, 0, 1, 5]
|
||
|
|
expected = [0, 5]
|
||
|
|
maxima = find_maxima(values)
|
||
|
|
assert maxima == expected
|
||
|
|
|
||
|
|
|
||
|
|
def test_find_maxima_empty():
|
||
|
|
values = []
|
||
|
|
expected = []
|
||
|
|
maxima = find_maxima(values)
|
||
|
|
assert maxima == expected
|
||
|
|
|
||
|
|
|
||
|
|
def test_find_maxima_plateau():
|
||
|
|
values = [1, 2, 2, 1]
|
||
|
|
expected = [1]
|
||
|
|
maxima = find_maxima(values)
|
||
|
|
assert maxima == expected
|
||
|
|
|
||
|
|
|
||
|
|
def test_find_maxima_not_a_plateau():
|
||
|
|
values = [1, 2, 2, 3, 1]
|
||
|
|
expected = [3]
|
||
|
|
maxima = find_maxima(values)
|
||
|
|
assert maxima == expected
|
||
|
|
|
||
|
|
|
||
|
|
# the tests below here fail, can you get them to pass?
|
||
|
|
|
||
|
|
|
||
|
|
def test_find_maxima_correct_order():
|
||
|
|
# TASK: get this test to pass
|
||
|
|
values = [2, 1, 5, 1, 9]
|
||
|
|
expected = [0, 2, 4]
|
||
|
|
maxima = find_maxima(values)
|
||
|
|
assert maxima == expected
|
||
|
|
|
||
|
|
|
||
|
|
def test_find_maxima_one_value():
|
||
|
|
# TASK: get this test to pass
|
||
|
|
values = [1]
|
||
|
|
expected = [0]
|
||
|
|
maxima = find_maxima(values)
|
||
|
|
assert maxima == expected
|
||
|
|
|
||
|
|
|
||
|
|
def test_find_maxima_long_plateau():
|
||
|
|
# TASK: Change the implementation for when there is a plateau
|
||
|
|
# for uneven plateau length, return the middle index, e.g. [1, 2, *2*, 2, 1] --> [2]
|
||
|
|
# for even plateau length, return the index left of the middle e.g. [1, 2, *2*, 2, 2, 1] --> [2]
|
||
|
|
values = [1, 2, 2, 2, 2, 2, 1, 8, 0]
|
||
|
|
expected = [3, 7]
|
||
|
|
maxima = find_maxima(values)
|
||
|
|
assert maxima == expected
|
||
|
|
|
||
|
|
|
||
|
|
def test_find_maxima_plateau_at_end():
|
||
|
|
# TASK: make sure plateaus at the end are handled properly (see test above)
|
||
|
|
values = [1, 2, 2]
|
||
|
|
expected = [1]
|
||
|
|
maxima = find_maxima(values)
|
||
|
|
assert maxima == expected
|
||
|
|
|
||
|
|
|
||
|
|
def test_find_maxima_plateau_at_start():
|
||
|
|
# TASK: make sure plateaus at the start are handled properly (see test above)
|
||
|
|
values = [1, 1, 0, 0]
|
||
|
|
expected = [0]
|
||
|
|
maxima = find_maxima(values)
|
||
|
|
assert maxima == expected
|
||
|
|
|
||
|
|
|
||
|
|
def test_find_maxima_all_same_values():
|
||
|
|
# TASK: implement a check for lists where there is no local maximum
|
||
|
|
values = [1, 1]
|
||
|
|
expected = [0]
|
||
|
|
maxima = find_maxima(values)
|
||
|
|
assert maxima == expected
|
||
|
|
|
||
|
|
|
||
|
|
def test_find_maxima_letters():
|
||
|
|
# stings can be evaluated with > and <, who knew!
|
||
|
|
# Find an easy solution so that both "t"s are recognised as local maxima
|
||
|
|
values = ["T", "e", "s", "t", "s", "!"]
|
||
|
|
expected = [0, 3]
|
||
|
|
maxima = find_maxima(values)
|
||
|
|
assert maxima == expected
|
||
|
|
|
||
|
|
|
||
|
|
def test_find_maxima_new_inputs_to_make_current_function_fail():
|
||
|
|
# should you actually be done with all tests, then you can think of other cases where the current function fails
|
||
|
|
# and write tests for them and fix them
|
||
|
|
assert False
|