Skip to content

Commit 3a18e55

Browse files
authored
Assert palette is not None (#8877)
2 parents 3d21c16 + 75d3f1d commit 3a18e55

File tree

3 files changed

+4
-0
lines changed

3 files changed

+4
-0
lines changed

Tests/test_file_gif.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -224,6 +224,7 @@ def test_optimize_if_palette_can_be_reduced_by_half() -> None:
224224
out = BytesIO()
225225
im.save(out, "GIF", optimize=optimize)
226226
with Image.open(out) as reloaded:
227+
assert reloaded.palette is not None
227228
assert len(reloaded.palette.palette) // 3 == colors
228229

229230

@@ -1361,6 +1362,7 @@ def test_palette_save_all_P(tmp_path: Path) -> None:
13611362
# Assert that the frames are correct, and each frame has the same palette
13621363
assert_image_equal(im.convert("RGB"), frames[0].convert("RGB"))
13631364
assert im.palette is not None
1365+
assert im.global_palette is not None
13641366
assert im.palette.palette == im.global_palette.palette
13651367

13661368
im.seek(1)

Tests/test_image_quantize.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -70,6 +70,7 @@ def test_quantize_no_dither() -> None:
7070
converted = image.quantize(dither=Image.Dither.NONE, palette=palette)
7171
assert converted.mode == "P"
7272
assert converted.palette is not None
73+
assert palette.palette is not None
7374
assert converted.palette.palette == palette.palette.palette
7475

7576

Tests/test_image_transform.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,7 @@ def test_palette(self) -> None:
4848
im.size, Image.Transform.AFFINE, [1, 0, 0, 0, 1, 0]
4949
)
5050
assert im.palette is not None
51+
assert transformed.palette is not None
5152
assert im.palette.palette == transformed.palette.palette
5253

5354
def test_extent(self) -> None:

0 commit comments

Comments
 (0)