def test_import_ordering(self): filenames = ['cities1000.txt', 'countryInfo.txt'] filepaths = [get_tst_filepath(fn) for fn in filenames] importers = get_importer_instances(self.session, *filepaths) eq_(importers[0].filename, 'countryInfo.txt') filenames = ['featureCodes_en.txt', 'cities1000.txt', 'countryInfo.txt'] filepaths = [get_tst_filepath(fn) for fn in filenames] importers = get_importer_instances(self.session, *filepaths) eq_(importers[-1].filename, 'cities1000.txt')
def run_importers(db_session, download_dir, local_filepaths): from sqlalchemy_geonames.imports import get_importer_instances for importer in get_importer_instances(db_session, download_dir, *local_filepaths): print("Running importer for {}...".format(importer.filename)) importer.run()
def test_imports(self): filepaths = [get_tst_filepath(fn) for fn in test_filenames] importers = get_importer_instances(self.session, *filepaths) for importer in importers: importer.run() assert_greater(self.session.query(importer.model).count(), 1)