add tests for cleaning housenumbers

This commit is contained in:
Sarah Hoffmann
2022-01-20 23:47:20 +01:00
parent 3ce123ab69
commit c170d323d9
7 changed files with 106 additions and 1 deletions

View File

@@ -58,6 +58,14 @@ class MockIcuWordTable:
self.conn.commit()
def add_housenumber(self, word_id, word_token):
with self.conn.cursor() as cur:
cur.execute("""INSERT INTO word (word_id, word_token, type)
VALUES (%s, %s, 'H')
""", (word_id, word_token))
self.conn.commit()
def count(self):
with self.conn.cursor() as cur:
return cur.scalar("SELECT count(*) FROM word")
@@ -68,6 +76,11 @@ class MockIcuWordTable:
return cur.scalar("SELECT count(*) FROM word WHERE type = 'S'")
def count_housenumbers(self):
with self.conn.cursor() as cur:
return cur.scalar("SELECT count(*) FROM word WHERE type = 'H'")
def get_special(self):
with self.conn.cursor() as cur:
cur.execute("SELECT word_token, info, word FROM word WHERE type = 'S'")