def setUp(self):
     self.ivo = Panda("Ivo", "fwadwa", "male")
     self.rado = Panda("Rado", "retre", "male")
     self.canko = Panda("Canko", "retre", "male")
     self.network = SocialNetwork()
 def setUp(self):
     self.ivo = Panda("Ivo", "fwadwa", "male")
     self.rado = Panda("Rado", "retre", "male")
     self.canko = Panda("Canko", "retre", "male")
     self.network = SocialNetwork()
class TestSocialNetwork(unittest.TestCase):

    def setUp(self):
        self.ivo = Panda("Ivo", "fwadwa", "male")
        self.rado = Panda("Rado", "retre", "male")
        self.canko = Panda("Canko", "retre", "male")
        self.network = SocialNetwork()

    def test_has_and_add_panda_in_network(self):
        self.network.add_panda(self.ivo)
        self.assertTrue(self.network.has_panda(self.ivo))

    def test_are_friedns_and_make_friends(self):
        self.network.make_friends(self.ivo, self.rado)
        self.assertTrue(self.network.are_friends(self.ivo, self.rado))

    def test_friends_of(self):
        self.network.make_friends(self.ivo, self.rado)
        self.network.make_friends(self.ivo, self.canko)
        my_list = [self.rado, self.canko]
        #self.assertTrue(self.rado in self.network.friends_of(self.ivo))

    def test_connection_level(self):
        rado = Panda("Rado", "*****@*****.**", "male")
        pavli = Panda("Pavli", "*****@*****.**", "male")
        maria = Panda("maria", "*****@*****.**", "female")

        self.network.make_friends(self.ivo, rado)
        self.network.make_friends(rado, pavli)
        self.network.make_friends(pavli, maria)

        self.assertEqual(self.network.connection_level(self.ivo, rado), 1)
        self.assertEqual(self.network.connection_level(self.ivo, pavli), 2)
        self.assertEqual(self.network.connection_level(self.ivo, maria), 3)
class TestSocialNetwork(unittest.TestCase):
    def setUp(self):
        self.ivo = Panda("Ivo", "fwadwa", "male")
        self.rado = Panda("Rado", "retre", "male")
        self.canko = Panda("Canko", "retre", "male")
        self.network = SocialNetwork()

    def test_has_and_add_panda_in_network(self):
        self.network.add_panda(self.ivo)
        self.assertTrue(self.network.has_panda(self.ivo))

    def test_are_friedns_and_make_friends(self):
        self.network.make_friends(self.ivo, self.rado)
        self.assertTrue(self.network.are_friends(self.ivo, self.rado))

    def test_friends_of(self):
        self.network.make_friends(self.ivo, self.rado)
        self.network.make_friends(self.ivo, self.canko)
        my_list = [self.rado, self.canko]
        #self.assertTrue(self.rado in self.network.friends_of(self.ivo))

    def test_connection_level(self):
        rado = Panda("Rado", "*****@*****.**", "male")
        pavli = Panda("Pavli", "*****@*****.**", "male")
        maria = Panda("maria", "*****@*****.**", "female")

        self.network.make_friends(self.ivo, rado)
        self.network.make_friends(rado, pavli)
        self.network.make_friends(pavli, maria)

        self.assertEqual(self.network.connection_level(self.ivo, rado), 1)
        self.assertEqual(self.network.connection_level(self.ivo, pavli), 2)
        self.assertEqual(self.network.connection_level(self.ivo, maria), 3)