def test_equality(self):
case_res = [([1, 2, 2, 1, 1], [2]),
([1, 2, 2, 3, 1], [3]),
([1, 3, 2, 2, 1], [1]),
([3, 2, 2, 2, 3], [0, 4])]
for x, desired in case_res:
idx = find_maxima(x)
self.assertEqual(idx, desired)
def test_last_element(self):
x = [1, 2]
idx = find_maxima(x)
self.assertEqual(idx, [1])
def test_first_element(self):
x = [3, 1, 4]
idx = find_maxima(x)
self.assertEqual(idx, [0, 2])
