test the last couple of overloads
they are actually defined in terms of other overloads, but test them anyway
This commit is contained in:
parent
d0f7efcd1e
commit
d14a6ea2c7
@ -139,27 +139,17 @@ class TestArithmetic(unittest.TestCase):
|
|||||||
def div(x, y):
|
def div(x, y):
|
||||||
return x / y
|
return x / y
|
||||||
|
|
||||||
# div(const / image) needs (image ** -1), which won't work for complex
|
# (const / image) needs (image ** -1), which won't work for complex
|
||||||
# images ... just test with non-complex
|
|
||||||
self.run_arith_const(div, fmt = noncomplex_formats)
|
self.run_arith_const(div, fmt = noncomplex_formats)
|
||||||
self.run_arith(div)
|
self.run_arith(div)
|
||||||
|
|
||||||
# run a function on an image,
|
def test_floordiv(self):
|
||||||
# 50,50 and 10,10 should have different values on the test image
|
def my_floordiv(x, y):
|
||||||
def run_testunary(self, message, im, fn):
|
return x // y
|
||||||
self.run_cmp(message, im, 50, 50, lambda x: run_fn(fn, x))
|
|
||||||
self.run_cmp(message, im, 10, 10, lambda x: run_fn(fn, x))
|
|
||||||
|
|
||||||
def run_unary(self, images, fn, fmt = all_formats):
|
# (const // image) needs (image ** -1), which won't work for complex
|
||||||
[self.run_testunary(fn.func_name + ' image', x.cast(y), fn)
|
self.run_arith_const(my_floordiv, fmt = noncomplex_formats)
|
||||||
for x in images for y in fmt]
|
self.run_arith(my_floordiv, fmt = noncomplex_formats)
|
||||||
|
|
||||||
def test_abs(self):
|
|
||||||
def my_abs(x):
|
|
||||||
return abs(x)
|
|
||||||
|
|
||||||
im = -self.colour
|
|
||||||
self.run_unary([im], my_abs, fmt = all_formats)
|
|
||||||
|
|
||||||
def test_pow(self):
|
def test_pow(self):
|
||||||
def my_pow(x, y):
|
def my_pow(x, y):
|
||||||
@ -169,26 +159,6 @@ class TestArithmetic(unittest.TestCase):
|
|||||||
self.run_arith_const(my_pow, fmt = noncomplex_formats)
|
self.run_arith_const(my_pow, fmt = noncomplex_formats)
|
||||||
self.run_arith(my_pow, fmt = noncomplex_formats)
|
self.run_arith(my_pow, fmt = noncomplex_formats)
|
||||||
|
|
||||||
def test_lshift(self):
|
|
||||||
def my_lshift(x):
|
|
||||||
# python doesn't allow float << int
|
|
||||||
if isinstance(x, float):
|
|
||||||
x = int(x)
|
|
||||||
return x << 2
|
|
||||||
|
|
||||||
# we don't support constant << image, treat as a unary
|
|
||||||
self.run_unary(self.all_images, my_lshift, fmt = noncomplex_formats)
|
|
||||||
|
|
||||||
def test_rshift(self):
|
|
||||||
def my_rshift(x):
|
|
||||||
# python doesn't allow float >> int
|
|
||||||
if isinstance(x, float):
|
|
||||||
x = int(x)
|
|
||||||
return x >> 2
|
|
||||||
|
|
||||||
# we don't support constant >> image, treat as a unary
|
|
||||||
self.run_unary(self.all_images, my_rshift, fmt = noncomplex_formats)
|
|
||||||
|
|
||||||
def test_and(self):
|
def test_and(self):
|
||||||
def my_and(x, y):
|
def my_and(x, y):
|
||||||
# python doesn't allow bools on float
|
# python doesn't allow bools on float
|
||||||
@ -225,6 +195,73 @@ class TestArithmetic(unittest.TestCase):
|
|||||||
self.run_arith_const(my_xor, fmt = noncomplex_formats)
|
self.run_arith_const(my_xor, fmt = noncomplex_formats)
|
||||||
self.run_arith(my_xor, fmt = noncomplex_formats)
|
self.run_arith(my_xor, fmt = noncomplex_formats)
|
||||||
|
|
||||||
|
# run a function on an image,
|
||||||
|
# 50,50 and 10,10 should have different values on the test image
|
||||||
|
def run_testunary(self, message, im, fn):
|
||||||
|
self.run_cmp(message, im, 50, 50, lambda x: run_fn(fn, x))
|
||||||
|
self.run_cmp(message, im, 10, 10, lambda x: run_fn(fn, x))
|
||||||
|
|
||||||
|
def run_unary(self, images, fn, fmt = all_formats):
|
||||||
|
[self.run_testunary(fn.func_name + ' image', x.cast(y), fn)
|
||||||
|
for x in images for y in fmt]
|
||||||
|
|
||||||
|
def test_abs(self):
|
||||||
|
def my_abs(x):
|
||||||
|
return abs(x)
|
||||||
|
|
||||||
|
im = -self.colour
|
||||||
|
self.run_unary([im], my_abs)
|
||||||
|
|
||||||
|
def test_lshift(self):
|
||||||
|
def my_lshift(x):
|
||||||
|
# python doesn't allow float << int
|
||||||
|
if isinstance(x, float):
|
||||||
|
x = int(x)
|
||||||
|
return x << 2
|
||||||
|
|
||||||
|
# we don't support constant << image, treat as a unary
|
||||||
|
self.run_unary(self.all_images, my_lshift, fmt = noncomplex_formats)
|
||||||
|
|
||||||
|
def test_rshift(self):
|
||||||
|
def my_rshift(x):
|
||||||
|
# python doesn't allow float >> int
|
||||||
|
if isinstance(x, float):
|
||||||
|
x = int(x)
|
||||||
|
return x >> 2
|
||||||
|
|
||||||
|
# we don't support constant >> image, treat as a unary
|
||||||
|
self.run_unary(self.all_images, my_rshift, fmt = noncomplex_formats)
|
||||||
|
|
||||||
|
def test_mod(self):
|
||||||
|
def my_mod(x):
|
||||||
|
return x % 2
|
||||||
|
|
||||||
|
# we don't support constant % image, treat as a unary
|
||||||
|
self.run_unary(self.all_images, my_mod, fmt = noncomplex_formats)
|
||||||
|
|
||||||
|
def test_pos(self):
|
||||||
|
def my_pos(x):
|
||||||
|
return +x
|
||||||
|
|
||||||
|
self.run_unary(self.all_images, my_pos)
|
||||||
|
|
||||||
|
def test_neg(self):
|
||||||
|
def my_neg(x):
|
||||||
|
return -x
|
||||||
|
|
||||||
|
self.run_unary(self.all_images, my_neg)
|
||||||
|
|
||||||
|
def test_invert(self):
|
||||||
|
def my_invert(x):
|
||||||
|
if isinstance(x, float):
|
||||||
|
x = int(x)
|
||||||
|
return ~x & 0xff
|
||||||
|
|
||||||
|
# ~image is trimmed to image max so it's hard to test for all formats
|
||||||
|
# just test uchar
|
||||||
|
self.run_unary(self.all_images, my_invert,
|
||||||
|
fmt = [Vips.BandFormat.UCHAR])
|
||||||
|
|
||||||
# test the rest of VipsArithmetic
|
# test the rest of VipsArithmetic
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
|
Loading…
Reference in New Issue
Block a user