def test_counts_datetimes(): 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='DateTime'), categorical=True) counts = attr.counts(normalize=False) assert sum(counts) == len(dates) assert array_equal(counts, [1, 1, 2, 1, 2, 1, 1, 1]) counts = attr.counts(bins=['12/03/2001', '10/22/2002'], normalize=False) assert array_equal(counts, [1, 1])
def test_counts_numerical_attribute(): ints = random.randint(1, 100, size) attr = Attribute(Series(ints, name='Integer')) counts = attr.counts(normalize=False) assert sum(counts) == 30 assert len(counts) == 20 counts = attr.counts(bins=[0, 10, 20, 30, 100], normalize=False) assert sum(counts) == 30 assert len(counts) == 4 # categorical ints attr = Attribute(Series([1, 10, 11, 10, 20, 15, 16, 25], name='Integer'), categorical=True) counts = attr.counts(normalize=False) assert sum(counts) == 8 assert len(counts) == 7 counts = attr.counts(bins=[5, 10, 15], normalize=False) assert sum(counts) == 3 assert len(counts) == 3
def test_counts_categorical_attribute(): ints = random.randint(1, 10, size) attr = Attribute(Series(ints, name='Integer'), categorical=True) assert sum(attr.counts()) == 30