def test_quickshift(self):
        # Test grayscaled image.
        image = mnist.test.next_batch(1, shuffle=False)[0][0]
        segmentation = quickshift(
            image, ratio=1, kernel_size=2, max_dist=2, sigma=0)