def testGetTotalReservableCapacity(self): default_capacity = getDefaultValue(Room, 'capacity') for l, loc in self.iterLocations(): total = sum([r.get('capacity', default_capacity) for r in l.get('rooms', []) if r.get('is_reservable', True)]) assert total == loc.getTotalReservableCapacity()
def testGetDefaultValue(self): # TODO: defaults must be put into config assert utils.getDefaultValue(Room, 'capacity') == 20 assert utils.getDefaultValue(Reservation, 'is_cancelled') == False with self.assertRaises(RuntimeError): utils.getDefaultValue(Room, 'comments')