Pillow/Tests/test_image_putpalette.py

import pytest
from PIL import ImagePalette

from .helper import hopper


def test_putpalette():
    def palette(mode):
        im = hopper(mode).copy()
        im.putpalette(list(range(256)) * 3)
        p = im.getpalette()
        if p:
            return im.mode, p[:10]
        return im.mode

    with pytest.raises(ValueError):
        palette("1")
    for mode in ["L", "LA", "P", "PA"]:
        assert palette(mode) == (
            "PA" if "A" in mode else "P",
            [0, 1, 2, 3, 4, 5, 6, 7, 8, 9],
        )
    with pytest.raises(ValueError):
        palette("I")
    with pytest.raises(ValueError):
        palette("F")
    with pytest.raises(ValueError):
        palette("RGB")
    with pytest.raises(ValueError):
        palette("RGBA")
    with pytest.raises(ValueError):
        palette("YCbCr")


def test_imagepalette():
    im = hopper("P")
    im.putpalette(ImagePalette.negative())
    im.putpalette(ImagePalette.random())
    im.putpalette(ImagePalette.sepia())
    im.putpalette(ImagePalette.wedge())
Convert some tests to pytest style To better follow conventional pytest style, this removes the outer wrapper class in favor of a function for some tests. These tests were picked as they are relatively simple and presented no barriers to a quick port. The assert* methods are replaced with assert statements. When necessary, a fixture is used to create a temporary directory. This commit does not convert the entire test suite to this style as some test classes use methods or other advanced features that are difficult to automatically convert. The goal is to address these issues in followup commits. Refs #4193 2019-11-04 01:18:55 +03:00			`import pytest`
py3k: Import Christoph Gohlke's test suite This is Christoph Gohlke's test suite from his personal PIL package found at http://www.lfd.uci.edu/~gohlke/pythonlibs/. This is just to bring it in as a separate commit. Future commits will align it with Pillow. 2012-10-16 00:26:38 +04:00			`from PIL import ImagePalette`

Convert some tests to pytest style To better follow conventional pytest style, this removes the outer wrapper class in favor of a function for some tests. These tests were picked as they are relatively simple and presented no barriers to a quick port. The assert* methods are replaced with assert statements. When necessary, a fixture is used to create a temporary directory. This commit does not convert the entire test suite to this style as some test classes use methods or other advanced features that are difficult to automatically convert. The goal is to address these issues in followup commits. Refs #4193 2019-11-04 01:18:55 +03:00			`from .helper import hopper`
Introduce isort to automate import ordering and formatting Similar to the recent adoption of Black. isort is a Python utility to sort imports alphabetically and automatically separate into sections. By using isort, contributors can quickly and automatically conform to the projects style without thinking. Just let the tool do it. Uses the configuration recommended by the Black to avoid conflicts of style. Rewrite TestImageQt.test_deprecated to no rely on import order. 2019-07-06 23:40:53 +03:00
Convert old tests to use unittest 2014-06-10 13:10:47 +04:00
Convert some tests to pytest style To better follow conventional pytest style, this removes the outer wrapper class in favor of a function for some tests. These tests were picked as they are relatively simple and presented no barriers to a quick port. The assert* methods are replaced with assert statements. When necessary, a fixture is used to create a temporary directory. This commit does not convert the entire test suite to this style as some test classes use methods or other advanced features that are difficult to automatically convert. The goal is to address these issues in followup commits. Refs #4193 2019-11-04 01:18:55 +03:00			`def test_putpalette():`
			`def palette(mode):`
			`im = hopper(mode).copy()`
			`im.putpalette(list(range(256)) * 3)`
			`p = im.getpalette()`
			`if p:`
			`return im.mode, p[:10]`
			`return im.mode`
Format with Black 2019-06-13 18:54:24 +03:00
Convert some tests to pytest style To better follow conventional pytest style, this removes the outer wrapper class in favor of a function for some tests. These tests were picked as they are relatively simple and presented no barriers to a quick port. The assert* methods are replaced with assert statements. When necessary, a fixture is used to create a temporary directory. This commit does not convert the entire test suite to this style as some test classes use methods or other advanced features that are difficult to automatically convert. The goal is to address these issues in followup commits. Refs #4193 2019-11-04 01:18:55 +03:00			`with pytest.raises(ValueError):`
			`palette("1")`
			`for mode in ["L", "LA", "P", "PA"]:`
			`assert palette(mode) == (`
			`"PA" if "A" in mode else "P",`
			`[0, 1, 2, 3, 4, 5, 6, 7, 8, 9],`
			`)`
			`with pytest.raises(ValueError):`
			`palette("I")`
			`with pytest.raises(ValueError):`
			`palette("F")`
			`with pytest.raises(ValueError):`
			`palette("RGB")`
			`with pytest.raises(ValueError):`
			`palette("RGBA")`
			`with pytest.raises(ValueError):`
			`palette("YCbCr")`
Convert old tests to use unittest 2014-06-10 13:10:47 +04:00
Convert some tests to pytest style To better follow conventional pytest style, this removes the outer wrapper class in favor of a function for some tests. These tests were picked as they are relatively simple and presented no barriers to a quick port. The assert* methods are replaced with assert statements. When necessary, a fixture is used to create a temporary directory. This commit does not convert the entire test suite to this style as some test classes use methods or other advanced features that are difficult to automatically convert. The goal is to address these issues in followup commits. Refs #4193 2019-11-04 01:18:55 +03:00
			`def test_imagepalette():`
			`im = hopper("P")`
			`im.putpalette(ImagePalette.negative())`
			`im.putpalette(ImagePalette.random())`
			`im.putpalette(ImagePalette.sepia())`
			`im.putpalette(ImagePalette.wedge())`