47 lines
1.7 KiB
Python
47 lines
1.7 KiB
Python
# vim: set fileencoding=utf-8 :
|
|
import pytest
|
|
|
|
import pyvips
|
|
from helpers import MOSAIC_FILES, MOSAIC_MARKS, MOSAIC_VERTICAL_MARKS
|
|
|
|
class TestMosaicing:
|
|
def test_mosaic(self):
|
|
# ported from https://github.com/libvips/nip2/tree/master/share/nip2/data/examples/1_point_mosaic
|
|
|
|
mosaiced_image = None
|
|
|
|
for i in range(0, len(MOSAIC_FILES), 2):
|
|
files = MOSAIC_FILES[i:i + 2]
|
|
marks = MOSAIC_MARKS[i:i + 2]
|
|
|
|
im = pyvips.Image.new_from_file(files[0])
|
|
sec_im = pyvips.Image.new_from_file(files[1])
|
|
horizontal_part = im.mosaic(sec_im,
|
|
pyvips.Direction.HORIZONTAL,
|
|
marks[0][0], marks[0][1],
|
|
marks[1][0], marks[1][1])
|
|
|
|
if mosaiced_image is None:
|
|
mosaiced_image = horizontal_part
|
|
else:
|
|
vertical_marks = MOSAIC_VERTICAL_MARKS[i - 2:i]
|
|
mosaiced_image = mosaiced_image.mosaic(horizontal_part,
|
|
pyvips.Direction.VERTICAL,
|
|
vertical_marks[1][0], vertical_marks[1][1],
|
|
vertical_marks[0][0], vertical_marks[0][1])
|
|
|
|
mosaiced_image = mosaiced_image.globalbalance()
|
|
|
|
# Uncomment to see output file
|
|
#mosaiced_image.write_to_file('after.jpg')
|
|
|
|
# hard to test much more than this
|
|
assert mosaiced_image.width == 1005
|
|
assert mosaiced_image.height == 1295
|
|
assert mosaiced_image.interpretation == pyvips.Interpretation.B_W
|
|
assert mosaiced_image.format == pyvips.BandFormat.FLOAT
|
|
assert mosaiced_image.bands == 1
|
|
|
|
if __name__ == '__main__':
|
|
pytest.main()
|