def create_school(iped, data): school = School(school_id=iped, data_json=STARTER_DATA_JSON) for field in data: setattr(school, field, data[field]) school.zip5 = school.zip5[:5] school.save() _alias = data.get('alias') if _alias: create_alias(_alias, school)
def test_get_school(self): """test grabbing a school by ID""" closed_school = School(pk=999999, operating=False) closed_school.save() test1 = get_school('155317') self.assertTrue(test1.pk == 155317) test2 = get_school('xxx') self.assertTrue(test2 is None) test3 = get_school('999999') self.assertTrue(test3 is None)
def create_school(id, data): school = School(school_id=id, data_json=STARTER_DATA_JSON) for field in data: if field == 'alias': ALIAS = data['alias'] else: setattr(school, field, data[field]) school.zip5 = school.zip5[:5] school.save() create_alias(ALIAS, school)
def create_school(id, data): school = School(school_id=id, data_json=STARTER_DATA_JSON) for field in data: if field == 'alias': ALIAS = data['alias'] else: setattr(school, field, data[field]) school.zip5 = school.zip5[:5] school.save() alias = create_alias(ALIAS, school)
def test_get_program_length(self): school = School(school_id=123456, degrees_highest='2') program = Program(institution=school, level='2') bad_school = School(school_id=999999, degrees_highest='5') test1 = get_program_length(program=program, school=school) self.assertTrue(test1 == 2) test2 = get_program_length(program='', school=school) self.assertTrue(test2 == 2) test3 = get_program_length(program='', school='') self.assertIs(test3, None) program.level = '3' test4 = get_program_length(program=program, school='') self.assertEqual(test4, 4) bad_school_test = get_program_length(program='', school=bad_school) self.assertIs(bad_school_test, None)
def test_get_program_length(self): school = School(school_id='123456', degrees_highest='2') program = Program(institution=school, level='2') test1 = get_program_length(program=program, school=school) self.assertTrue(test1 == 2) test2 = get_program_length(program='', school=school) self.assertTrue(test2 == 2) test3 = get_program_length(program='', school='') self.assertTrue(test3 is None) program.level = '3' test4 = get_program_length(program=program, school='') self.assertTrue(test4 == 4)
def test_prepare(self): school = School(school_id="999999", city="Example City", state="VA", zip5="12345") doc = SchoolDocument() prepared_data = doc.prepare(school) self.assertEqual( prepared_data, { 'autocomplete': doc.prepare_autocomplete(school), 'city': school.city, 'nicknames': doc.prepare_nicknames(school), 'school_id': school.school_id, 'state': school.state, 'text': school.primary_alias, 'zip5': school.zip5, })
def test_get_region_failure(self): school = School(school_id='123456', state='') self.assertTrue(get_region(school) == '')
def test_get_region(self): school = School(school_id='123456', state='NE') self.assertTrue(get_region(school) == 'MW')