41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
|
import numpy as np
|
||
|
|
|
||
|
|
from damask import Colormap
|
||
|
|
|
||
|
|
class TestColormap:
|
||
|
|
|
||
|
|
def test_conversion(self):
|
||
|
|
|
||
|
|
specials = np.array([[0.,0.,0.],
|
||
|
|
[1.,0.,0.],
|
||
|
|
[0.,1.,0.],
|
||
|
|
[0.,0.,1.],
|
||
|
|
[1.,1.,0.],
|
||
|
|
[0.,1.,1.],
|
||
|
|
[1.,0.,1.],
|
||
|
|
[1.,1.,1.]
|
||
|
|
])
|
||
|
|
rgbs = np.vstack((specials,np.random.rand(10000,3)))
|
||
|
|
pass # class not integrated
|
||
|
|
for rgb in rgbs:
|
||
|
|
print('rgb',rgb)
|
||
|
|
|
||
|
|
# rgb2hsv2rgb
|
||
|
|
assert np.allclose(Colormap._hsv2rgb(Colormap._rgb2hsv(rgb)),rgb)
|
||
|
|
|
||
|
|
# rgb2hsl2rgb
|
||
|
|
assert np.allclose(Colormap._hsl2rgb(Colormap._rgb2hsl(rgb)),rgb)
|
||
|
|
|
||
|
|
# rgb2xyz2rgb
|
||
|
|
xyz = Colormap._rgb2xyz(rgb)
|
||
|
|
print('xyz',xyz)
|
||
|
|
assert np.allclose(Colormap._xyz2rgb(xyz),rgb,atol=1.e-6,rtol=0)
|
||
|
|
|
||
|
|
# xyz2lab2xyz
|
||
|
|
lab = Colormap._xyz2lab(xyz)
|
||
|
|
print('lab',lab)
|
||
|
|
assert np.allclose(Colormap._lab2xyz(lab),xyz)
|
||
|
|
|
||
|
|
# lab2msh2lab
|
||
|
|
assert np.allclose(Colormap._msh2lab(Colormap._lab2msh(lab)),lab)
|