from helper import unittest, PillowTestCase from PIL import Image class TestImageGetBands(PillowTestCase): def test_getbands(self): self.assertEqual(Image.new("1", (1, 1)).getbands(), ("1",)) self.assertEqual(Image.new("L", (1, 1)).getbands(), ("L",)) self.assertEqual(Image.new("I", (1, 1)).getbands(), ("I",)) self.assertEqual(Image.new("F", (1, 1)).getbands(), ("F",)) self.assertEqual(Image.new("P", (1, 1)).getbands(), ("P",)) self.assertEqual(Image.new("RGB", (1, 1)).getbands(), ("R", "G", "B")) self.assertEqual( Image.new("RGBA", (1, 1)).getbands(), ("R", "G", "B", "A")) self.assertEqual( Image.new("CMYK", (1, 1)).getbands(), ("C", "M", "Y", "K")) self.assertEqual( Image.new("YCbCr", (1, 1)).getbands(), ("Y", "Cb", "Cr")) if __name__ == '__main__': unittest.main() # End of file