1import tempfile 2import unittest 3 4import PIL.Image 5 6import pillowfight 7 8 9class TestGrayFilter(unittest.TestCase): 10 def test_grayfilter(self): 11 with tempfile.NamedTemporaryFile(suffix='.jpg') as tmpfile: 12 in_img = PIL.Image.open("tests/data/black_border_problem.jpg") 13 out_img = pillowfight.unpaper_grayfilter(in_img) 14 in_img.close() 15 16 # beware of JPG compression 17 self.assertEqual(out_img.mode, "RGB") 18 out_img.save(tmpfile.name) 19 out_img.close() 20 out_img = PIL.Image.open(tmpfile.name) 21 22 expected_img = PIL.Image.open( 23 "tests/data/black_border_problem_grayfilter.jpg" 24 ) 25 self.assertEqual(out_img.tobytes(), expected_img.tobytes()) 26 expected_img.close() 27