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
65
66
67
68
69
70
71
72
73
74
75
76
| import unittest, test.test_support
| import colorsys
|
| def frange(start, stop, step):
| while start <= stop:
| yield start
| start += step
|
| class ColorsysTest(unittest.TestCase):
|
| def assertTripleEqual(self, tr1, tr2):
| self.assertEqual(len(tr1), 3)
| self.assertEqual(len(tr2), 3)
| self.assertAlmostEqual(tr1[0], tr2[0])
| self.assertAlmostEqual(tr1[1], tr2[1])
| self.assertAlmostEqual(tr1[2], tr2[2])
|
| def test_hsv_roundtrip(self):
| for r in frange(0.0, 1.0, 0.2):
| for g in frange(0.0, 1.0, 0.2):
| for b in frange(0.0, 1.0, 0.2):
| rgb = (r, g, b)
| self.assertTripleEqual(
| rgb,
| colorsys.hsv_to_rgb(*colorsys.rgb_to_hsv(*rgb))
| )
|
| def test_hsv_values(self):
| values = [
| # rgb, hsv
| ((0.0, 0.0, 0.0), ( 0 , 0.0, 0.0)), # black
| ((0.0, 0.0, 1.0), (4./6., 1.0, 1.0)), # blue
| ((0.0, 1.0, 0.0), (2./6., 1.0, 1.0)), # green
| ((0.0, 1.0, 1.0), (3./6., 1.0, 1.0)), # cyan
| ((1.0, 0.0, 0.0), ( 0 , 1.0, 1.0)), # red
| ((1.0, 0.0, 1.0), (5./6., 1.0, 1.0)), # purple
| ((1.0, 1.0, 0.0), (1./6., 1.0, 1.0)), # yellow
| ((1.0, 1.0, 1.0), ( 0 , 0.0, 1.0)), # white
| ((0.5, 0.5, 0.5), ( 0 , 0.0, 0.5)), # grey
| ]
| for (rgb, hsv) in values:
| self.assertTripleEqual(hsv, colorsys.rgb_to_hsv(*rgb))
| self.assertTripleEqual(rgb, colorsys.hsv_to_rgb(*hsv))
|
| def test_hls_roundtrip(self):
| for r in frange(0.0, 1.0, 0.2):
| for g in frange(0.0, 1.0, 0.2):
| for b in frange(0.0, 1.0, 0.2):
| rgb = (r, g, b)
| self.assertTripleEqual(
| rgb,
| colorsys.hls_to_rgb(*colorsys.rgb_to_hls(*rgb))
| )
|
| def test_hls_values(self):
| values = [
| # rgb, hls
| ((0.0, 0.0, 0.0), ( 0 , 0.0, 0.0)), # black
| ((0.0, 0.0, 1.0), (4./6., 0.5, 1.0)), # blue
| ((0.0, 1.0, 0.0), (2./6., 0.5, 1.0)), # green
| ((0.0, 1.0, 1.0), (3./6., 0.5, 1.0)), # cyan
| ((1.0, 0.0, 0.0), ( 0 , 0.5, 1.0)), # red
| ((1.0, 0.0, 1.0), (5./6., 0.5, 1.0)), # purple
| ((1.0, 1.0, 0.0), (1./6., 0.5, 1.0)), # yellow
| ((1.0, 1.0, 1.0), ( 0 , 1.0, 0.0)), # white
| ((0.5, 0.5, 0.5), ( 0 , 0.5, 0.0)), # grey
| ]
| for (rgb, hls) in values:
| self.assertTripleEqual(hls, colorsys.rgb_to_hls(*rgb))
| self.assertTripleEqual(rgb, colorsys.hls_to_rgb(*hls))
|
| def test_main():
| test.test_support.run_unittest(ColorsysTest)
|
| if __name__ == "__main__":
| test_main()
|
|
