import unittest from dataset.problem_11.buggy import binary_search class TestBinarySearch(unittest.TestCase): def test_found_middle(self): self.assertEqual(binary_search([1, 3, 5, 7], 5), 2) def test_found_last(self): self.assertEqual(binary_search([1, 3, 5, 7], 7), 3) def test_not_found(self): self.assertEqual(binary_search([1, 3, 5, 7], 4), -1) if __name__ == "__main__": unittest.main()