forked from BCCN-Prog/2018_day2
-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_maxima.py
46 lines (35 loc) · 1.03 KB
/
test_maxima.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
from maxima import *
import numpy as np
def test_find_maxima_1():
x = [0, 1, 2, 1, 2, 1, 0]
assert find_maxima(x) == [2,4]
def test_find_maxima_2():
x = [-i**2 for i in range(-3, 4)]
assert find_maxima(x) == [3]
def test_find_maxima_3():
x = [np.sin(2*alpha) for alpha in np.linspace(0.0, 5.0, 100)]
assert find_maxima(x) == [16,78]
def test_find_maxima_4():
x = [4, 2, 1, 3, 1, 2]
assert find_maxima(x) == [0,3,5]
def test_find_maxima_5():
x = [4, 2, 1, 3, 1, 5]
assert find_maxima(x) == [0,3,5]
def test_find_maxima_6():
x = [4, 2, 1, 3, 1]
assert find_maxima(x) == [0,3]
def test_find_maxima_7():
x = [1, 2, 2, 1]
assert find_maxima(x) == [1,2]
def test_find_maxima_8():
x = [1, 2, 2, 3, 1]
assert find_maxima(x) == [3]
def test_find_maxima_9():
x = [1, 3, 2, 2, 1]
assert find_maxima(x) == [1]
def test_find_maxima_10():
x = [3, 2, 2, 3]
assert find_maxima(x) == [0,3]
def test_find_maxima_11():
x = [1, 2, 2, 2, 1]
assert find_maxima(x) == [1,2,3]