python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2025-03-03 11:35:52 +03:00
Code Issues Packages Projects Releases Wiki Activity

Refactor and add two more tests

Browse Source
This commit is contained in:
hugovk 2014-07-01 22:53:30 +03:00
parent 775307113b
commit b6f836b5b8
1 changed files with 24 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
Tests/test_file_palm.py
View File

@ -5,10 +5,10 @@ import os.path
class TestFilePalm(PillowTestCase): class TestFilePalm(PillowTestCase):
def test_save_palm_p(self): def helper_save_as_palm(self, mode):
# Arrange # Arrange
im = lena("P") im = lena(mode)
outfile = self.tempfile('temp_p.palm') outfile = self.tempfile("temp_" + mode + ".palm")
# Act # Act
im.save(outfile) im.save(outfile)
@ -17,6 +17,27 @@ class TestFilePalm(PillowTestCase):
self.assertTrue(os.path.isfile(outfile)) self.assertTrue(os.path.isfile(outfile))
self.assertGreater(os.path.getsize(outfile), 0) self.assertGreater(os.path.getsize(outfile), 0)
def test_monochrome(self):
# Arrange
mode = "1"
# Act / Assert
self.helper_save_as_palm(mode)
def test_p_mode(self):
# Arrange
mode = "P"
# Act / Assert
self.helper_save_as_palm(mode)
def test_rgb_ioerror(self):
# Arrange
mode = "RGB"
# Act / Assert
self.assertRaises(IOError, lambda: self.helper_save_as_palm(mode))
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()