<filename>hackerrank/Algorithms/Birthday Chocolate/test.py
import unittest

import solution


class TestQ(unittest.TestCase):
    def test_case_0(self):
        self.assertEqual(solution.birthday([1, 2, 1, 3, 2], 3, 2), 2)

    def test_case_1(self):
        self.assertEqual(solution.birthday([1, 1, 1, 1, 1, 1], 3, 2), 0)

    def test_case_2(self):
        self.assertEqual(solution.birthday([4], 4, 1), 1)


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

<filename>hackerrank/Algorithms/Birthday Chocolate/test.py
import unittest

import solution


class TestQ(unittest.TestCase):
    def test_case_0(self):
        self.assertEqual(solution.birthday([1, 2, 1, 3, 2], 3, 2), 2)

    def test_case_1(self):
        self.assertEqual(solution.birthday([1, 1, 1, 1, 1, 1], 3, 2), 0)

    def test_case_2(self):
        self.assertEqual(solution.birthday([4], 4, 1), 1)


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