def test_merge_layer_from_file():
with TempFile('.xcf') as dst, TempFile('.xcf') as src:
layer_bg = np.array([
[[255, 255, 255], [0, 0, 0], [255, 255, 255]],
[[255, 255, 255], [255, 255, 255], [255, 255, 255]],
],
dtype=np.uint8)
src_file = GimpFile(src)
src_file.create('Background', np.zeros(shape=(2, 3, 3),
dtype=np.uint8))
src_file.add_layer_from_numpy(
'Yellow',
np.array([
[[240, 255, 0], [240, 255, 0], [240, 255, 0]],
[[240, 255, 0], [240, 255, 0], [240, 255, 0]],
],
dtype=np.uint8))
dst_file = GimpFile(dst)
dst_file.create('Yellow', layer_bg)
dst_file.merge_layer_from_file(src_file, 'Yellow')
new_layer_contents = dst_file.layer_to_numpy('Yellow')
assert np.all([240, 255, 0] == new_layer_contents)
def test_add_layer_from_file():
with TempFile('.xcf') as dst, TempFile('.xcf') as src:
layer_bg = np.array([
[[255, 255, 255], [0, 0, 0], [255, 255, 255]],
[[255, 255, 255], [255, 255, 255], [255, 255, 255]],
],
dtype=np.uint8)
position = 1
src_file = GimpFile(src)
src_file.create('Background', np.zeros(shape=(2, 3, 3),
dtype=np.uint8))
src_file.add_layer_from_numpy(
'Yellow',
np.array([
[[240, 255, 0], [240, 255, 0], [240, 255, 0]],
[[240, 255, 0], [240, 255, 0], [240, 255, 0]],
],
dtype=np.uint8))
dst_file = GimpFile(dst)
dst_file.create('Background', layer_bg)
dst_file.add_layer_from_file(src_file,
'Yellow',
new_name='Yellow (copied)',
new_position=position)
assert 'Yellow (copied)' == dst_file.layers()[position].name
assert np.all(
[240, 255, 0] == dst_file.layer_to_numpy('Yellow (copied)'))
def test_add_layer_from_numpy():
tmp_file = tempfile.mktemp(suffix='.xcf')
layer_bg = np.array([
[[255, 255, 255], [0, 0, 0], [255, 255, 255]],
[[255, 255, 255], [255, 255, 255], [255, 255, 255]],
],
dtype=np.uint8)
layer_fg = np.array([
[[255, 255, 255], [255, 255, 255], [255, 255, 255]],
[[255, 255, 255], [0, 0, 0], [255, 255, 255]],
],
dtype=np.uint8)
gimp_file = GimpFile(tmp_file)
gimp_file.create('Background', layer_bg)
gimp_file.add_layer_from_numpy('Foreground',
layer_fg,
opacity=55.,
visible=False)
actual_bg = gimp_file.layer_to_numpy('Background')
actual_fg = gimp_file.layer_to_numpy('Foreground')
os.remove(tmp_file)
assert np.all(layer_bg == actual_bg)
assert np.all(layer_fg == actual_fg)
def test_add_layer_from_numpy_with_position_layer_name():
data = np.array([[[255, 255, 255]]], dtype=np.uint8)
with TempFile('.xcf') as tmp_file:
gimp_file = GimpFile(tmp_file)
gimp_file.create('Background', data)
gimp_file.add_layer_from_numpy('Layer 1', data, position='Background')
gimp_file.add_layer_from_numpy('Layer 2', data, position='Background')
assert gimp_file.layer_names() == ['Layer 1', 'Layer 2', 'Background']
def test_create():
filename = file.relative_to(__file__, 'test-resources/test-create.xcf')
layer_bg = np.array([
[[255, 255, 255], [0, 0, 0], [255, 255, 255]],
[[255, 255, 255], [255, 255, 255], [255, 255, 255]],
],
dtype=np.uint8)
gimp_file = GimpFile(filename)
gimp_file.create('Background', layer_bg)
exists = os.path.exists(filename)
actual = gimp_file.layer_to_numpy('Background')
os.remove(filename)
assert exists
assert np.all(layer_bg == actual)
def test_merge_layer_from_file_with_cleared_selection():
src = file.relative_to(__file__, 'test-resources/selection.xcf')
with TempFile('.xcf') as dst:
src_file = GimpFile(src)
dst_file = GimpFile(dst)
dst_file.create('Background', np.zeros(shape=(3, 3, 3),
dtype=np.uint8))
dst_file.merge_layer_from_file(src_file, 'Background')
new_layer_contents = dst_file.layer_to_numpy('Background')
assert np.all(
np.array([
[[255, 255, 255], [255, 255, 255], [255, 255, 255]],
[[255, 0, 0], [255, 0, 0], [255, 255, 255]],
[[255, 255, 255], [255, 255, 255], [255, 255, 255]],
],
dtype=np.uint8) == new_layer_contents)
def test_remove_layer():
tmp_file = tempfile.mktemp(suffix='.xcf')
layer = np.array([
[[255, 255, 255], [0, 0, 0], [255, 255, 255]],
[[255, 255, 255], [255, 255, 255], [255, 255, 255]],
],
dtype=np.uint8)
gimp_file = GimpFile(tmp_file)
gimp_file.create('Background', layer)
gimp_file.add_layer_from_numpy('Layer', layer)
all_layers = gimp_file.layer_names()
gimp_file.remove_layer('Background')
remaining_layers1 = gimp_file.layer_names()
gimp_file.remove_layer('Layer')
remaining_layers2 = gimp_file.layer_names()
os.remove(tmp_file)
assert ['Layer', 'Background'] == all_layers
assert ['Layer'] == remaining_layers1
assert [] == remaining_layers2