def test_complete_attributes(self):
""" Test preset complete function
"""
colorspace = REC709
outlutfile = os.path.join(self.tmp_dir, "default_ascii_1D.lut")
default_preset = ASCII_HELPER.get_default_preset()
cust_preset = {}
cust_preset = ASCII_HELPER.complete_preset(cust_preset)
expression = set(default_preset).issubset(set(cust_preset))
self.assert_(expression,
("Something went wrong in preset completion :\n"
"Completed preset:\n{0}\nDefault one:\n{1}"
).format(cust_preset, default_preset))
ASCII_HELPER.check_preset(cust_preset)
# try to write a float ascii lut without forcing float mode
cust_preset[presets.IN_RANGE] = [0, 1.0]
self.failUnlessRaises(PresetException, ASCII_HELPER.write_1d_lut,
colorspace.decode_gradation,
outlutfile,
cust_preset)
# force float mode
cust_preset[presets.IS_FLOAT] = True
ASCII_HELPER.write_1d_lut(colorspace.decode_gradation,
outlutfile,
cust_preset)
def test_ascii_lut(self):
""" Test ascii 1D / 2D export
"""
colorspace = REC709
# 2D LUT
outlutfile = os.path.join(self.tmp_dir, "default_2D.lut")
preset = ASCII_HELPER.get_default_preset()
ASCII_HELPER.write_2d_lut(colorspace.decode_gradation,
outlutfile,
preset)
# 1D LUT
outlutfile = os.path.join(self.tmp_dir, "default_1D.lut")
preset = ASCII_HELPER.get_default_preset()
ASCII_HELPER.write_1d_lut(colorspace.decode_gradation,
outlutfile,
preset)
# test out bit depth inadequate with output range
preset[OUT_BITDEPTH] = 12
self.failUnlessRaises(AsciiHelperException, ASCII_HELPER.write_1d_lut,
colorspace.decode_gradation, outlutfile, preset)
