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_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 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')
