test_saveload.py

# vim: set fileencoding=utf-8 :

import os
import tempfile

import pyvips
from pathlib import Path
from helpers import temp_filename, skip_if_no, IMAGES, JPEG_FILE


class TestSaveLoad:
    @classmethod
    def setup_class(cls):
        cls.tempdir = tempfile.mkdtemp()

    @skip_if_no('jpegload')
    def test_save_file(self):
        filename = temp_filename(self.tempdir, '.jpg')

        im = pyvips.Image.black(10, 20)
        im.write_to_file(filename)
        assert os.path.isfile(filename)

        os.remove(filename)

    @skip_if_no('jpegload')
    def test_load_file(self):
        filename = temp_filename(self.tempdir, '.jpg')

        im = pyvips.Image.black(10, 20)
        im.write_to_file(filename)

        x = pyvips.Image.new_from_file(filename)
        assert x.width == 10
        assert x.height == 20
        assert x.bands == 1

        os.remove(x.filename)

    @skip_if_no('jpegload')
    def test_save_file_pathlib(self):
        filename = Path(temp_filename(self.tempdir, '.jpg'))

        im = pyvips.Image.black(10, 20)
        im.write_to_file(filename)
        assert filename.exists()
        filename.unlink()

    @skip_if_no('jpegload')
    def test_load_file_pathlib(self):
        filename = Path(IMAGES) / 'sample.jpg'
        assert filename.exists()

        im_a = pyvips.Image.new_from_file(JPEG_FILE)
        im_b = pyvips.Image.new_from_file(filename)

        assert im_a.bands == im_b.bands
        assert im_a.width == im_b.width
        assert im_a.height == im_b.height

    @skip_if_no('jpegload')
    def test_save_buffer(self):
        im = pyvips.Image.black(10, 20)
        buf = im.write_to_buffer('.jpg')
        assert len(buf) > 100

    @skip_if_no('jpegload')
    def test_load_buffer(self):
        im = pyvips.Image.black(10, 20)
        buf = im.write_to_buffer('.jpg')

        x = pyvips.Image.new_from_buffer(buf, '')
        assert x.width == 10
        assert x.height == 20
        assert x.bands == 1