def test_cached(self) -> None:
"""Tests the case when the pickle cache is already available."""
refdir = os.path.join(os.path.dirname(__file__), "refdir")
refpath = os.path.join(refdir, "hazszamok_20190511.tsv")
util.build_reference_cache(refpath, "01")
memory_cache = util.build_reference_cache(refpath, "01")
expected = {'01': {'011': {'Hamzsabégi út': hnr_list(['1']),
'Ref Name 1': hnr_list(['1', '2']),
'Törökugrató utca': hnr_list(['1', '10', '11', '12', '2', '7']),
'Tűzkő utca': hnr_list(['1', '10', '2', '9'])}}}
self.assertEqual(memory_cache, expected)
os.unlink(util.get_reference_cache_path(refpath, "01"))
def test_missing(self) -> None:
"""Tests the case when the street is not in the reference."""
relations = get_relations()
refdir = os.path.join(os.path.dirname(__file__), "refdir")
refpath = os.path.join(refdir, "hazszamok_20190511.tsv")
memory_cache = util.build_reference_cache(refpath, "01")
relation_name = "gazdagret"
street = "No such utca"
relation = relations.get_relation(relation_name)
ret = relation.build_ref_housenumbers(memory_cache, street, "")
self.assertEqual(ret, [])
def test_happy(self) -> None:
"""Tests the happy path."""
refdir = os.path.join(os.path.dirname(__file__), "refdir")
refpath = os.path.join(refdir, "hazszamok_20190511.tsv")
memory_cache = util.build_reference_cache(refpath)
expected = {
'01': {
'011': {
'Ref Name 1': ['1', '2'],
'Törökugrató utca': ['1', '10', '11', '12', '2', '7'],
'Tűzkő utca': ['1', '10', '2', '9'],
'Hamzsabégi út': ['1']
}
}
}
self.assertEqual(memory_cache, expected)
os.unlink(refpath + ".pickle")
def test_happy(self) -> None:
"""Tests the happy path."""
refdir = os.path.join(os.path.dirname(__file__), "refdir")
relations = get_relations()
refpath = os.path.join(refdir, "hazszamok_20190511.tsv")
memory_cache = util.build_reference_cache(refpath, "01")
relation_name = "gazdagret"
street = "Törökugrató utca"
relation = relations.get_relation(relation_name)
ret = relation.build_ref_housenumbers(memory_cache, street, "")
expected = [
'Törökugrató utca\t1\tcomment',
'Törökugrató utca\t10\t',
'Törökugrató utca\t11\t',
'Törökugrató utca\t12\t',
'Törökugrató utca\t2\t',
'Törökugrató utca\t7\t',
]
self.assertEqual(ret, expected)
def test_happy(self) -> None:
"""Tests the happy path."""
with unittest.mock.patch('util.get_abspath', get_abspath):
refdir = os.path.join(os.path.dirname(__file__), "refdir")
relations = get_relations()
refpath = os.path.join(refdir, "hazszamok_20190511.tsv")
memory_cache = util.build_reference_cache(refpath)
relation_name = "gazdagret"
street = "Törökugrató utca"
relation = relations.get_relation(relation_name)
ret = relation.build_ref_housenumbers(memory_cache, street, "")
expected = [
'Törökugrató utca\t1',
'Törökugrató utca\t10',
'Törökugrató utca\t11',
'Törökugrató utca\t12',
'Törökugrató utca\t2',
'Törökugrató utca\t7',
]
self.assertEqual(ret, expected)
