def test_get_from_file(self):
     office = Space(1, 'prooms', 1)
     d = office.get_from_file(r'data/prooms.txt')
     # check if the number list items equals number of lines ina file
     with open('data/prooms.txt') as f:
         total = sum(1 for _ in f)
     self.assertTrue(type(d), type([]))
     # check that the lit has equal number lines and
     self.assertTrue(len(d), total)
 def test_allocate_room(self):
     office = Space(1, 'prooms')
     # check if dictionary has been created
     self.assertTrue(type(office.room_list), type({}))
     # check if number passed is corrected
     self.assertEqual(office.num_rooms, 1)
     # check if the list of room coincides with the number of rooms
     # generated
     self.assertEqual(len(office.room_list), 1)
     # check if naming convention is right and it exists
     self.assertTrue('prooms 0' in office.room_list.keys(), True)
     # check if the list created has empty list
     self.assertListEqual(office.room_list['prooms 0'], [])
     # allocate a room
     name = 'david'
     r_alloc = (office.allocate_room(name))
     # check if name assigned is returned
     self.assertIn(name, r_alloc,)
     # remove test file
     os.remove(office.path)