From b6f836b5b8fdb5884172e87d15fa443bdd80a058 Mon Sep 17 00:00:00 2001 From: hugovk Date: Tue, 1 Jul 2014 22:53:30 +0300 Subject: [PATCH] Refactor and add two more tests --- Tests/test_file_palm.py | 27 ++++++++++++++++++++++++--- 1 file changed, 24 insertions(+), 3 deletions(-) diff --git a/Tests/test_file_palm.py b/Tests/test_file_palm.py index 62c8396b2..c9e3c8c37 100644 --- a/Tests/test_file_palm.py +++ b/Tests/test_file_palm.py @@ -5,10 +5,10 @@ import os.path class TestFilePalm(PillowTestCase): - def test_save_palm_p(self): + def helper_save_as_palm(self, mode): # Arrange - im = lena("P") - outfile = self.tempfile('temp_p.palm') + im = lena(mode) + outfile = self.tempfile("temp_" + mode + ".palm") # Act im.save(outfile) @@ -17,6 +17,27 @@ class TestFilePalm(PillowTestCase): self.assertTrue(os.path.isfile(outfile)) 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__': unittest.main()