from helper import unittest, PillowTestCase, tearDownModule, lena

from PIL import Image
import io


class TestFileBmp(PillowTestCase):

    def roundtrip(self, im):
        outfile = self.tempfile("temp.bmp")

        im.save(outfile, 'BMP')

        reloaded = Image.open(outfile)
        reloaded.load()
        self.assertEqual(im.mode, reloaded.mode)
        self.assertEqual(im.size, reloaded.size)
        self.assertEqual(reloaded.format, "BMP")

    def test_sanity(self):
        self.roundtrip(lena())

        self.roundtrip(lena("1"))
        self.roundtrip(lena("L"))
        self.roundtrip(lena("P"))
        self.roundtrip(lena("RGB"))

    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
