Mercurial > hg > config
view python/example/binarysearch.py @ 701:de7bf9523e21
-
author | Jeff Hammel <k0scist@gmail.com> |
---|---|
date | Wed, 13 Aug 2014 13:16:56 -0700 |
parents | e3341b7ce4ef |
children |
line wrap: on
line source
#!/usr/bin/env python def find_index(ordered_array, to_find): """ Return index of ``to_find`` via binary search. Returns None if not in ``ordered_array`` ordered_array -- array of ordered values to """ if not ordered_array: return minimum = 0 maximum = len(ordered_array)-1 while True: middle = (minimum + maximum)/2 value = ordered_array[middle] if value == to_find: return middle if maximum == minimum: return if value < to_find: minimum = middle+1 continue if value > to_find: maximum = middle continue if __name__ == '__main__': import unittest class TestFindIndex(unittest.TestCase): values = [1,2,4,8,16,17] def test_spotcheck(self): for index, value in enumerate(self.values): self.assertEqual(find_index(self.values, value), index) unittest.main()