diff --git a/Tests/test_image_getdominantcolors.py b/Tests/test_image_getdominantcolors.py new file mode 100644 index 000000000..c6c734bb7 --- /dev/null +++ b/Tests/test_image_getdominantcolors.py @@ -0,0 +1,22 @@ +from .helper import hopper + + +def test_getdominantcolors(): + def getdominantcolors(mode, numcolors=None): + im = hopper(mode) + + if numcolors: + colors = im.getdominantcolors(numcolors) + else: + colors = im.getdominantcolors() + return len(colors) + + assert getdominantcolors("F") == 3 + assert getdominantcolors("I") == 3 + assert getdominantcolors("L") == 3 + assert getdominantcolors("P") == 3 + assert getdominantcolors("RGB") == 3 + assert getdominantcolors("YCbCr") == 3 + assert getdominantcolors("CMYK") == 3 + assert getdominantcolors("RGBA") == 3 + assert getdominantcolors("HSV") == 3