def test_consolidate():
c = TheCount()
c.tally("january", 2)
c.tally("february", 3)
c.tally("march", 1)
c.tally("may", 5)
c.tally("june", 4)
c.tally("july", 9)
c.tally("august", 14)
c.tally("september", 8)
c.tally("october", 5)
c.tally("november", 1)
d = c.consolidate({
"winter": ["december", "january", "february", "march"],
"spring": ["march", "april", "may", "june"],
"summer": ["june", "july", "august", "september"],
"autumn": ["september", "october", "november", "december"],
# Ignore a key if there are 0 occurrences
"ignore": ["fooooo"]
})
assert set(d.keys()) == set(["winter", "spring", "summer", "autumn"])
assert d.get("winter") == 6
assert d.get("spring") == 10
assert d.get("summer") == 35
assert d.get("autumn") == 14
def test_total_returns_number_of_occurrences_when_key_was_tallied():
c = TheCount()
times = random.randint(3, 10)
for _ in xrange(times):
c.tally("foo")
assert c.total("foo") == times
def test_get_returns_0_by_default():
c = TheCount()
times = random.randint(3, 10)
for _ in xrange(times):
c.tally("foo")
assert c.get("foo") == times
assert c.get("bar") == 0
def test_total_returns_0_if_key_was_never_tallied():
c = TheCount()
times = random.randint(3, 10)
for _ in xrange(times):
c.tally("foo")
assert c.total("foo") == times
assert c.total("bar") == 0
def test_clear_resets_the_sums():
c = TheCount()
c.tally("foo")
c.tally("bar")
assert len(c) == 2
c.clear()
assert len(c) == 0
def test_grand_total_sums_everything_up():
c = TheCount()
for _ in xrange(5):
c.tally("foo")
for _ in xrange(2):
c.tally("bar")
for _ in xrange(4):
c.tally(153)
assert c.grand_total() == 11
def test_iter_methods_work_as_expected():
c = TheCount()
c.tally("foo")
c.tally("bar")
c.tally("foo")
c.tally(4.2)
c.tally((1, 2, 3))
assert set(iter(c)) == set(["foo", "bar", 4.2, (1, 2, 3)])
assert set(c.iterkeys()) == set(["foo", "bar", 4.2, (1, 2, 3)])
assert set(c.itervalues()) == set([2, 1])
assert set(c.iteritems()) == set([("foo", 2), ("bar", 1), (4.2, 1),
((1, 2, 3), 1)])
def test_counts_can_be_aggregated():
dracula = TheCount()
dracula.tally("one")
dracula.tally(2)
dracula.tally(2)
dracula.tally("THREE")
chocula = TheCount()
chocula.tally("one")
chocula.tally(2)
chocula.tally("THREE")
chocula.tally(4.0)
chocula.tally(4.0)
orlok = dracula + chocula
assert set(orlok.keys()) == set(["one", 2, "THREE", 4.0])
assert orlok["one"] == 2
assert orlok[2] == 3
assert orlok["THREE"] == 2
assert orlok[4.0] == 2
def test_total_returns_the_sum_of_the_specified_keys():
c = TheCount()
foo_times = random.randint(3, 10)
for _ in xrange(foo_times):
c.tally("foo")
bar_times = random.randint(3, 10)
for _ in xrange(bar_times):
c.tally("bar")
assert c.total("foo") == foo_times
assert c.total("bar") == bar_times
assert c.total("foo", "bar") == foo_times + bar_times
assert c.total("foo", "bar", "quux") == foo_times + bar_times
def test_len_returns_number_of_unique_keys():
c = TheCount()
for _ in xrange(3):
c.tally("foo")
assert len(c) == 1
for _ in xrange(2):
c.tally("bar")
assert len(c) == 2
for _ in xrange(2):
c.tally("foo")
assert len(c) == 2
def test_reject_keeps_iteritems_such_that_func_returns_something_falsey():
c = TheCount()
for _ in xrange(6):
c.tally("apple")
for _ in xrange(2):
c.tally("banana")
for _ in xrange(10):
c.tally("carrot")
for _ in xrange(3):
c.tally(42)
d = c.reject(lambda k, v: v > 3)
assert set(d.keys()) == set(["banana", 42])
assert d["banana"] == 2
assert d[42] == 3
def test_select_keeps_iteritems_such_that_func_returns_something_truthy():
c = TheCount()
for _ in xrange(6):
c.tally("apple")
for _ in xrange(2):
c.tally("banana")
for _ in xrange(10):
c.tally("carrot")
for _ in xrange(3):
c.tally(42)
d = c.select(lambda k, v: v > 3)
assert set(d.keys()) == set(["apple", "carrot"])
assert d["apple"] == 6
assert d["carrot"] == 10
def test_select_and_reject_do_not_modify_the_original_TheCount_instance():
c = TheCount()
for _ in xrange(6):
c.tally("apple")
for _ in xrange(2):
c.tally("banana")
for _ in xrange(10):
c.tally("carrot")
for _ in xrange(3):
c.tally(42)
d = c.select(lambda k, v: v > 3)
assert set(c.keys()) == set(["apple", "banana", "carrot", 42])
assert set(d.keys()) == set(["apple", "carrot"])
e = c.reject(lambda k, v: v > 3)
assert set(c.keys()) == set(["apple", "banana", "carrot", 42])
assert set(e.keys()) == set(["banana", 42])
def test_tally_counts_stuff():
c = TheCount()
assert set(c.keys()) == set()
for _ in xrange(3):
c.tally("foo")
assert set(c.keys()) == set(["foo"])
assert c["foo"] == 3
for _ in xrange(2):
c.tally(10, 2)
assert set(c.keys()) == set(["foo", 10])
assert c["foo"] == 3
assert c[10] == 4
for _ in xrange(2):
c.tally("foo")
assert set(c.keys()) == set(["foo", 10])
assert c["foo"] == 5
assert c[10] == 4
def test_total_returns_0_when_no_keys_are_requested():
c = TheCount()
assert c.total() == 0