def test_set_domain_for_datetime_attribute(): dates = ['05/29/1988', '06/22/1988', '07/30/1992', '07/30/1992', '11/12/2000', '01/02/2001', '01/02/2001', '12/03/2001', '07/09/2002', '10/22/2002'] attr = Attribute(Series(dates, name='String'), categorical=True) bins = attr.bins attr.domain = ['07/01/1997', '12/20/1999', '01/01/2004'] assert len(bins) + 3 == len(attr.bins)
def test_set_domain_for_integer_categorical_attribute(): ints = random.randint(1, 100, size) attr = Attribute(Series(ints, name='Integer'), categorical=True) assert attr.bins[0] >= 1 assert attr.bins[-1] <= 100 attr.domain = [-2, 120] assert attr.bins[0] == -2 assert attr.bins[-1] == 120
def test_set_domain_for_float_attribute(): floats = random.uniform(1, 100, size) attr = Attribute(Series(floats, name='Float')) assert attr._min >= 1 assert attr._max <= 100 attr.domain = [-2, 120] assert attr._min == -2 assert attr._max == 120
def test_set_domain_for_integer_attribute(): ints = random.randint(1, 100, size) attr = Attribute(Series(ints, name='Integer')) assert attr._min >= 1 assert attr._max <= 100 attr.domain = [-2, 120] assert attr._min == -2 assert attr._max == 120
def test_set_domain_for_string_attribute(): strings = list(map(lambda x: randomize_string(5), range(size))) attr = Attribute(Series(strings, name='String'), categorical=True) bins = attr.bins attr.domain = ['a', 'b', 'China', 'USA'] assert len(bins) + 4 == len(attr.bins)