85 lines
2.7 KiB
Python
Executable File
85 lines
2.7 KiB
Python
Executable File
#!/usr/bin/python
|
|
|
|
import unittest
|
|
import math
|
|
|
|
#import logging
|
|
#logging.basicConfig(level = logging.DEBUG)
|
|
|
|
from gi.repository import Vips
|
|
|
|
Vips.leak_set(True)
|
|
|
|
# an expanding zip ... if either of the args is a scalar or a one-element list,
|
|
# duplicate it down the other side
|
|
def zip_expand(x, y):
|
|
# handle singleton list case
|
|
if isinstance(x, list) and len(x) == 1:
|
|
x = x[0]
|
|
if isinstance(y, list) and len(y) == 1:
|
|
y = y[0]
|
|
|
|
if isinstance(x, list) and isinstance(y, list):
|
|
return list(zip(x, y))
|
|
elif isinstance(x, list):
|
|
return [[i, y] for i in x]
|
|
elif isinstance(y, list):
|
|
return [[x, j] for j in y]
|
|
else:
|
|
return [[x, y]]
|
|
|
|
class TestMorphology(unittest.TestCase):
|
|
# test a pair of things which can be lists for approx. equality
|
|
def assertAlmostEqualObjects(self, a, b, places = 4, msg = ''):
|
|
# print 'assertAlmostEqualObjects %s = %s' % (a, b)
|
|
for x, y in zip_expand(a, b):
|
|
self.assertAlmostEqual(x, y, places = places, msg = msg)
|
|
|
|
def test_countlines(self):
|
|
im = Vips.Image.black(100, 100)
|
|
im = im.draw_line(255, 0, 50, 100, 50)
|
|
n_lines = im.countlines(Vips.Direction.HORIZONTAL)
|
|
self.assertEqual(n_lines, 1)
|
|
|
|
def test_labelregions(self):
|
|
im = Vips.Image.black(100, 100)
|
|
im = im.draw_circle(255, 50, 50, 25, fill = True)
|
|
mask, opts = im.labelregions(segments = True)
|
|
|
|
self.assertEqual(opts['segments'], 3)
|
|
self.assertEqual(mask.max(), 2)
|
|
|
|
def test_erode(self):
|
|
im = Vips.Image.black(100, 100)
|
|
im = im.draw_circle(255, 50, 50, 25, fill = True)
|
|
im2 = im.erode([[128, 255, 128],
|
|
[255, 255, 255],
|
|
[128, 255, 128]])
|
|
self.assertEqual(im.width, im2.width)
|
|
self.assertEqual(im.height, im2.height)
|
|
self.assertEqual(im.bands, im2.bands)
|
|
self.assertTrue(im.avg() > im2.avg())
|
|
|
|
def test_dilate(self):
|
|
im = Vips.Image.black(100, 100)
|
|
im = im.draw_circle(255, 50, 50, 25, fill = True)
|
|
im2 = im.dilate([[128, 255, 128],
|
|
[255, 255, 255],
|
|
[128, 255, 128]])
|
|
self.assertEqual(im.width, im2.width)
|
|
self.assertEqual(im.height, im2.height)
|
|
self.assertEqual(im.bands, im2.bands)
|
|
self.assertTrue(im2.avg() > im.avg())
|
|
|
|
def test_rank(self):
|
|
im = Vips.Image.black(100, 100)
|
|
im = im.draw_circle(255, 50, 50, 25, fill = True)
|
|
im2 = im.rank(3, 3, 8)
|
|
self.assertEqual(im.width, im2.width)
|
|
self.assertEqual(im.height, im2.height)
|
|
self.assertEqual(im.bands, im2.bands)
|
|
self.assertTrue(im2.avg() > im.avg())
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|