def test_add_panda(self):
panda = Panda("Ivo", "*****@*****.**", "male")
soc_net = PandaSocialNetwork()
soc_net.add_panda(panda)
lst = soc_net.get_pandas_list()
panda1 = lst[0]
self.assertEqual(str(panda1), 'Ivo')
try:
soc_net.add_panda(panda)
except ValueError:
bra = 'this is an error'
self.assertEqual(bra, 'this is an error')
from PandaSocialNetwork import Panda, PandaSocialNetwork
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.are_friends(ivo, tony))
print(network.are_friends(ivo, rado))
print(network.friends_of(rado))
print(network.friends_of(ivo))
print(network.has_panda(ivo))
