def test_img_to_buf_source(self): img = new("RGB", (256, 256), "red") img.save("test2.jpg") img.format = "JPEG" data = img_to_buf(img, 'source').read() # a 'JFIF' chunk in the bitstream indicates a .jpg image assert 'JFIF' in data
def test_img_to_buf_png(self): img = new("RGB", (256, 256), "red") img.save("test1.png", 'PNG') data = img_to_buf(img, 'png').read() # ImageHeaDeR, ImageDATa, and ImageEND are # all necessary chunks in a .PNG bitstream assert 'IHDR' in data and 'IDAT' in data and 'IEND' in data
def test_insert_image_blob(self): img = new("RGB", (256, 256), "red") data = img_to_buf(img, 'jpeg').read() tempDB = self.__make_tempDB() tempDB.insert_image_blob(0, 0, 0, Binary(data)) result = tempDB.cursor.execute("select count(*) from tiles;") assert result.fetchone()[0] == 1
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_img_to_buf_jpg(self): img = new("RGB", (256, 256), "red") data = img_to_buf(img, 'jpeg').read() # a 'JFIF' chunk in the bitstream indicates a .jpg image assert 'JFIF' in data