def setUp(self):
self.ivo = Panda("Ivo", "*****@*****.**", "male")
self.network = PandaSocialNetwork()
class TestSocialNetwork(unittest.TestCase):
def setUp(self):
self.ivo = Panda("Ivo", "*****@*****.**", "male")
self.network = PandaSocialNetwork()
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_has_panda_when_panda_not_in_network(self):
rado = Panda("Rado", "*****@*****.**", "male")
self.assertFalse(self.network.has_panda(rado))
def test_make_and_are_friends(self):
rado = Panda("Rado", "*****@*****.**", "male")
self.network.make_friends(self.ivo, rado)
self.assertTrue(self.network.are_friends(self.ivo, rado))
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)
