UCL-RITS#167

week09/refactoring/initial_person_class.py

@@ -16,7 +16,7 @@ def add_connection(self, person, relation):
 
     def forget(self, person):
         """Removes any connections to a person"""
-        pass
+        self.connections.pop(person,None)
 
 
 def average_age(group):
@@ -28,13 +28,22 @@ def average_age(group):
 if __name__ == "__main__":
     # ...then create the group members one by one...
     jill = Person("Jill", 26, "biologist")
+    john = Person("John", 27, "writer") 
+    zalika = Person("Zalika", 28, "artist")   
+    nash = Person("Nash", 34, "chef")  
 
     # ...then add the connections one by one...
     # Note: this will fail from here if the person objects aren't created
     jill.add_connection(zalika, "friend")
-
+    jill.add_connection(john, "partner")
+    john.add_connection(jill, "partner")
+    zalika.add_connection(jill, "friend")
+    nash.add_connection(john, "cousin")
+    nash.add_connection(zalika, "landlord")
+
     # ... then forget Nash and John's connection
     nash.forget(john)
+
     # Then create the group
     my_group = {jill, zalika, john, nash}