# Vecina testov uporablja graf, objavljen na
# http://ucilnica.pef.uni-lj.si/pluginfile.php/10364/mod_folder/content/0/naloga_09/mreza.png

# Ko test javi napako, poglejte, kaj pričakuje, kakšen je graf in kaj vrača
# vaša metoda.

import unittest

class TestPrijatelji(unittest.TestCase):
    def setUp(self):
        mreza = self.mreza = Prijatelji()
        mreza.dodaj("Ana", "Berta")
        mreza.dodaj("Ana", "Cilka")
        mreza.dodaj("Ana", "Dani")
        mreza.dodaj("Berta", "Cilka")
        mreza.dodaj("Dani", "Eva")
        mreza.dodaj("Eva", "Francka")
        mreza.dodaj("Eva", "Greta")
        mreza.dodaj("Eva", "Helga")
        mreza.dodaj("Francka", "Greta")
        mreza.dodaj("Greta", "Helga")

        self.dva = Prijatelji()
        self.dva.dodaj("Samo", "Edini")

        self.myspace = Prijatelji()


    def test_6_dodaj_sta(self):
        mreza = self.mreza

        self.assertTrue(mreza.sta_prijatelja("Ana", "Berta"))
        self.assertTrue(mreza.sta_prijatelja("Berta", "Ana"))

        self.assertTrue(mreza.sta_prijatelja("Eva", "Francka"))
        self.assertTrue(mreza.sta_prijatelja("Francka", "Eva"))

        self.assertTrue(mreza.sta_prijatelja("Greta", "Helga"))
        self.assertTrue(mreza.sta_prijatelja("Helga", "Greta"))

        self.assertFalse(mreza.sta_prijatelja("Dani", "Francka"))
        self.assertFalse(mreza.sta_prijatelja("Francka", "Dani"))

        self.assertTrue(self.dva.sta_prijatelja("Samo", "Edini"))
        self.assertTrue(self.dva.sta_prijatelja("Edini", "Samo"))


    def test_6_brisi(self):
        mreza = self.mreza
        mreza.pobrisi("Helga", "Greta")

        self.assertFalse(mreza.sta_prijatelja("Greta", "Helga"))
        self.assertFalse(mreza.sta_prijatelja("Helga", "Greta"))

        self.assertTrue(mreza.sta_prijatelja("Greta", "Eva"))
        self.assertTrue(mreza.sta_prijatelja("Eva", "Greta"))
        self.assertTrue(mreza.sta_prijatelja("Greta", "Francka"))
        self.assertTrue(mreza.sta_prijatelja("Francka", "Greta"))
        self.assertTrue(mreza.sta_prijatelja("Helga", "Eva"))
        self.assertTrue(mreza.sta_prijatelja("Eva", "Helga"))


    def test_7_prijatelji(self):
        mreza = self.mreza
        self.assertSetEqual(mreza.prijatelji("Ana"), {"Berta", "Cilka", "Dani"})
        self.assertSetEqual(mreza.prijatelji("Berta"), {"Ana", "Cilka"})
        self.assertSetEqual(mreza.prijatelji("Cilka"), {"Berta", "Ana"})
        self.assertSetEqual(mreza.prijatelji("Dani"), {"Ana", "Eva"})
        self.assertSetEqual(mreza.prijatelji("Eva"),
                            {"Dani", "Francka", "Greta", "Helga"})
        self.assertSetEqual(mreza.prijatelji("Francka"), {"Eva", "Greta"})
        self.assertSetEqual(mreza.prijatelji("Greta"),
                            {"Francka", "Eva", "Helga"})
        self.assertSetEqual(mreza.prijatelji("Helga"), {"Eva", "Greta"})

        self.assertSetEqual(self.dva.prijatelji("Samo"), {"Edini"})
        self.assertSetEqual(self.dva.prijatelji("Edini"), {"Samo"})


    def test_7_skupni_prijatelji(self):
        mreza = self.mreza
        self.assertSetEqual(mreza.skupni_prijatelji("Ana", "Berta"), {"Cilka"})
        self.assertSetEqual(mreza.skupni_prijatelji("Eva", "Greta"),
                            {"Francka", "Helga"})
        self.assertSetEqual(mreza.skupni_prijatelji("Dani", "Greta"), {"Eva"})
        self.assertSetEqual(mreza.skupni_prijatelji("Ana", "Helga"), set())
        self.assertSetEqual(mreza.skupni_prijatelji("Ana", "Dani"), set())

        self.assertSetEqual(self.dva.skupni_prijatelji("Samo", "Edini"), set())


    def test_7_skupni_prijatelj(self):
        mreza = self.mreza
        self.assertTrue(mreza.skupni_prijatelj("Ana", "Berta"))
        self.assertTrue(mreza.skupni_prijatelj("Eva", "Greta"))
        self.assertTrue(mreza.skupni_prijatelj("Dani", "Greta"))

        self.assertFalse(mreza.skupni_prijatelj("Ana", "Helga"))
        self.assertFalse(mreza.skupni_prijatelj("Ana", "Dani"))

        self.assertFalse(self.dva.skupni_prijatelj("Samo", "Edini"))


    def test_8_priporoci(self):
        mreza = self.mreza
        self.assertSetEqual(mreza.priporoci("Berta"), {"Dani"})
        self.assertSetEqual(mreza.priporoci("Dani"),
                            {"Berta", "Cilka", "Francka", "Greta", "Helga"})
        self.assertSetEqual(mreza.priporoci("Greta"), {"Dani"})

        self.assertSetEqual(self.dva.priporoci("Samo"), set())
        self.assertSetEqual(self.dva.priporoci("Edini"), set())


    def test_9_klika(self):
        mreza = self.mreza
        self.assertTrue(mreza.klika(["Ana", "Berta", "Cilka"]))
        self.assertTrue(mreza.klika(["Dani", "Eva"]))
        self.assertTrue(mreza.klika(["Eva", "Francka", "Greta", "Helga"]))

        self.assertFalse(mreza.klika(["Ana", "Dani", "Eva"]))

        self.assertTrue(self.dva.klika(["Samo", "Edini"]))

    def test_10_bolj_priljubljen(self):
        mreza = self.mreza
        self.assertTrue(mreza.bolj_priljubljen("Ana"))
        self.assertTrue(mreza.bolj_priljubljen("Eva"))

        self.assertFalse(mreza.bolj_priljubljen("Berta"))
        self.assertFalse(mreza.bolj_priljubljen("Greta"))

        self.assertTrue(self.dva.bolj_priljubljen("Samo"))
        self.assertTrue(self.dva.bolj_priljubljen("Edini"))

    def test_10_najbolj_priljubljeni(self):
        self.assertSetEqual(self.mreza.najbolj_priljubljeni(), {"Ana", "Eva"})
        self.assertSetEqual(self.dva.najbolj_priljubljeni(), {"Samo", "Edini"})
        self.assertSetEqual(self.myspace.najbolj_priljubljeni(), set())