def setUpClass(self):
make_test_databases.remove_existing_databases()
make_test_databases.make_assignee_db()
make_test_databases.make_inventor_db()
self.conn = sqlite3.connect("hashTbl.sqlite3")
self.cursor = self.conn.cursor()
geocode_setup.create_sql_helper_functions(self.conn)
geocode_setup.create_hashtbl(self.cursor, self.conn)
def test_fix_city_country(self):
make_test_databases.make_assignee_db()
geocode_setup.fix_city_country(self.c)
# Inspect loc table in hashTbl, find something to assert.
rows = self.c.execute('select * from loc order by city').fetchall()
element = rows[0][1]
self.assertTrue('BERLIN'==element,"{0} should be {1}".format(element,'BERLIN'))
element = rows[0][3]
self.assertTrue('DE' == element,"{0} should be {1}".format(element,'DE'))
element = rows[2][1]
self.assertTrue('FAIRHAVEN' == element,"{0} should be {1}".format(element,'FAIRHAVEN'))
element = rows[2][2]
self.assertTrue('MA' == element,"{0} should be {1}".format(element,'MA'))
pass
