def test_dummy_coding_zero_based(self):
var = DiscreteVariable("foo", values=list("abc"))
varb, varc = owcontinuize.dummy_coding(var)
self.assertEqual(varb.name, "foo=b")
self.assertIsInstance(varb.compute_value, transformation.Indicator)
self.assertEqual(varb.compute_value.value, 1)
self.assertIs(varb.compute_value.variable, var)
self.assertEqual(varc.name, "foo=c")
self.assertIsInstance(varc.compute_value, transformation.Indicator)
self.assertEqual(varc.compute_value.value, 2)
self.assertIs(varc.compute_value.variable, var)
varb, varc = owcontinuize.dummy_coding(var, zero_based=False)
self.assertEqual(varb.name, "foo=b")
self.assertIsInstance(varb.compute_value, transformation.Indicator1)
self.assertEqual(varb.compute_value.value, 1)
self.assertIs(varb.compute_value.variable, var)
self.assertEqual(varc.name, "foo=c")
self.assertIsInstance(varc.compute_value, transformation.Indicator1)
self.assertEqual(varc.compute_value.value, 2)
self.assertIs(varb.compute_value.variable, var)
def test_dummy_coding_base_value(self):
var = DiscreteVariable("foo", values=tuple("abc"))
varb, varc = owcontinuize.dummy_coding(var, base_value=0)
self.assertEqual(varb.name, "foo=b")
self.assertIsInstance(varb.compute_value, transformation.Indicator)
self.assertEqual(varb.compute_value.value, 1)
self.assertEqual(varc.name, "foo=c")
self.assertIsInstance(varc.compute_value, transformation.Indicator)
self.assertEqual(varc.compute_value.value, 2)
varb, varc = owcontinuize.dummy_coding(var, base_value=1)
self.assertEqual(varb.name, "foo=a")
self.assertIsInstance(varb.compute_value, transformation.Indicator)
self.assertEqual(varb.compute_value.value, 0)
self.assertEqual(varc.name, "foo=c")
self.assertIsInstance(varc.compute_value, transformation.Indicator)
self.assertEqual(varc.compute_value.value, 2)
