def testStorageV2(self):
state = b'\x00\x00\x00\x10\x00C\x00o\x00l\x00o\x00r\x00m\x00a\x00p\x00'\
b'\x00\x00\x02\x00\x00\x00\x0e\x00v\x00i\x00r\x00i\x00d\x00i\x00'\
b's\x00\x00\x00\x00\x06\x00?\xf0\x00\x00\x00\x00\x00\x00\x00\x00'\
b'\x00\x00\x06\x00@\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x06'\
b'\x00l\x00o\x00g\x00\x00\x00\x0c\x00m\x00i\x00n\x00m\x00a\x00x'
state = qt.QByteArray(state)
colormap = Colormap()
colormap.restoreState(state)
expected = Colormap(name="viridis", vmin=1, vmax=2, normalization=Colormap.LOGARITHM)
expected.setGammaNormalizationParameter(1.5)
self.assertEqual(colormap, expected)
def testStoreRestore(self):
colormaps = [
Colormap(name="viridis"),
Colormap(normalization=Colormap.SQRT)
]
cmap = Colormap(normalization=Colormap.GAMMA)
cmap.setGammaNormalizationParameter(1.2)
cmap.setNaNColor('red')
colormaps.append(cmap)
for expected in colormaps:
with self.subTest(colormap=expected):
state = expected.saveState()
result = Colormap()
result.restoreState(state)
self.assertEqual(expected, result)
