def test_conversion_size(self): domain = Domain([age, gender, income], [race]) self.assertRaises(ValueError, domain.convert, [0] * 3) self.assertRaises(ValueError, domain.convert, [0] * 5) domain = Domain([age, income], [race], [gender, education, ssn]) self.assertRaises(ValueError, domain.convert, [0] * 2) self.assertRaises(ValueError, domain.convert, [0] * 4) self.assertRaises(ValueError, domain.convert, [0] * 7) domain.convert([0] * 3) domain.convert([0] * 6)
def test_conversion(self): domain = Domain([age, income], [race], [gender, education, ssn]) values, metas = domain.convert([42, 13, "White"]) assert_array_equal(values, np.array([42, 13, 0])) assert_array_equal(metas, np.array([Unknown, Unknown, None])) values, metas = domain.convert([42, 13, "White", "M", "HS", "1234567"]) assert_array_equal(values, np.array([42, 13, 0])) assert_array_equal(metas, np.array([0, 1, "1234567"], dtype=object))
def test_conversion(self): domain = Domain([age, income], [race], [gender, education, ssn]) x, y, metas = domain.convert([42, 13, "White"]) assert_array_equal(x, np.array([42, 13])) assert_array_equal(y, np.array([0])) self.assertTrue(all(np.isnan(np.array(metas, dtype=float)))) x, y, metas = domain.convert([42, 13, "White", "M", "HS", "1234567"]) assert_array_equal(x, np.array([42, 13])) assert_array_equal(y, np.array([0])) assert_array_equal(metas, np.array([0, 1, "1234567"], dtype=object))
def test_conversion(self): domain = Domain([age, income], [race], [gender, education, ssn]) x, y, metas = domain.convert([42, 13, "White"]) assert_array_equal(x, np.array([42, 13])) assert_array_equal(y, np.array([0])) metas_exp = [gender.Unknown, education.Unknown, ssn.Unknown] def eq(a, b): if isinstance(a, Real) and isinstance(b, Real) and \ np.isnan(a) and np.isnan(b): return True else: return a == b self.assertTrue(all(starmap(eq, zip(metas, metas_exp)))) x, y, metas = domain.convert([42, 13, "White", "M", "HS", "1234567"]) assert_array_equal(x, np.array([42, 13])) assert_array_equal(y, np.array([0])) assert_array_equal(metas, np.array([0, 1, "1234567"], dtype=object))