class TestSovialNetwork(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 setUp(self): self.sn = PandaSocialNetwork() self.pan1 = Panda("Peter", "*****@*****.**", "male") self.pan2 = Panda("Sanja", "*****@*****.**", "female") self.pan3 = Panda("Rada", "*****@*****.**", "female") self.pan4 = Panda("Pafel", "*****@*****.**", "male") self.sn.add_panda(self.pan1)
def main(): network = PandaSocialNetwork() ivo = Panda("Ivo", "*****@*****.**", "male") rado = Panda("Rado", "*****@*****.**", "male") tony = Panda("Tony", "*****@*****.**", "female") for panda in [ivo, rado, tony]: network.add_panda(panda) network.make_friends(ivo, rado) network.make_friends(rado, tony) print(network.connection_level(ivo, rado) == 1) # True print(network.connection_level(ivo, tony) == 2) # True print(network.how_many_gender_in_network(1, rado, "female") == 1) # True
def setUp(self): self.ivo = Panda("Ivo", "*****@*****.**", "male") self.network = PandaSocialNetwork()
class TestSocialNetwork(unittest.TestCase): def setUp(self): self.sn = PandaSocialNetwork() self.pan1 = Panda("Peter", "*****@*****.**", "male") self.pan2 = Panda("Sanja", "*****@*****.**", "female") self.pan3 = Panda("Rada", "*****@*****.**", "female") self.pan4 = Panda("Pafel", "*****@*****.**", "male") self.sn.add_panda(self.pan1) def test_has_panda(self): self.assertFalse(self.sn.has_panda(self.pan2)) self.assertTrue(self.sn.has_panda(self.pan1)) def test_connection_level(self): self.sn.make_friends(self.pan1, self.pan2) self.sn.make_friends(self.pan1, self.pan3) self.sn.make_friends(self.pan4, self.pan2) self.sn.make_friends(self.pan2, self.pan3) self.assertEqual(self.sn.connection_level(self.pan1, self.pan4), 2) def test_add_panda(self): self.sn.add_panda(self.pan2) self.assertTrue(self.sn.has_panda(self.pan2)) self.assertFalse(self.sn.has_panda(self.pan3)) def test_print_network(self): self.assertEqual(self.sn.print_network(), {"Peter":[]}) def test_are_make_friends(self): self.sn.make_friends(self.pan1, self.pan3) self.assertTrue(self.sn.are_friends(self.pan1, self.pan3)) def test_are_connected(self): self.sn.make_friends(self.pan2, self.pan3) self.assertFalse(self.sn.are_connected(self.pan1, self.pan3)) self.assertTrue(self.sn.are_connected(self.pan2, self.pan3)) def test_friends_of(self): self.sn.make_friends(self.pan1, self.pan4) self.assertEqual(self.sn.friends_of(self.pan1), ["Pafel"]) def test_how_many_gender_in_network(self): self.sn.make_friends(self.pan1, self.pan4) self.sn.make_friends(self.pan1, self.pan2) self.sn.make_friends(self.pan2, self.pan3) self.assertEqual(self.sn.how_many_gender_in_network(2, self.pan2,"male"), 1) self.assertEqual(self.sn.how_many_gender_in_network(1, self.pan3,"male"), 0)
from panda import Panda from social_network import PandaSocialNetwork network = PandaSocialNetwork() ivo = Panda("Ivo", "*****@*****.**", "male") rado = Panda("Rado", "*****@*****.**", "male") tony = Panda("Tony", "*****@*****.**", "female") ony = Panda("ony", "*****@*****.**", "female") ado = Panda("ado", "*****@*****.**", "female") network.add_panda(rado) network.add_panda(ivo) network.add_panda(tony)