dataset/problem_4/test.py

import unittest
from dataset.problem_4.buggy import merge_sorted


class TestMergeSorted(unittest.TestCase):
    def test_regular_merge(self):
        self.assertEqual(merge_sorted([1, 3, 5], [2, 4, 6]), [1, 2, 3, 4, 5, 6])

    def test_uneven_lengths(self):
        self.assertEqual(merge_sorted([1, 2], [3, 4, 5]), [1, 2, 3, 4, 5])


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