from __future__ import annotations import pytest from PIL import BufrStubImagePlugin, Image from .helper import hopper TEST_FILE = "Tests/images/gfs.t06z.rassda.tm00.bufr_d" def test_open(): # Act with Image.open(TEST_FILE) as im: # Assert assert im.format == "BUFR" # Dummy data from the stub assert im.mode == "F" assert im.size == (1, 1) def test_invalid_file(): # Arrange invalid_file = "Tests/images/flower.jpg" # Act / Assert with pytest.raises(SyntaxError): BufrStubImagePlugin.BufrStubImageFile(invalid_file) def test_load(): # Arrange with Image.open(TEST_FILE) as im: # Act / Assert: stub cannot load without an implemented handler with pytest.raises(OSError): im.load() def test_save(tmp_path): # Arrange im = hopper() tmpfile = str(tmp_path / "temp.bufr") # Act / Assert: stub cannot save without an implemented handler with pytest.raises(OSError): im.save(tmpfile) def test_handler(tmp_path): class TestHandler: opened = False loaded = False saved = False def open(self, im): self.opened = True def load(self, im): self.loaded = True im.fp.close() return Image.new("RGB", (1, 1)) def save(self, im, fp, filename): self.saved = True handler = TestHandler() BufrStubImagePlugin.register_handler(handler) with Image.open(TEST_FILE) as im: assert handler.opened assert not handler.loaded im.load() assert handler.loaded temp_file = str(tmp_path / "temp.bufr") im.save(temp_file) assert handler.saved BufrStubImagePlugin._handler = None