def test_fit_string(self): domain = CategoricalDomain() domain = domain.fit(numpy.array(["1", None, "3", "2", None, "2"])) self.assertEqual(numpy.array(["1", "2", "3"]).tolist(), domain.data_.tolist())
def test_fit_int(self): domain = CategoricalDomain() self.assertEqual("return_invalid", domain.invalid_value_treatment) self.assertFalse(hasattr(domain, "data_")) domain = domain.fit(numpy.array([1, 3, 2, 2])) self.assertEqual(numpy.array([1, 2, 3]).tolist(), domain.data_.tolist())