def test_resize_truncated_image(self):
engine = Engine(self.context)
with open(join(STORAGE_PATH, 'BlueSquare_truncated.jpg'), 'r') as im:
buffer = im.read()
engine.load(buffer, '.jpg')
engine.resize(10, 10)
mode, _ = engine.image_data_as_rgb()
expect(mode).to_equal('RGB')
def test_convert_png_1bit_to_png(self):
engine = Engine(self.context)
with open(join(STORAGE_PATH, '1bit.png'), 'r') as im:
buffer = im.read()
engine.load(buffer, '.png')
engine.resize(10, 10)
mode, _ = engine.image_data_as_rgb()
expect(mode).to_equal('RGBA')
def test_convert_png_1bit_to_png(self):
engine = Engine(self.context)
with open(join(STORAGE_PATH, '1bit.png'), 'r') as im:
buffer = im.read()
engine.load(buffer, '.png')
engine.resize(10, 10)
mode, _ = engine.image_data_as_rgb()
expect(mode).to_equal('P') # Note that this is not a true 1bit image, it's 8bit in black/white.
def test_convert_should_preserve_palette_mode(self):
engine = Engine(self.context)
with open(join(STORAGE_PATH, '256_color_palette.png'), 'r') as im:
buffer = im.read()
engine.load(buffer, '.png')
engine.resize(10, 10)
mode, _ = engine.image_data_as_rgb()
expect(mode).to_equal('P')
def test_convert_should_preserve_palette_mode(self):
engine = Engine(self.context)
with open(join(STORAGE_PATH, '256_color_palette.png'), 'r') as im:
buffer = im.read()
engine.load(buffer, '.png')
engine.resize(10, 10)
mode, _ = engine.image_data_as_rgb()
expect(mode).to_equal('P')
def test_resize_truncated_image(self):
engine = Engine(self.context)
with open(join(STORAGE_PATH, "BlueSquare_truncated.jpg"),
"rb") as image_file:
buffer = image_file.read()
engine.load(buffer, ".jpg")
engine.resize(10, 10)
mode, _ = engine.image_data_as_rgb()
expect(mode).to_equal("RGB")
def test_convert_png_1bit_to_png(self):
engine = Engine(self.context)
with open(join(STORAGE_PATH, '1bit.png'), 'r') as im:
buffer = im.read()
engine.load(buffer, '.png')
engine.resize(10, 10)
mode, _ = engine.image_data_as_rgb()
expect(mode).to_equal(
'P'
) # Note that this is not a true 1bit image, it's 8bit in black/white.
def test_compare_image_data_and_mode(engine):
pil_engine = PileEngine(get_context())
with open(join(STORAGE_PATH, "1bit.png"), "rb") as image_file:
buffer = image_file.read()
assert buffer is not None
engine.load(buffer, ".png")
pil_engine.load(buffer, ".png")
pil_mode, pil_data = pil_engine.image_data_as_rgb()
mode, data = engine.image_data_as_rgb()
assert mode == pil_mode
assert len(data) == len(pil_data)
assert data[:100] == pil_data[:100]
assert data[-100:] == pil_data[-100:]
def test_convert_should_preserve_palette_mode(self):
engine = Engine(self.context)
with open(join(STORAGE_PATH, "256_color_palette.png"),
"rb") as image_file:
buffer = image_file.read()
engine.load(buffer, ".png")
expect(engine.original_mode).to_equal("P")
engine.resize(10, 10)
mode, _ = engine.image_data_as_rgb()
expect(mode).to_equal("RGB")
final_bytes = BytesIO(engine.read())
mode = Image.open(final_bytes).mode
expect(mode).to_equal("P")
def test_convert_png_1bit_to_png(self):
engine = Engine(self.context)
with open(join(STORAGE_PATH, "1bit.png"), "rb") as image_file:
buffer = image_file.read()
engine.load(buffer, ".png")
expect(engine.original_mode).to_equal(
"P"
) # Note that this is not a true 1bit image, it's 8bit in black/white.
engine.resize(10, 10)
mode, _ = engine.image_data_as_rgb()
expect(mode).to_equal("RGB")
final_bytes = BytesIO(engine.read())
mode = Image.open(final_bytes).mode
expect(mode).to_equal("P")
