local_min.py

local_min.py

import unittest

def find_local_min(A):
    if len(A) == 0:
        return None
    elif len(A) == 1:
        return A[0]
    else:
        if A[0] < A[1]:
            return A[0]
        for i in xrange(1, len(A)):
            if A[i-1] > A[i]:
                if i+1 < len(A):
                    if A[i+1] > A[i]:
                        return A[i]
                else:
                    return A[i]

class TestMin(unittest.TestCase):
    def test_33(self):
        self.assertEqual(find_local_min([3,3]),None)
    def test_123(self):
        self.assertEqual(find_local_min([1,2,3]), 1)
    def test_323(self):
        self.assertEqual(find_local_min([3,2,3]), 2)
    def test_334561(self):
        self.assertEqual(find_local_min([3,3,4,5,6,1]), 1)

if __name__ == "__main__":
    unittest.main()