def testRemoveLocationByName(self): test_location_name = 'test_location' db.session.add(Location(name=test_location_name)) db.session.commit() assert Location.getLocationByName(test_location_name) is not None Location.removeLocationByName(test_location_name) db.session.commit() assert Location.getLocationByName(test_location_name) is None
def iterLocations(self): for l in LOCATIONS: loc = Location.getLocationByName(l['name']) yield l, loc db.session.add(loc) db.session.commit()