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)