Pillow/Tests/test_file_fits.py

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

58 lines
1.4 KiB
Python
Raw Normal View History

from __future__ import annotations
2024-01-20 14:23:03 +03:00
from io import BytesIO
2020-01-27 14:46:52 +03:00
import pytest
from PIL import FitsImagePlugin, Image
2022-02-15 03:22:46 +03:00
2024-03-26 13:29:25 +03:00
from .helper import assert_image_equal, assert_image_equal_tofile, hopper
TEST_FILE = "Tests/images/hopper.fits"
2015-07-03 08:03:25 +03:00
def test_open() -> None:
2020-01-27 14:46:52 +03:00
# Act
with Image.open(TEST_FILE) as im:
# Assert
assert im.format == "FITS"
assert im.size == (128, 128)
assert im.mode == "L"
2020-01-27 14:46:52 +03:00
2022-02-15 03:22:46 +03:00
assert_image_equal(im, hopper("L"))
2024-03-26 13:29:25 +03:00
def test_gzip1() -> None:
with Image.open("Tests/images/m13_gzip.fits") as im:
assert_image_equal_tofile(im, "Tests/images/m13.fits")
def test_invalid_file() -> None:
2020-01-27 14:46:52 +03:00
# Arrange
invalid_file = "Tests/images/flower.jpg"
2020-01-27 14:46:52 +03:00
# Act / Assert
with pytest.raises(SyntaxError):
2022-02-15 03:22:46 +03:00
FitsImagePlugin.FitsImageFile(invalid_file)
def test_truncated_fits() -> None:
# No END to headers
image_data = b"SIMPLE = T" + b" " * 50 + b"TRUNCATE"
with pytest.raises(OSError):
2022-02-15 03:22:46 +03:00
FitsImagePlugin.FitsImageFile(BytesIO(image_data))
def test_naxis_zero() -> None:
# This test image has been manually hexedited
# to set the number of data axes to zero
with pytest.raises(ValueError):
with Image.open("Tests/images/hopper_naxis_zero.fits"):
pass
2022-02-19 06:41:12 +03:00
def test_comment() -> None:
2023-02-25 08:44:07 +03:00
image_data = b"SIMPLE = T / comment string"
with pytest.raises(OSError):
FitsImagePlugin.FitsImageFile(BytesIO(image_data))