def test_3dl_preset(self): """ Test 3dl preset """ preset = presets.get_default_preset() # test type must be 3D self.failUnlessRaises(presets.PresetException, THREEDL_HELPER.check_preset, preset ) preset[presets.TYPE] = '3D' # test shaper attr exists self.failUnlessRaises(presets.PresetException, THREEDL_HELPER.check_preset, preset ) preset[SHAPER] = True # test mesh attr exists self.failUnlessRaises(presets.PresetException, THREEDL_HELPER.check_preset, preset ) preset[MESH] = True # test preset is ok THREEDL_HELPER.check_preset(preset) # test ranges are int outlutfile = os.path.join(self.tmp_dir, "test.3dl") self.failUnlessRaises(PresetException, THREEDL_HELPER.write_3d_lut, self.processor_3d.applyRGB, outlutfile, preset)
def test_check_attributes(self): """ Test preset check function """ outlutfile = os.path.join(self.tmp_dir, "test.cube") default_preset = presets.get_default_preset() CUBE_HELPER.check_preset(default_preset) # test missing attr cust_preset = {} self.failUnlessRaises(presets.PresetException, CUBE_HELPER.check_preset, cust_preset) for attr in presets.BASIC_ATTRS: cust_preset[attr] = default_preset[attr] self.failUnlessRaises(presets.PresetException, CUBE_HELPER.check_preset, cust_preset) ## test specific attr # change type to 1D cust_preset[presets.TYPE] = '1D' self.failUnlessRaises(presets.PresetException, CUBE_HELPER.check_preset, cust_preset) cust_preset[presets.OUT_BITDEPTH] = 12 CUBE_HELPER.check_preset(cust_preset) # try to write a 3D LUT with a 1D preset self.failUnlessRaises(alh.AbstractLUTException, CUBE_HELPER.write_3d_lut, self.processor_1d, outlutfile, cust_preset) # change type to 2D cust_preset[presets.TYPE] = '3D' self.failUnlessRaises(presets.PresetException, CUBE_HELPER.check_preset, cust_preset) cust_preset[presets.CUBE_SIZE] = 17 CUBE_HELPER.check_preset(cust_preset) # try to write a 1D LUT with a 3D preset self.failUnlessRaises(alh.AbstractLUTException, CUBE_HELPER.write_1d_lut, self.processor_1d, outlutfile, cust_preset) # # test value type # cube size cust_preset[presets.CUBE_SIZE] = presets.CUBE_SIZE_MAX_VALUE + 1 self.failUnlessRaises(presets.PresetException, CUBE_HELPER.check_preset, cust_preset) cust_preset[presets.CUBE_SIZE] = default_preset[presets.CUBE_SIZE] # range tests = 'test', ['a', 'a'], [0.0, 0.5, 1.0], 0.1 for test in tests: cust_preset[presets.IN_RANGE] = test self.failUnlessRaises(presets.PresetException, CUBE_HELPER.check_preset, cust_preset) cust_preset[presets.IN_RANGE] = 0.1, 1 CUBE_HELPER.check_preset(cust_preset) cust_preset[presets.IN_RANGE] = (0.1, 1) CUBE_HELPER.check_preset(cust_preset)