import unittest
import random

class Test01Obvezna(unittest.TestCase):
    def test_najvec_n_mali(self):
        self.assertEqual(najvec_n([1, 2, 3, 1, 1, 2, 1, 1, 2], 3), [1, 2, 3, 1, 1, 2, 2])
        self.assertEqual(najvec_n([1, 2, 3, 1, 1, 2, 1, 1, 2], 2), [1, 2, 3, 1, 2])
        self.assertEqual(najvec_n([1, 2, 3, 1, 1, 2, 1, 1, 2], 1), [1, 2, 3])

    def test_najvec_n_veliki(self):
        self.assertEqual(najvec_n(list(range(10000)) * 10, 2), list(range(10000)) * 2)
        x = [random.randint(0, 10000) for _ in range(100000)]
        najvec_n(x, 5)

class Test02Dodatna(unittest.TestCase):
    def test_najvec_n_na_mestu_mali(self):
        s = [1, 2, 3, 1, 1, 2, 1, 1, 2]
        self.assertIsNone(najvec_n_na_mestu(s, 3))
        self.assertEqual(s, [1, 2, 3, 1, 1, 2, 2])

        s = [1, 2, 3, 1, 1, 2, 1, 1, 2]
        najvec_n_na_mestu(s, 2)
        self.assertEqual(s, [1, 2, 3, 1, 2])

        s = [1, 2, 3, 1, 1, 2, 1, 1, 2]
        najvec_n_na_mestu(s, 1)
        self.assertEqual(s, [1, 2, 3])

    def test_najvec_n_na_mestu_veliki(self):
        s = najvec_n(list(range(10000)) * 10, 2)
        najvec_n_na_mestu(s, 3)
        self.assertEqual(s, list(range(10000)) * 2)
        x = [random.randint(0, 10000) for _ in range(100000)]
        najvec_n_na_mestu(x, 5)

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