X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/8f63f9516b276af8391d71e285db44a7b7214803..4bca5e838bc9f0aa5ebdb412b702eedc97df30c2:/test/python/test_tools_tiger_data.py diff --git a/test/python/test_tools_tiger_data.py b/test/python/test_tools_tiger_data.py index f97a5713..2e524087 100644 --- a/test/python/test_tools_tiger_data.py +++ b/test/python/test_tools_tiger_data.py @@ -1,61 +1,137 @@ """ Test for tiger data function """ -from pathlib import Path +import tarfile +from textwrap import dedent import pytest -import tarfile -from nominatim.tools import tiger_data, database_import +from nominatim.tools import tiger_data +from nominatim.errors import UsageError +class MockTigerTable: -@pytest.mark.parametrize("threads", (1, 5)) -def test_add_tiger_data(def_config, tmp_path, sql_preprocessor, - temp_db_cursor, threads, temp_db_with_extensions): - temp_db_cursor.execute('CREATE TABLE place (id INT)') - sqlfile = tmp_path / '1010.sql' - sqlfile.write_text("""INSERT INTO place values (1); - INSERT INTO non_existant_table values (1);""") - tiger_data.add_tiger_data(str(tmp_path), def_config, threads) + def __init__(self, conn): + self.conn = conn + with conn.cursor() as cur: + cur.execute("""CREATE TABLE tiger (linegeo GEOMETRY, + start INTEGER, + stop INTEGER, + interpol TEXT, + token_info JSONB, + postcode TEXT)""") + + def count(self): + with self.conn.cursor() as cur: + return cur.scalar("SELECT count(*) FROM tiger") + + def row(self): + with self.conn.cursor() as cur: + cur.execute("SELECT * FROM tiger LIMIT 1") + return cur.fetchone() + +@pytest.fixture +def tiger_table(def_config, temp_db_conn, sql_preprocessor, + temp_db_with_extensions, tmp_path): + def_config.lib_dir.sql = tmp_path / 'sql' + def_config.lib_dir.sql.mkdir() + + (def_config.lib_dir.sql / 'tiger_import_start.sql').write_text( + """CREATE OR REPLACE FUNCTION tiger_line_import(linegeo GEOMETRY, start INTEGER, + stop INTEGER, interpol TEXT, + token_info JSONB, postcode TEXT) + RETURNS INTEGER AS $$ + INSERT INTO tiger VALUES(linegeo, start, stop, interpol, token_info, postcode) + RETURNING 1 + $$ LANGUAGE SQL;""") + (def_config.lib_dir.sql / 'tiger_import_finish.sql').write_text( + """DROP FUNCTION tiger_line_import (linegeo GEOMETRY, in_startnumber INTEGER, + in_endnumber INTEGER, interpolationtype TEXT, + token_info JSONB, in_postcode TEXT);""") - assert temp_db_cursor.table_rows('place') == 1 + return MockTigerTable(temp_db_conn) + + +@pytest.fixture +def csv_factory(tmp_path): + def _mk_file(fname, hnr_from=1, hnr_to=9, interpol='odd', street='Main St', + city='Newtown', state='AL', postcode='12345', + geometry='LINESTRING(-86.466995 32.428956,-86.466923 32.428933)'): + (tmp_path / (fname + '.csv')).write_text(dedent("""\ + from;to;interpolation;street;city;state;postcode;geometry + {};{};{};{};{};{};{};{} + """.format(hnr_from, hnr_to, interpol, street, city, state, + postcode, geometry))) + + return _mk_file @pytest.mark.parametrize("threads", (1, 5)) -def test_add_tiger_data_bad_file(def_config, tmp_path, sql_preprocessor, - temp_db_cursor, threads, temp_db_with_extensions): - temp_db_cursor.execute('CREATE TABLE place (id INT)') - sqlfile = tmp_path / '1010.txt' +def test_add_tiger_data(def_config, src_dir, tiger_table, tokenizer_mock, threads): + tiger_data.add_tiger_data(str(src_dir / 'test' / 'testdb' / 'tiger'), + def_config, threads, tokenizer_mock()) + + assert tiger_table.count() == 6213 + + +def test_add_tiger_data_no_files(def_config, tiger_table, tokenizer_mock, + tmp_path): + tiger_data.add_tiger_data(str(tmp_path), def_config, 1, tokenizer_mock()) + + assert tiger_table.count() == 0 + + +def test_add_tiger_data_bad_file(def_config, tiger_table, tokenizer_mock, + tmp_path): + sqlfile = tmp_path / '1010.csv' sqlfile.write_text("""Random text""") - tiger_data.add_tiger_data(str(tmp_path), def_config, threads) - assert temp_db_cursor.table_rows('place') == 0 + tiger_data.add_tiger_data(str(tmp_path), def_config, 1, tokenizer_mock()) + + assert tiger_table.count() == 0 + + +def test_add_tiger_data_hnr_nan(def_config, tiger_table, tokenizer_mock, + csv_factory, tmp_path): + csv_factory('file1', hnr_from=99) + csv_factory('file2', hnr_from='L12') + csv_factory('file3', hnr_to='12.4') + + tiger_data.add_tiger_data(str(tmp_path), def_config, 1, tokenizer_mock()) + + assert tiger_table.count() == 1 + assert tiger_table.row()['start'] == 99 @pytest.mark.parametrize("threads", (1, 5)) -def test_add_tiger_data_tarfile(def_config, tmp_path, temp_db_cursor, - threads, temp_db_with_extensions, sql_preprocessor): - temp_db_cursor.execute('CREATE TABLE place (id INT)') - sqlfile = tmp_path / '1010.sql' - sqlfile.write_text("""INSERT INTO place values (1); - INSERT INTO non_existant_table values (1);""") - tar = tarfile.open("sample.tar.gz", "w:gz") - tar.add(sqlfile) +def test_add_tiger_data_tarfile(def_config, tiger_table, tokenizer_mock, + tmp_path, src_dir, threads): + tar = tarfile.open(str(tmp_path / 'sample.tar.gz'), "w:gz") + tar.add(str(src_dir / 'test' / 'testdb' / 'tiger' / '01001.csv')) tar.close() - tiger_data.add_tiger_data(str(tmp_path), def_config, threads) - assert temp_db_cursor.table_rows('place') == 1 + tiger_data.add_tiger_data(str(tmp_path / 'sample.tar.gz'), def_config, threads, + tokenizer_mock()) + assert tiger_table.count() == 6213 -@pytest.mark.parametrize("threads", (1, 5)) -def test_add_tiger_data_bad_tarfile(def_config, tmp_path, temp_db_cursor, threads, - temp_db_with_extensions, sql_preprocessor): - temp_db_cursor.execute('CREATE TABLE place (id INT)') - sqlfile = tmp_path / '1010.txt' - sqlfile.write_text("""Random text""") - tar = tarfile.open("sample.tar.gz", "w:gz") - tar.add(sqlfile) + +def test_add_tiger_data_bad_tarfile(def_config, tiger_table, tokenizer_mock, + tmp_path): + tarfile = tmp_path / 'sample.tar.gz' + tarfile.write_text("""Random text""") + + with pytest.raises(UsageError): + tiger_data.add_tiger_data(str(tarfile), def_config, 1, tokenizer_mock()) + + +def test_add_tiger_data_empty_tarfile(def_config, tiger_table, tokenizer_mock, + tmp_path): + tar = tarfile.open(str(tmp_path / 'sample.tar.gz'), "w:gz") + tar.add(__file__) tar.close() - tiger_data.add_tiger_data(str(tmp_path), def_config, threads) - assert temp_db_cursor.table_rows('place') == 0 + tiger_data.add_tiger_data(str(tmp_path / 'sample.tar.gz'), def_config, 1, + tokenizer_mock()) + + assert tiger_table.count() == 0