Pillow/Tests/test_tiff_ifdrational.py

59 lines
1.7 KiB
Python
Raw Normal View History

from .helper import PillowTestCase, hopper
from PIL import TiffImagePlugin, Image
from PIL.TiffImagePlugin import IFDRational
from fractions import Fraction
2016-02-05 01:57:13 +03:00
class Test_IFDRational(PillowTestCase):
def _test_equal(self, num, denom, target):
2016-01-24 04:44:31 +03:00
t = IFDRational(num, denom)
self.assertEqual(target, t)
self.assertEqual(t, target)
2016-01-24 04:44:31 +03:00
def test_sanity(self):
2016-01-24 04:44:31 +03:00
self._test_equal(1, 1, 1)
2016-02-05 01:57:13 +03:00
self._test_equal(1, 1, Fraction(1, 1))
self._test_equal(2, 2, 1)
2016-02-05 01:57:13 +03:00
self._test_equal(1.0, 1, Fraction(1, 1))
2016-01-24 04:44:31 +03:00
2016-02-05 01:57:13 +03:00
self._test_equal(Fraction(1, 1), 1, Fraction(1, 1))
self._test_equal(IFDRational(1, 1), 1, 1)
2016-01-24 04:44:31 +03:00
2016-02-05 01:57:13 +03:00
self._test_equal(1, 2, Fraction(1, 2))
self._test_equal(1, 2, IFDRational(1, 2))
def test_nonetype(self):
2019-02-03 07:59:24 +03:00
# Fails if the _delegate function doesn't return a valid function
2016-01-24 04:44:31 +03:00
xres = IFDRational(72)
yres = IFDRational(72)
self.assertIsNotNone(xres._val)
self.assertIsNotNone(xres.numerator)
self.assertIsNotNone(xres.denominator)
self.assertIsNotNone(yres._val)
2016-01-24 04:44:31 +03:00
self.assertTrue(xres and 1)
self.assertTrue(xres and yres)
def test_ifd_rational_save(self):
methods = (True, False)
2019-06-13 18:54:46 +03:00
if "libtiff_encoder" not in dir(Image.core):
methods = (False,)
2016-08-04 09:40:12 +03:00
for libtiff in methods:
TiffImagePlugin.WRITE_LIBTIFF = libtiff
2016-01-24 04:44:31 +03:00
im = hopper()
2019-06-13 18:54:46 +03:00
out = self.tempfile("temp.tiff")
2016-02-05 01:57:13 +03:00
res = IFDRational(301, 1)
2019-06-13 18:54:46 +03:00
im.save(out, dpi=(res, res), compression="raw")
reloaded = Image.open(out)
2019-06-13 18:54:46 +03:00
self.assertEqual(float(IFDRational(301, 1)), float(reloaded.tag_v2[282]))