def test_combine_worker_dbs(): session_folder = make_session_folder() # make a random number of tempdbs with dummy data img = new("RGB", (256, 256), "red") data = img_to_buf(img, 'jpeg').read() z = randint(2, 5) for x in xrange(z): TempDB(session_folder).insert_image_blob(x, 0, 0, Binary(data)) # confirm that combine_worker_dbs assimilates all tempdb's into gpkg chdir(session_folder) # necessary to put gpkg in session_folder gpkg = Geopackage("test.gpkg", 4326) combine_worker_dbs(gpkg) result = gpkg.execute("select count(*) from tiles;") assert (result.fetchone())[0] == z
def test_assimilate_error(self): session_folder = make_session_folder() chdir(session_folder) with Geopackage(session_folder, 3395) as gpkg: with raises(IOError): gpkg.assimilate("None") remove(join(getcwd(), gpkg.file_path))
def test_assimilate_error(self): session_folder = make_session_folder() chdir(session_folder) gpkg = Geopackage(session_folder, 3395) remove(join(getcwd(), gpkg.file_path)) gpkg.execute("") OE = None try: gpkg.assimilate("None") except OperationalError as OE: print "OperationalError should occur." finally: assert OE is not None and type(OE) == OperationalError
def make_gpkg(): filename = uuid4().hex + '.gpkg' tmp_file = join(gettempdir(), filename) return Geopackage(tmp_file, 4326)
def test_execute_return(self): session_folder = make_session_folder() chdir(session_folder) gpkg = Geopackage(session_folder, 9804) result = gpkg.execute("select count(*) from tiles;") assert (result.fetchone())[0] == 0
def test_file_path(self): filename = uuid4().hex + '.gpkg' tmp_dir = gettempdir() tmp_file = join(tmp_dir, filename) gpkg = Geopackage(tmp_file, 3857) assert gpkg.file_path == tmp_file