def test_choice_datetimes():
dates = ['05/29/1988', '06/22/1988', '07/30/1992', '01/02/2001',
'11/12/2000', '07/09/2002', '08/30/1998', '06/03/1997',
'10/22/2002', '12/03/2001']
attr = Attribute(Series(dates, name='DateTime'))
choices = attr.choice()
assert len(choices) == len(dates)
def test_choice_floats():
floats = random.uniform(1, 10, size)
attr = Attribute(Series(floats, name='Float'))
assert len(attr.bins) == 20
choices = attr.choice()
assert len(choices) == size
def test_choice_strings():
strings = list(map(lambda x: randomize_string(5), range(size)))
attr = Attribute(Series(strings, name='String'))
choices = attr.choice()
assert len(choices) == size
def test_choice_integers():
ints = random.randint(1, 100, size)
attr = Attribute(Series(ints, name='Integer'))
assert len(attr.bins) == 20
choices = attr.choice()
assert len(choices) == size
