def test_resize_grey():
"""Enlarge grey image, check new size and intensities"""
data = numpy.zeros((5, 5))
data[2, 2] = 255
output_shape = (10, 10)
parameters = {'data': [data], 'output_shape': output_shape}
resized = geometric.resize(parameters)
assert_equal(resized.shape, output_shape)
assert_equal(resized[4, 4], 1)
assert_equal(resized.sum(), 16)
def test_resize_rgb():
"""Enlarge rgb image, check new size and intensities"""
data = numpy.zeros((5, 5, 3))
data[2, 2, 1] = 255
output_shape = (10, 10)
parameters = {'data': [data], 'output_shape': output_shape}
resized = geometric.resize(parameters)
assert_equal(resized.shape, (10, 10, 3))
assert_equal(resized[4, 4, 0], 0)
assert_equal(resized[4, 4, 1], 1)
assert_equal(resized[4, 4, 2], 0)
assert_equal(resized.sum(), 16)
