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
|
import sys
import os
import contextlib
from subprocess import run, PIPE
from unittest import TestCase
from colorsys import hls_to_rgb
def cc(bin, src, *opts):
print(f"{src} -> {bin} {opts}")
with contextlib.suppress(OSError):
os.unlink(bin)
run(["cc", "-Os", *opts, "-o", bin, src])
cc("test_color_int32.bin", "test_color.c", "-DUSE_FLOAT=0", "-DUSE_64BIT_MUL=0")
cc("test_color_int64.bin", "test_color.c", "-DUSE_FLOAT=0", "-DUSE_64BIT_MUL=1")
cc("test_color_float.bin", "test_color.c", "-DUSE_FLOAT=1")
class Color_Test(TestCase):
def __runtest(self, mode):
thres = 1 if mode == "float" else 6
for h in range(0, 0xFFFF+1, 0x1000):
for s in range(0, 0xFFFF+1, 0x1000):
for l in range(0, 0xFFFF+1, 0x1000):
p = run([f"./test_color_{mode}.bin", f"{h}", f"{s}", f"{l}"],
stdout=PIPE)
res = (int(x) for x in p.stdout.decode("UTF-8").split())
exp = (round(x * 0xFFFF)
for x in hls_to_rgb(h / 0xFFFF, l / 0xFFFF, s / 0xFFFF))
if not all( abs(a - b) <= thres for a, b in zip(res, exp) ):
print("H = %X S = %X L = %X" % (h, s, l))
print("res =", list(res), ", exp =", list(exp))
assert 0, "Invalid RGB conversion."
def test_float(self):
self.__runtest(mode="float")
def test_int32(self):
self.__runtest(mode="int32")
def test_int64(self):
self.__runtest(mode="int64")
# vim: ts=4 sw=4 expandtab
|
