def test_add_friend(self, friendToAdd):
"""
Tests that the add_friend() function adds the specified person to the set
Takes a Person, friendToAdd and returns True if the specified friend was added to the set or False if it wasn't
"""
Network.add_friend(self, friendToAdd)
#check that the friend was added to the set
if (Network.setOfFriends.__contains__(friendToAdd)):
return True
else:
return False
def test_add_friend(self, friendToAdd):
"""
Tests that the add_friend() function adds the specified person to the set
Takes a Person, friendToAdd and returns True if the specified friend was added to the set or False if it wasn't
"""
Network.add_friend(self, friendToAdd)
#check that the friend was added to the set
if (Network.setOfFriends.__contains__(friendToAdd)):
return True
else:
return False
def test_add_friend(self):
"""
Tests that the add_friend() function adds the specified person to the set
Takes a Person, friendToAdd and returns True if the specified friend was added to the set or False if it wasn't
"""
#creates a friend Person object and sets their name and age
friendV = Person()
friendV.set_name("Vanessa")
friendV.set_age(18)
Network.add_friend(self, friendV)
#check that the friend was added to the set
self.assertTrue(friendV in Network.setOfFriends)
def test_add_friend(self):
"""
Tests that the add_friend() function adds the specified person to the set
Takes a Person, friendToAdd and returns True if the specified friend was added to the set or False if it wasn't
"""
#creates a friend Person object and sets their name and age
friendV = Person()
friendV.set_name("Vanessa")
friendV.set_age(18)
Network.add_friend(self, friendV)
#check that the friend was added to the set
self.assertTrue(friendV in Network.setOfFriends)
def main():
"""
calls the test functions and will print whether or not the functions in the Network class work
depending if the test functions return True or False
"""
#create a network object
socialN = Network()
#creates a user person object
userM = Person()
userM.set_name("Marilyn")
userM.set_age(18)
#creates a friend Person object and sets their name and age
friendV = Person()
friendV.set_name("Vanessa")
friendV.set_age(18)
#call test_add_friend() passing the friend as parameters
#if the method returns True then the add_friend() function works
if (TestNetwork.test_add_friend(socialN, friendV) == True):
print("add_friend() function works")
else:
print("add_friend() function doesn't work")
#creates another friend object and adds her to the set
friendA = Person()
friendA.set_name("Andrea")
friendA.set_age(15)
Network.add_friend(socialN, friendA)
#call the test_get_friend() passing socialN as parameters
#if the method returns True then the get_friend() function works
if (TestNetwork.test_get_friends(socialN) == True):
print("get_friends() function works")
else:
print("get_friends() function doesn't work")
#call the test_get_friends_same_age() passing socialN as parameters
#if the method returns True then the get_friend_same_age() function works
if (TestNetwork.test_get_friends_same_age(socialN, userM) == True):
print("get_friends_same_age() function works")
else:
print("get_friends_same_age() function doesn't work")
#call the test_friend_to_remove() function
#if the function returns True then the friend_to_remove() function works
if (TestNetwork.test_friend_to_remove(socialN) == True):
print("friend_to_remove() function works")
else:
print("friend_to_remove() function doesn't work")
#call test_remove_friend() passing the friend as parameters
#if the method returns True then the remove_friend() function works
if (TestNetwork.test_remove_friend(socialN, friendV) == True):
print("remove_friend() function works")
else:
print("remove_friend() function doesn't work")
#call test_remove_all_friends()
#if the method returns True then the remove_all_friends() function works
if (TestNetwork.test_remove_all_friends(socialN) == True):
print("remove_all_friends() function works")
else:
print("remove_all_friends() function doesn't work")
def main():
"""
calls the test functions and will print whether or not the functions in the Network class work
depending if the test functions return True or False
"""
#create a network object
socialN = Network()
#creates a user person object
userM = Person()
userM.set_name("Marilyn")
userM.set_age(18)
#creates a friend Person object and sets their name and age
friendV = Person()
friendV.set_name("Vanessa")
friendV.set_age(18)
#call test_add_friend() passing the friend as parameters
#if the method returns True then the add_friend() function works
if (TestNetwork.test_add_friend(socialN,friendV) == True):
print ("add_friend() function works")
else:
print ("add_friend() function doesn't work")
#creates another friend object and adds her to the set
friendA = Person()
friendA.set_name("Andrea")
friendA.set_age(15)
Network.add_friend(socialN, friendA)
#call the test_get_friend() passing socialN as parameters
#if the method returns True then the get_friend() function works
if(TestNetwork.test_get_friends(socialN)==True):
print("get_friends() function works")
else:
print("get_friends() function doesn't work")
#call the test_get_friends_same_age() passing socialN as parameters
#if the method returns True then the get_friend_same_age() function works
if(TestNetwork.test_get_friends_same_age(socialN, userM)==True):
print("get_friends_same_age() function works")
else:
print("get_friends_same_age() function doesn't work")
#call the test_friend_to_remove() function
#if the function returns True then the friend_to_remove() function works
if(TestNetwork.test_friend_to_remove(socialN) == True):
print("friend_to_remove() function works")
else:
print("friend_to_remove() function doesn't work")
#call test_remove_friend() passing the friend as parameters
#if the method returns True then the remove_friend() function works
if (TestNetwork.test_remove_friend(socialN, friendV) == True):
print ("remove_friend() function works")
else:
print ("remove_friend() function doesn't work")
#call test_remove_all_friends()
#if the method returns True then the remove_all_friends() function works
if (TestNetwork.test_remove_all_friends(socialN) == True):
print ("remove_all_friends() function works")
else:
print ("remove_all_friends() function doesn't work")
