libvips/test/test-suite/test_create.py

465 lines
15 KiB
Python

# vim: set fileencoding=utf-8 :
import pytest
import pyvips
from helpers import assert_almost_equal_objects
class TestCreate:
def test_black(self):
im = pyvips.Image.black(100, 100)
assert im.width == 100
assert im.height == 100
assert im.format == pyvips.BandFormat.UCHAR
assert im.bands == 1
for i in range(0, 100):
pixel = im(i, i)
assert len(pixel) == 1
assert pixel[0] == 0
im = pyvips.Image.black(100, 100, bands=3)
assert im.width == 100
assert im.height == 100
assert im.format == pyvips.BandFormat.UCHAR
assert im.bands == 3
for i in range(0, 100):
pixel = im(i, i)
assert len(pixel) == 3
assert_almost_equal_objects(pixel, [0, 0, 0])
def test_buildlut(self):
M = pyvips.Image.new_from_array([[0, 0],
[255, 100]])
lut = M.buildlut()
assert lut.width == 256
assert lut.height == 1
assert lut.bands == 1
p = lut(0, 0)
assert p[0] == 0.0
p = lut(255, 0)
assert p[0] == 100.0
p = lut(10, 0)
assert p[0] == 100 * 10.0 / 255.0
M = pyvips.Image.new_from_array([[0, 0, 100],
[255, 100, 0],
[128, 10, 90]])
lut = M.buildlut()
assert lut.width == 256
assert lut.height == 1
assert lut.bands == 2
p = lut(0, 0)
assert_almost_equal_objects(p, [0.0, 100.0])
p = lut(64, 0)
assert_almost_equal_objects(p, [5.0, 95.0])
def test_eye(self):
im = pyvips.Image.eye(100, 90)
assert im.width == 100
assert im.height == 90
assert im.bands == 1
assert im.format == pyvips.BandFormat.FLOAT
assert im.max() == 1.0
assert im.min() == -1.0
im = pyvips.Image.eye(100, 90, uchar=True)
assert im.width == 100
assert im.height == 90
assert im.bands == 1
assert im.format == pyvips.BandFormat.UCHAR
assert im.max() == 255.0
assert im.min() == 0.0
@pytest.mark.skipif(pyvips.type_find("VipsOperation", "fwfft") == 0,
reason="no FFTW, skipping test")
def test_fractsurf(self):
im = pyvips.Image.fractsurf(100, 90, 2.5)
assert im.width == 100
assert im.height == 90
assert im.bands == 1
assert im.format == pyvips.BandFormat.FLOAT
def test_gaussmat(self):
im = pyvips.Image.gaussmat(1, 0.1)
assert im.width == 5
assert im.height == 5
assert im.bands == 1
assert im.format == pyvips.BandFormat.DOUBLE
assert im.max() == 20
total = im.avg() * im.width * im.height
scale = im.get("scale")
assert total == scale
p = im(im.width / 2, im.height / 2)
assert p[0] == 20.0
im = pyvips.Image.gaussmat(1, 0.1,
separable=True, precision="float")
assert im.width == 5
assert im.height == 1
assert im.bands == 1
assert im.format == pyvips.BandFormat.DOUBLE
assert im.max() == 1.0
total = im.avg() * im.width * im.height
scale = im.get("scale")
assert total == scale
p = im(im.width / 2, im.height / 2)
assert p[0] == 1.0
def test_gaussnoise(self):
im = pyvips.Image.gaussnoise(100, 90)
assert im.width == 100
assert im.height == 90
assert im.bands == 1
assert im.format == pyvips.BandFormat.FLOAT
im = pyvips.Image.gaussnoise(100, 90, sigma=10, mean=100)
assert im.width == 100
assert im.height == 90
assert im.bands == 1
assert im.format == pyvips.BandFormat.FLOAT
sigma = im.deviate()
mean = im.avg()
assert sigma == pytest.approx(10, abs=0.4)
assert mean == pytest.approx(100, abs=0.4)
def test_grey(self):
im = pyvips.Image.grey(100, 90)
assert im.width == 100
assert im.height == 90
assert im.bands == 1
assert im.format == pyvips.BandFormat.FLOAT
p = im(0, 0)
assert p[0] == 0.0
p = im(99, 0)
assert p[0] == 1.0
p = im(0, 89)
assert p[0] == 0.0
p = im(99, 89)
assert p[0] == 1.0
im = pyvips.Image.grey(100, 90, uchar=True)
assert im.width == 100
assert im.height == 90
assert im.bands == 1
assert im.format == pyvips.BandFormat.UCHAR
p = im(0, 0)
assert p[0] == 0
p = im(99, 0)
assert p[0] == 255
p = im(0, 89)
assert p[0] == 0
p = im(99, 89)
assert p[0] == 255
def test_identity(self):
im = pyvips.Image.identity()
assert im.width == 256
assert im.height == 1
assert im.bands == 1
assert im.format == pyvips.BandFormat.UCHAR
p = im(0, 0)
assert p[0] == 0.0
p = im(255, 0)
assert p[0] == 255.0
p = im(128, 0)
assert p[0] == 128.0
im = pyvips.Image.identity(ushort=True)
assert im.width == 65536
assert im.height == 1
assert im.bands == 1
assert im.format == pyvips.BandFormat.USHORT
p = im(0, 0)
assert p[0] == 0
p = im(99, 0)
assert p[0] == 99
p = im(65535, 0)
assert p[0] == 65535
def test_invertlut(self):
lut = pyvips.Image.new_from_array([[0.1, 0.2, 0.3, 0.1],
[0.2, 0.4, 0.4, 0.2],
[0.7, 0.5, 0.6, 0.3]])
im = lut.invertlut()
assert im.width == 256
assert im.height == 1
assert im.bands == 3
assert im.format == pyvips.BandFormat.DOUBLE
p = im(0, 0)
assert_almost_equal_objects(p, [0, 0, 0])
p = im(255, 0)
assert_almost_equal_objects(p, [1, 1, 1])
p = im(0.2 * 255, 0)
assert p[0] == pytest.approx(0.1, abs=0.1)
p = im(0.3 * 255, 0)
assert p[1] == pytest.approx(0.1, abs=0.1)
p = im(0.1 * 255, 0)
assert p[2] == pytest.approx(0.1, abs=0.1)
def test_matrixinvert(self):
# 4x4 matrix to check if PLU decomposition works
mat = pyvips.Image.new_from_array([[4, 0, 0, 0],
[0, 0, 2, 0],
[0, 1, 2, 0],
[1, 0, 0, 1]])
im = mat.matrixinvert()
assert im.width == 4
assert im.height == 4
assert im.bands == 1
assert im.format == pyvips.BandFormat.DOUBLE
p = im(0, 0)
assert p[0] == 0.25
p = im(3, 3)
assert p[0] == 1.0
def test_logmat(self):
im = pyvips.Image.logmat(1, 0.1)
assert im.width == 7
assert im.height == 7
assert im.bands == 1
assert im.format == pyvips.BandFormat.DOUBLE
assert im.max() == 20
total = im.avg() * im.width * im.height
scale = im.get("scale")
assert total == scale
p = im(im.width / 2, im.height / 2)
assert p[0] == 20.0
im = pyvips.Image.logmat(1, 0.1,
separable=True, precision="float")
assert im.width == 7
assert im.height == 1
assert im.bands == 1
assert im.format == pyvips.BandFormat.DOUBLE
assert im.max() == 1.0
total = im.avg() * im.width * im.height
scale = im.get("scale")
assert total == scale
p = im(im.width / 2, im.height / 2)
assert p[0] == 1.0
def test_mask_butterworth_band(self):
im = pyvips.Image.mask_butterworth_band(128, 128, 2,
0.5, 0.5, 0.7,
0.1)
assert im.width == 128
assert im.height == 128
assert im.bands == 1
assert im.format == pyvips.BandFormat.FLOAT
assert im.max() == pytest.approx(1, abs=0.01)
p = im(32, 32)
assert p[0] == 1.0
im = pyvips.Image.mask_butterworth_band(128, 128, 2,
0.5, 0.5, 0.7,
0.1, uchar=True, optical=True)
assert im.width == 128
assert im.height == 128
assert im.bands == 1
assert im.format == pyvips.BandFormat.UCHAR
assert im.max() == 255
p = im(32, 32)
assert p[0] == 255.0
p = im(64, 64)
assert p[0] == 255.0
im = pyvips.Image.mask_butterworth_band(128, 128, 2,
0.5, 0.5, 0.7,
0.1, uchar=True, optical=True,
nodc=True)
assert im.width == 128
assert im.height == 128
assert im.bands == 1
assert im.format == pyvips.BandFormat.UCHAR
assert im.max() == 255
p = im(32, 32)
assert p[0] == 255.0
p = im(64, 64)
assert p[0] != 255
def test_mask_butterworth(self):
im = pyvips.Image.mask_butterworth(128, 128, 2, 0.7, 0.1,
nodc=True)
assert im.width == 128
assert im.height == 128
assert im.bands == 1
assert im.format == pyvips.BandFormat.FLOAT
assert im.min() == pytest.approx(0, abs=0.01)
p = im(0, 0)
assert p[0] == 0.0
v, x, y = im.maxpos()
assert x == 64
assert y == 64
im = pyvips.Image.mask_butterworth(128, 128, 2, 0.7, 0.1,
optical=True, uchar=True)
assert im.width == 128
assert im.height == 128
assert im.bands == 1
assert im.format == pyvips.BandFormat.UCHAR
assert im.min() == pytest.approx(0, abs=0.01)
p = im(64, 64)
assert p[0] == 255
def test_mask_butterworth_ring(self):
im = pyvips.Image.mask_butterworth_ring(128, 128, 2, 0.7, 0.1, 0.5,
nodc=True)
assert im.width == 128
assert im.height == 128
assert im.bands == 1
assert im.format == pyvips.BandFormat.FLOAT
p = im(45, 0)
assert p[0] == pytest.approx(1.0, abs=0.0001)
v, x, y = im.minpos()
assert x == 64
assert y == 64
def test_mask_fractal(self):
im = pyvips.Image.mask_fractal(128, 128, 2.3)
assert im.width == 128
assert im.height == 128
assert im.bands == 1
assert im.format == pyvips.BandFormat.FLOAT
def test_mask_gaussian_band(self):
im = pyvips.Image.mask_gaussian_band(128, 128, 0.5, 0.5, 0.7, 0.1)
assert im.width == 128
assert im.height == 128
assert im.bands == 1
assert im.format == pyvips.BandFormat.FLOAT
assert im.max() == pytest.approx(1, abs=0.01)
p = im(32, 32)
assert p[0] == 1.0
def test_mask_gaussian(self):
im = pyvips.Image.mask_gaussian(128, 128, 0.7, 0.1,
nodc=True)
assert im.width == 128
assert im.height == 128
assert im.bands == 1
assert im.format == pyvips.BandFormat.FLOAT
assert im.min() == pytest.approx(0, abs=0.01)
p = im(0, 0)
assert p[0] == 0.0
def test_mask_gaussian_ring(self):
im = pyvips.Image.mask_gaussian_ring(128, 128, 0.7, 0.1, 0.5,
nodc=True)
assert im.width == 128
assert im.height == 128
assert im.bands == 1
assert im.format == pyvips.BandFormat.FLOAT
p = im(45, 0)
assert p[0] == pytest.approx(1.0, abs=0.001)
def test_mask_ideal_band(self):
im = pyvips.Image.mask_ideal_band(128, 128, 0.5, 0.5, 0.7)
assert im.width == 128
assert im.height == 128
assert im.bands == 1
assert im.format == pyvips.BandFormat.FLOAT
assert im.max() == pytest.approx(1, abs=0.01)
p = im(32, 32)
assert p[0] == 1.0
def test_mask_ideal(self):
im = pyvips.Image.mask_ideal(128, 128, 0.7,
nodc=True)
assert im.width == 128
assert im.height == 128
assert im.bands == 1
assert im.format == pyvips.BandFormat.FLOAT
assert im.min() == pytest.approx(0, abs=0.01)
p = im(0, 0)
assert p[0] == 0.0
def test_mask_gaussian_ring_2(self):
im = pyvips.Image.mask_ideal_ring(128, 128, 0.7, 0.5,
nodc=True)
assert im.width == 128
assert im.height == 128
assert im.bands == 1
assert im.format == pyvips.BandFormat.FLOAT
p = im(45, 0)
assert p[0] == pytest.approx(1, abs=0.001)
def test_sines(self):
im = pyvips.Image.sines(128, 128)
assert im.width == 128
assert im.height == 128
assert im.bands == 1
assert im.format == pyvips.BandFormat.FLOAT
@pytest.mark.skipif(pyvips.type_find("VipsOperation", "text") == 0,
reason="no text, skipping test")
def test_text(self):
im = pyvips.Image.text("Hello, world!")
assert im.width > 10
assert im.height > 10
assert im.bands == 1
assert im.format == pyvips.BandFormat.UCHAR
assert im.max() == 255
assert im.min() == 0
# test autofit
im = pyvips.Image.text("Hello, world!", width=500, height=500)
# quite a large threshold, since we need to work with a huge range of
# text rendering systems
assert abs(im.width - 500) < 50
def test_tonelut(self):
im = pyvips.Image.tonelut()
assert im.bands == 1
assert im.format == pyvips.BandFormat.USHORT
assert im.width == 32768
assert im.height == 1
assert im.hist_ismonotonic()
def test_xyz(self):
im = pyvips.Image.xyz(128, 128)
assert im.bands == 2
assert im.format == pyvips.BandFormat.UINT
assert im.width == 128
assert im.height == 128
p = im(45, 35)
assert_almost_equal_objects(p, [45, 35])
def test_zone(self):
im = pyvips.Image.zone(128, 128)
assert im.width == 128
assert im.height == 128
assert im.bands == 1
assert im.format == pyvips.BandFormat.FLOAT
@pytest.mark.skipif(pyvips.type_find("VipsOperation", "worley") == 0,
reason="no worley, skipping test")
def test_worley(self):
im = pyvips.Image.worley(512, 512)
assert im.width == 512
assert im.height == 512
assert im.bands == 1
assert im.format == pyvips.BandFormat.FLOAT
@pytest.mark.skipif(pyvips.type_find("VipsOperation", "perlin") == 0,
reason="no perlin, skipping test")
def test_perlin(self):
im = pyvips.Image.perlin(512, 512)
assert im.width == 512
assert im.height == 512
assert im.bands == 1
assert im.format == pyvips.BandFormat.FLOAT
if __name__ == '__main__':
pytest.main()