import unittest

class TestKlici(unittest.TestCase):
    def test_klici(self):
        self.assertEqual(klici([]), set())
        self.assertEqual(klici([508732, 556442, 334878, 951147, 508723, 508723]), {508732, 556442, 334878, 951147, 508723, 508723})
        self.assertEqual(klici([12345, 12345, 23456]), {12345, 23456})

class TestVRazredu(unittest.TestCase):
    def test_v_razredu(self):
        self.assertEqual(v_razredu(["Ana", "Berta"], []), {"Ana", "Berta"})
        self.assertEqual(v_razredu(["Ana", "Hana", "Barbara", "Manja", "Miha", "Peter"], ["Hana", "Manja", "Barbara", "Miha"]), {"Ana", "Peter"})

class TestPokaziCrke(unittest.TestCase):
    def test_pokazi_crke(self):
        self.assertEqual(pokazi_crke("PONUDNIK", set(["O", "N"])), '.ON..N..')
        self.assertEqual(pokazi_crke("PONUDNIK", set(["O", "I", "K"])), '.O....IK')
        self.assertEqual(pokazi_crke("PONUDNIK", set()), '........')
        self.assertEqual(pokazi_crke("PONUDNIK", set(["P", "O", "N", "I", "K", "U"])), 'PONU.NIK')

class TestIzvoz(unittest.TestCase):
    def test_drzave(self):
        self.assertSetEqual(set(kaj_izvaza("Slovenia")),
            {'transport equipment', 'manufactured goods', 'chemicals', 'food', 'machinery'})
        self.assertSetEqual(set(kaj_izvaza("Bermuda")),
            {"pharmaceuticals"})
        self.assertSetEqual(set(kaj_izvaza("Gambia, The")),
            {'palm kernels', 'peanut products', 'cotton lint', 'fish'})

    def test_produkti(self):
        self.assertSetEqual(set(kdo_izvaza("food")),
            {'Eritrea', 'Italy', 'Slovenia', 'Egypt', 'Guam', 'Saint Kitts and Nevis', 'Macedonia', 'United Kingdom', 'Lesotho', 'Antigua and Barbuda', 'Greece', 'Iraq', 'Poland'})
        self.assertSetEqual(set(kdo_izvaza("sand")),
            {'British Virgin Islands'})


class PodobnaBeseda(unittest.TestCase):
    def test_podobna(self):
        self.assertEqual(podobna("merjasec", ["ana", "berta", "cilka", "dani", "ema", "franci", "greta", "hilda"]), 'berta')
        self.assertEqual(podobna("zmaj", ["ana", "berta", "cilka", "dani", "ema", "franci", "greta", "hilda"]), 'ema')
        self.assertEqual(podobna("Krava", ["ana", "berta", "cilka", "dani", "ema", "franci", "greta", "hilda"]), 'berta')
        self.assertEqual(podobna("princesa", ["ana", "berta", "cilka", "dani", "ema", "franci", "greta", "hilda"]), 'franci')
        self.assertEqual(podobna("aaaaaaaa", ["ana", "berta", "cilka", "dani", "ema", "franci", "greta", "hilda"]), 'ana')
        self.assertEqual(podobna("aeiou", ["ana", "berta", "cilka", "dani", "ema", "franci", "greta", "hilda"]), 'berta')

class Raznolikost(unittest.TestCase):
    def test_raznolikost(self):
        self.assertEqual(najraznolika(["RABarbara", "izpit", "zmagA"]), 'izpit')
        self.assertEqual(najraznolika(["voda", "RABarbara", "izpit", "zmagA"]), 'voda')
        self.assertEqual(najraznolika(["voda", "RABarbara", "izpit", "zmagA", "veselje"]), 'veselje')
        self.assertEqual(najraznolika(["strawberry", "raspberry", "pi", "zmagA", "veselje"]), 'strawberry')

class Sifra(unittest.TestCase):
    def test_sifra(self):
        self.assertEqual(sifra('\x19\x14\x1c]\x19\x0f\x14\t\x13\x18\t]\x12\x0e[\n\x1a\t\x18\x15\x12\x13\x1c'),
            'big brother is watching')
        self.assertEqual(sifra('\xe1d\xe0q\xe5r\xf7b\xe0i'), 'strawberry')

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