Pillow/Tests/test_file_tar.py

36 lines
1.0 KiB
Python
Raw Normal View History

from PIL import Image, TarIO
from .helper import PillowTestCase
codecs = dir(Image.core)
2014-09-04 13:21:19 +04:00
# Sample tar archive
TEST_TAR_FILE = "Tests/images/hopper.tar"
2014-06-10 13:10:47 +04:00
class TestFileTar(PillowTestCase):
def setUp(self):
if "zip_decoder" not in codecs and "jpeg_decoder" not in codecs:
2018-08-17 12:40:13 +03:00
self.skipTest("neither jpeg nor zip support available")
2014-06-10 13:10:47 +04:00
def test_sanity(self):
for codec, test_path, format in [
2019-06-13 18:54:11 +03:00
["zip_decoder", "hopper.png", "PNG"],
["jpeg_decoder", "hopper.jpg", "JPEG"],
]:
if codec in codecs:
tar = TarIO.TarIO(TEST_TAR_FILE, test_path)
im = Image.open(tar)
im.load()
self.assertEqual(im.mode, "RGB")
self.assertEqual(im.size, (128, 128))
self.assertEqual(im.format, format)
def test_close(self):
2019-06-13 18:54:11 +03:00
tar = TarIO.TarIO(TEST_TAR_FILE, "hopper.jpg")
tar.close()
def test_contextmanager(self):
2019-06-13 18:54:11 +03:00
with TarIO.TarIO(TEST_TAR_FILE, "hopper.jpg"):
pass