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)