1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
import tempfile
import unittest
import PIL.Image
import pillowfight
class TestAll(unittest.TestCase):
def test_all_1(self):
with tempfile.NamedTemporaryFile(suffix='.png') as tmpfile:
in_img = PIL.Image.open("tests/data/black_border_problem.png")
out_img = pillowfight.ace(in_img, seed=0xDEADBEE)
# unpaper order
out_img = pillowfight.unpaper_blackfilter(out_img)
out_img = pillowfight.unpaper_noisefilter(out_img)
out_img = pillowfight.unpaper_blurfilter(out_img)
out_img = pillowfight.unpaper_masks(out_img)
out_img = pillowfight.unpaper_grayfilter(out_img)
out_img = pillowfight.unpaper_border(out_img)
in_img.close()
# beware of JPG compression
self.assertEqual(out_img.mode, "RGB")
out_img.save(tmpfile.name)
out_img.close()
out_img = PIL.Image.open(tmpfile.name)
expected_img = PIL.Image.open(
"tests/data/black_border_problem_all.png"
)
self.assertEqual(out_img.tobytes(), expected_img.tobytes())
expected_img.close()
def test_all_2(self):
with tempfile.NamedTemporaryFile(suffix='.png') as tmpfile:
in_img = PIL.Image.open("tests/data/brightness_problem.png")
out_img = pillowfight.ace(in_img, seed=0xBEEDEAD)
# unpaper order
out_img = pillowfight.unpaper_blackfilter(out_img)
out_img = pillowfight.unpaper_noisefilter(out_img)
out_img = pillowfight.unpaper_blurfilter(out_img)
out_img = pillowfight.unpaper_masks(out_img)
out_img = pillowfight.unpaper_grayfilter(out_img)
out_img = pillowfight.unpaper_border(out_img)
in_img.close()
# beware of JPG compression
self.assertEqual(out_img.mode, "RGB")
out_img.save(tmpfile.name)
out_img.close()
out_img = PIL.Image.open(tmpfile.name)
expected_img = PIL.Image.open(
"tests/data/brightness_problem_all.png"
)
self.assertEqual(out_img.tobytes(), expected_img.tobytes())
expected_img.close()
|
