def print_persons(self): """Print a list of persons in amity. Returns String: names and identifier of persons in Amity """ persons = self.amity.persons persons_string = "List of Persons with Id\n" for i in persons: person = persons[i] persons_string += (person.name + " " + Amity.get_person_type(person) + " " + person.identifier + "\n") return persons_string
def print_person(self, person_obj): """Build a string of person object printing out the name. Arguments: person_obj - person to printed to file Returns: string: of the persons name, person_type and accomodation status """ name = person_obj.name.upper() want_accomodation = "" person_type = Amity.get_person_type(person_obj) if person_type == "fellow": want_accomodation = person_obj.wants_accom return (name + ' ' + person_type.upper() + ' ' + want_accomodation + "\n")
def test_get_person_type_two(self): self.assertEqual(Amity.get_person_type(self.staffB), 'staff')
def test_get_person_type_three(self): self.assertIsNone(Amity.get_person_type({}))
def test_get_person_type(self): self.assertEqual(Amity.get_person_type(self.fellowA), 'fellow')