Pillow/Tests/test_file_bmp.py

45 lines
1.0 KiB
Python
Raw Normal View History

2014-06-10 13:10:47 +04:00
from helper import unittest, PillowTestCase, tearDownModule, lena
from PIL import Image
2014-01-23 08:40:37 +04:00
import io
2014-06-10 13:10:47 +04:00
class TestFileBmp(PillowTestCase):
2014-06-10 13:10:47 +04:00
def roundtrip(self, im):
outfile = self.tempfile("temp.bmp")
2014-06-10 13:10:47 +04:00
im.save(outfile, 'BMP')
2014-06-10 13:10:47 +04:00
reloaded = Image.open(outfile)
reloaded.load()
self.assertEqual(im.mode, reloaded.mode)
self.assertEqual(im.size, reloaded.size)
self.assertEqual(reloaded.format, "BMP")
2014-06-10 13:10:47 +04:00
def test_sanity(self):
self.roundtrip(lena())
2014-06-10 13:10:47 +04:00
self.roundtrip(lena("1"))
self.roundtrip(lena("L"))
self.roundtrip(lena("P"))
self.roundtrip(lena("RGB"))
2014-01-23 08:40:37 +04:00
2014-06-10 13:10:47 +04:00
def test_save_to_bytes(self):
output = io.BytesIO()
im = lena()
im.save(output, "BMP")
output.seek(0)
reloaded = Image.open(output)
self.assertEqual(im.mode, reloaded.mode)
self.assertEqual(im.size, reloaded.size)
self.assertEqual(reloaded.format, "BMP")
if __name__ == '__main__':
unittest.main()
# End of file