def test_undo_redo(
brush_shape,
brush_size,
mode,
selected_label,
preserve_labels,
n_dimensional,
):
blobs = data.binary_blobs(length=64, volume_fraction=0.3, n_dim=3)
layer = Labels(blobs)
data_history = [blobs.copy()]
with pytest.warns(FutureWarning):
layer.brush_shape = brush_shape
layer.brush_size = brush_size
layer.mode = mode
layer.selected_label = selected_label
layer.preserve_labels = preserve_labels
layer.n_edit_dimensions = 3 if n_dimensional else 2
coord = np.random.random((3, )) * (np.array(blobs.shape) - 1)
while layer.data[tuple(coord.astype(int))] == 0 and np.any(layer.data):
coord = np.random.random((3, )) * (np.array(blobs.shape) - 1)
if layer.mode == 'fill':
layer.fill(coord, layer.selected_label)
if layer.mode == 'erase':
layer.paint(coord, 0)
if layer.mode == 'paint':
layer.paint(coord, layer.selected_label)
data_history.append(np.copy(layer.data))
layer.undo()
np.testing.assert_array_equal(layer.data, data_history[0])
layer.redo()
np.testing.assert_array_equal(layer.data, data_history[1])
