from PIL import Image

from .helper import hopper


def test_palette():
    def palette(mode):
        p = hopper(mode).getpalette()
        if p:
            return p[:10]
        return None

    assert palette("1") is None
    assert palette("L") is None
    assert palette("I") is None
    assert palette("F") is None
    assert palette("P") == [0, 0, 0, 0, 0, 0, 0, 0, 0, 0]
    assert palette("RGB") is None
    assert palette("RGBA") is None
    assert palette("CMYK") is None
    assert palette("YCbCr") is None


def test_palette_rawmode():
    im = Image.new("P", (1, 1))
    im.putpalette((1, 2, 3))

    for rawmode in ("RGB", None):
        rgb = im.getpalette(rawmode)
        assert rgb == [1, 2, 3]

    # Convert the RGB palette to RGBA
    rgba = im.getpalette("RGBA")
    assert rgba == [1, 2, 3, 255]

    im.putpalette((1, 2, 3, 4), "RGBA")

    # Convert the RGBA palette to RGB
    rgb = im.getpalette("RGB")
    assert rgb == [1, 2, 3]

    for rawmode in ("RGBA", None):
        rgba = im.getpalette(rawmode)
        assert rgba == [1, 2, 3, 4]