From a9995b6a70bc0ec274a8e272504c36f14183eaa0 Mon Sep 17 00:00:00 2001 From: Stefan van der Walt Date: Fri, 11 Oct 2013 18:44:30 +0200 Subject: [PATCH] Test whether labels are left alone in label2rgb. --- skimage/color/tests/test_colorlabel.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/skimage/color/tests/test_colorlabel.py b/skimage/color/tests/test_colorlabel.py index ab1fc894..dcfbe4ea 100644 --- a/skimage/color/tests/test_colorlabel.py +++ b/skimage/color/tests/test_colorlabel.py @@ -3,7 +3,8 @@ import itertools import numpy as np from numpy import testing from skimage.color.colorlabel import label2rgb -from numpy.testing import assert_array_almost_equal as assert_close +from numpy.testing import (assert_array_almost_equal as assert_close, + assert_array_equal) def test_shape_mismatch(): @@ -80,6 +81,14 @@ def test_label_consistency(): for label_id in label_2.flat: assert_close(rgb_1[label_1 == label_id], rgb_2[label_2 == label_id]) +def test_leave_labels_alone(): + labels = np.array([-1, 0, 1]) + labels_saved = labels.copy() + + label2rgb(labels) + label2rgb(labels, bg_label=1) + assert_array_equal(labels, labels_saved) + if __name__ == '__main__': testing.run_module_suite()