import unittest

class Test01Bingo(unittest.TestCase):
    def test_bingo(self):
        listki = [[4, 1, 2, 3, 5], [6, 1, 2, 3, 4], [7, 6, 4, 3, 2]]
        self.assertEqual(bingo(listki, [1, 2, 3, 4, 5, 6, 7]), 5)
        self.assertEqual(bingo(listki, [7, 8, 9, 10, 6, 5, 4, 3, 2, 1, 0]), 9)
        self.assertEqual(bingo(listki[:-1], [7, 8, 9, 10, 6, 5, 4, 3, 2, 1, 0]), 10)



class Test02Nogavice(unittest.TestCase):
    def test_brez_parov(self):
        self.assertEqual(brez_parov([15, 12, 17, 18, 12, 4, 15, 4, 7], [3, 4, 3, 5, 17, 7]), (1, 2))
        self.assertEqual(brez_parov([3, 4, 3, 5, 17, 7], [15, 12, 17, 18, 12, 4, 15, 4, 7]), (2, 1))
        self.assertEqual(brez_parov([1, 2, 3], [3, 4, 5, 6, 7]), (2, 4))
        self.assertEqual(brez_parov([3, 4, 5, 6, 7], [1, 2, 3]), (4, 2))
        self.assertEqual(brez_parov([1, 2, 3], [3, 4, 4]), (2, 0))
        self.assertEqual(brez_parov([3, 4, 4], [1, 2, 3]), (0, 2))
        self.assertEqual(brez_parov([1, 2, 3, 3, 2, 1], [4, 4]), (0, 0))
        self.assertEqual(brez_parov([4, 4], [1, 2, 3, 3, 2, 1]), (0, 0))
        self.assertEqual(brez_parov([1, 2, 3, 2, 1], []), (1, 0))
        self.assertEqual(brez_parov([], [1, 2, 3, 2, 1]), (0, 1))


class Test03VSeznam(unittest.TestCase):
    def test_v_seznam(self):
        self.assertEqual(v_seznam((1, (2, (3, None)))), [1, 2, 3])
        self.assertEqual(v_seznam((1, None)), [1])
        self.assertEqual(v_seznam(None), [])

    def test_v_seznam_rek(self):
        self.assertEqual(v_seznam_rek((1, (2, (3, None)))), [1, 2, 3])
        self.assertEqual(v_seznam_rek((1, None)), [1])
        self.assertEqual(v_seznam_rek(None), [])


class Test04VPare(unittest.TestCase):
    def test_v_pare(self):
        self.assertEqual(v_pare([1, 2, 3]), (1, (2, (3, None))))
        self.assertEqual(v_pare([1]), (1, None))
        self.assertIsNone(v_pare([]))


class Test05Sledilnik(unittest.TestCase):
    def test_sledilnik(self):
        zacetek = {
            'Anja': 'Libanja', 'Ive': 'Županje Njive', 'Peter': 'Šempeter',
            'Ela': 'Koroška Bela', 'Nik': 'Kamnik', 'Jan': 'Ozeljan',
            'Jana': 'Ljubljana', 'Tina': 'Slatina', 'Lan': 'Jablan',
            'Iva': 'Lepa njiva', 'Tomaž': 'Stomaž', 'Ana': 'Mojstrana',
            'Bor': 'Maribor', 'Vid': 'Šentvid', 'Anže': 'Šentjanž',
            'Ajda': 'Odranci', 'Tim': 'Polževo', 'France': 'Lozice'}

        sledilnik = Sledilnik(zacetek)
        sledilnik2 = Sledilnik({"Ana": "Mala Polana", "Lan": "Begunje"})

    #    POBRIŠI SELITVE V ISTI KRAJ!

        self.assertEqual(sledilnik.selitev(), 0)
        self.assertEqual(sledilnik.kje_zivi("Jana"), "Ljubljana")
        self.assertEqual(sledilnik.prebivalci("Ljubljana"), {"Jana"})

        self.assertIsNone(sledilnik.preseli("Jana", "Šentvid"))
        self.assertEqual(sledilnik.selitev(), 1)
        self.assertEqual(sledilnik.kje_zivi("Jana"), "Šentvid")
        self.assertEqual(sledilnik.prebivalci("Ljubljana"), set())
        self.assertEqual(sledilnik.prebivalci("Šentvid"), {"Jana", "Vid"})

        self.assertIsNone(sledilnik.preseli("France", "Slatina"))
        self.assertEqual(sledilnik.selitev(), 2)
        self.assertEqual(sledilnik.kje_zivi("France"), "Slatina")
        self.assertEqual(sledilnik.prebivalci("Lozice"), set())
        self.assertEqual(sledilnik.prebivalci("Slatina"), {"France", "Tina"})

        self.assertIsNone(sledilnik.preseli_vse("Slatina", "Mojstrana"))
        self.assertEqual(sledilnik.selitev(), 4)
        self.assertEqual(sledilnik.kje_zivi("France"), "Mojstrana")
        self.assertEqual(sledilnik.kje_zivi("Tina"), "Mojstrana")
        self.assertEqual(sledilnik.prebivalci("Slatina"), set())
        self.assertEqual(sledilnik.prebivalci("Mojstrana"), {"Ana", "France", "Tina"})

        self.assertIsNone(sledilnik.preseli_vse("Mojstrana", "Šentvid"))
        self.assertEqual(sledilnik.selitev(), 7)
        self.assertEqual(sledilnik.kje_zivi("France"), "Šentvid")
        self.assertEqual(sledilnik.kje_zivi("Tina"), "Šentvid")
        self.assertEqual(sledilnik.kje_zivi("Ana"), "Šentvid")
        self.assertEqual(sledilnik.prebivalci("Mojstrana"), set())
        self.assertEqual(sledilnik.prebivalci("Šentvid"), {"Jana", "Vid", "Ana", "France", "Tina"})

        self.assertEqual(sledilnik2.kje_zivi("Ana"), "Mala Polana")
        self.assertEqual(sledilnik2.selitev(), 0)


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