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)
