def test_to_val(self):
string_var = StringVariable("x")
self.assertEqual(string_var.to_val("foo"), "foo")
self.assertEqual(string_var.to_val(42), "42")
cont_var = ContinuousVariable("x")
self.assertTrue(math.isnan(cont_var.to_val("?")))
self.assertTrue(math.isnan(Unknown))
var = Variable("x")
self.assertEqual(var.to_val("x"), "x")
def test_exclusive(self):
var = Variable("x")
colors = [Mock(), Mock()]
palette = Mock()
var.colors = colors
var.palette = palette
self.assertIsNone(var.colors)
self.assertTrue("palette" in var.attributes)
self.assertFalse("colors" in var.attributes)
var.colors = colors
self.assertIsNone(var.palette)
self.assertTrue("colors" in var.attributes)
self.assertFalse("palette" in var.attributes)
def test_exclusive(self):
var = Variable("x")
colors = [Mock(), Mock()]
palette = Mock()
var.colors = colors
# set_color for variable does not set this attribute; derived methods do
var.attributes["colors"] = colors
var.palette = palette
self.assertIsNone(var.colors)
self.assertTrue("palette" in var.attributes)
self.assertFalse("colors" in var.attributes)
var.colors = colors
# set_color for variable does not set this attribute; derived methods do
var.attributes["colors"] = colors
self.assertIsNone(var.palette)
self.assertTrue("colors" in var.attributes)
self.assertFalse("palette" in var.attributes)
def test_palette(self):
var = Variable("x")
palette = Mock()
var.palette = palette
self.assertIs(var.palette, palette)
def test_colors(self):
var = Variable("x")
colors = [Mock(), Mock()]
var.colors = colors
self.assertIs(var.colors, colors)
def setUpClass(cls):
cls.var = Variable("x")
def setUp(self):
self.var = Variable("x")