dataset/problem_2/test.py

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()