def test_site_list_urls(self): locations = xmlparsing.site_list_urls('testfiles/sitelistpartial.xml') assert locations[u'Pointe-à-la-Croix'] == ('QC', 's0000810') locations = xmlparsing.site_list_urls('testfiles/sitelist.xml') assert locations['Toronto Island'][0] == 'ON' assert locations['Toronto Island'][1] == 's0000785'
def test_db_add_locations(self): locations = xmlparsing.site_list_urls('testfiles/sitelist.xml') f = open('testfiles/sitecodesdict.txt', 'w') f.write(str(locations)) xmlparsing.db_add_locations(redis.test, locations) codes = redis.test.get('site_codes') f = open('testfiles/sitecodesdict.txt', 'r') # a failure here may just indicate that the codes have changed assert codes == f.read() codesDict = ast.literal_eval(codes) loc = u'Rivière-du-Loup' assert codesDict[loc] == ('QC', 's0000253') loc = u'Îles-de-la-Madeleine' assert codesDict[loc] == ('QC', 's0000174')