Python Office.rooms Examples

Example #1

File: office_tests.py Project: GDP254/a-21-project1

 def test_add_office(self):
     office = Office('MyOffice78')
     initial_room_count = len(Dojo.rooms())
     initial_office_count = len(Office.rooms())
     Office.add(office)
     new_room_count = len(Dojo.rooms())
     new_office_count = len(Office.rooms())
     self.assertEqual([initial_room_count + 1, initial_office_count + 1],
                      [new_room_count, new_office_count])

Example #2

File: office_tests.py Project: GDP254/a-21-project1

 def test_remove_office(self):
     office = Office('MyOffice89')
     Office.add(office)
     initial_room_count = len(Dojo.rooms())
     initial_office_count = len(Office.rooms())
     Office.remove(office)
     new_room_count = len(Dojo.rooms())
     new_office_count = len(Office.rooms())
     self.assertEqual([initial_room_count - 1, initial_office_count - 1],
                      [new_room_count, new_office_count])