from tester import *

from PIL import Image

def test_sanity():

    im = lena()

    im = im.quantize()
    assert_image(im, "P", im.size)

    im = lena()
    im = im.quantize(palette=lena("P"))
    assert_image(im, "P", im.size)

def test_octree_quantize():
    im = lena()

    im = im.quantize(100, Image.FASTOCTREE)
    assert_image(im, "P", im.size)

    assert len(im.getcolors()) == 100