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')
