def test_write_filehandle(self, format, name, tmp_path):
c = Colormap.from_predefined('Dark2') # noqa
fname = tmp_path / name
with open(fname, 'w') as f: # noqa
eval(f'c.save_{format}(f)')
for i in range(10):
if fname.exists(): return
time.sleep(.5)
assert False
def test_eq(self):
assert Colormap.from_predefined('strain') == Colormap.from_predefined(
'strain')
assert Colormap.from_predefined('strain') != Colormap.from_predefined(
'stress')
assert Colormap.from_predefined(
'strain', N=128) != Colormap.from_predefined('strain', N=64)
assert not Colormap.from_predefined('strain', N=128) == 1
def test_shade(self, ref_path, update, bounds):
data = np.add(*np.indices((10, 11)))
img_current = Colormap.from_predefined('orientation').shade(
data, bounds=bounds)
if update:
img_current.save(ref_path / f'shade_{bounds}.png')
else:
img_reference = Image.open(ref_path / f'shade_{bounds}.png')
diff = ImageChops.difference(img_reference.convert('RGB'),
img_current.convert('RGB'))
assert not diff.getbbox()
def test_compare_reference(self, format, ext, tmp_path, ref_path, update):
name = 'binary'
c = Colormap.from_predefined(name) # noqa
if update:
os.chdir(ref_path)
eval(f'c.save_{format}()')
else:
os.chdir(tmp_path)
eval(f'c.save_{format}()')
time.sleep(.5)
assert filecmp.cmp(tmp_path / (name + ext),
ref_path / (name + ext))
def test_repr(self, patch_plt_show):
print(Colormap.from_predefined('stress'))
def test_at_value(self, N, cmap, at, result):
assert np.allclose(Colormap.from_predefined(cmap, N=N).at(at)[..., :3],
result,
rtol=0.005)
def test_mul(self):
c = o = Colormap.from_predefined('jet')
o *= 2
assert c + c == o
def test_add(self):
c = Colormap.from_predefined('jet')
c += c
assert (np.allclose(c.colors[:len(c.colors) // 2],
c.colors[len(c.colors) // 2:]))
def test_invert(self):
c_1 = Colormap.from_predefined('strain')
c_2 = ~c_1
assert (not np.allclose(c_1.colors, c_2.colors)) and \
np.allclose(c_1.colors,(~c_2).colors)
def test_reversed(self):
c_1 = Colormap.from_predefined('stress')
c_2 = c_1.reversed()
assert (not np.allclose(c_1.colors,c_2.colors)) and \
np.allclose(c_1.colors,c_2.reversed().colors)
def test_from_predefined(self, name, format, tmp_path):
N = np.random.randint(2, 256)
c = Colormap.from_predefined(name, N) # noqa
os.chdir(tmp_path)
eval(f'c.save_{format}()')