from helper import unittest, PillowTestCase, hopper
from PIL import Image


class TestImageRotate(PillowTestCase):

    def test_rotate(self):
        def rotate(im, mode, angle):
            out = im.rotate(angle)
            self.assertEqual(out.mode, mode)
            self.assertEqual(out.size, im.size)  # default rotate clips output
            out = im.rotate(angle, expand=1)
            self.assertEqual(out.mode, mode)
            if angle % 180 == 0:
                self.assertEqual(out.size, im.size)
            elif im.size == (0, 0):
                self.assertEqual(out.size, im.size)
            else:
                self.assertNotEqual(out.size, im.size)

                
        for mode in ("1", "P", "L", "RGB", "I", "F"):
            im = hopper(mode)
            rotate(im, mode, 45)

        for angle in (0, 90, 180, 270):
            im = Image.open('Tests/images/test-card.png')
            rotate(im, im.mode, angle)

        for angle in (0, 45, 90, 180, 270):
            im = Image.new('RGB',(0,0))
            rotate(im, im.mode, angle)


if __name__ == '__main__':
    unittest.main()