def test_save_low(self): descA = owcolor.DiscAttrDesc( DiscreteVariable("varA", values=tuple("abc"))) descA.name = "a2" descB = owcolor.DiscAttrDesc( DiscreteVariable("varB", values=tuple("abc"))) descB.set_value(1, "X") descC = owcolor.ContAttrDesc(ContinuousVariable("varC")) descC.name = "c2" descD = owcolor.ContAttrDesc(ContinuousVariable("varD")) descD.palette_name = "linear_viridis" descE = owcolor.ContAttrDesc(ContinuousVariable("varE")) self.widget.disc_descs = [descA, descB] self.widget.cont_descs = [descC, descD, descE] with tempfile.TemporaryDirectory() as path: fname = os.path.join(path, "foo.colors") self.widget._save_var_defs(fname) with open(fname) as f: js = json.load(f) self.assertEqual(js["categorical"], {"varA": {"rename": "a2"}, "varB": {"renamed_values": {"b": "X"}}}) self.assertEqual(js["numeric"], {"varC": {"rename": "c2"}, "varD": {"colors": "linear_viridis"}})
def _create_descs(self): disc_vars = [DiscreteVariable(f"var{c}", values=("a", "b", "c")) for c in "AB"] cont_vars = [ContinuousVariable(f"var{c}") for c in "CDE"] self.widget.disc_descs = [owcolor.DiscAttrDesc(v) for v in disc_vars] self.widget.cont_descs = [owcolor.ContAttrDesc(v) for v in cont_vars] return disc_vars, cont_vars
def setUp(self): z = ContinuousVariable("z") w = ContinuousVariable("w") u = ContinuousVariable("u") self.descs = [owcolor.ContAttrDesc(v) for v in (z, w, u)] self.model = owcolor.ContColorTableModel() self.model.set_data(self.descs) self.table = owcolor.ContinuousTable(self.model)
def test_to_dict(self): x = ContinuousVariable("x") desc = owcolor.ContAttrDesc(x) self.assertEqual(desc.to_dict(), {}) desc2, warns = owcolor.ContAttrDesc.from_dict(x, desc.to_dict()) self.assertEqual(warns, []) self.assertIsNone(desc2.new_name) self.assertIsNone(desc2.new_palette_name) desc.name = "y" self.assertEqual(desc.to_dict(), {"rename": "y"}) desc2, warns = owcolor.ContAttrDesc.from_dict(x, desc.to_dict()) self.assertEqual(warns, []) self.assertEqual(desc2.new_name, "y") self.assertIsNone(desc2.new_palette_name) desc = owcolor.ContAttrDesc(x) desc.palette_name = "linear_viridis" self.assertEqual(desc.to_dict(), {"colors": "linear_viridis"})
def setUp(self): x = ContinuousVariable("x") self.desc = owcolor.ContAttrDesc(x)