def test_accum_add_get():
values = np.array([1.5])
acc = Accumulator(values, 10)
assert acc is not None
assert acc.size == 0
assert acc.peek() < 0
assert acc.remove() < 0
acc.add(0)
assert acc.size == 1
assert acc.peek() == 0
assert acc.remove() == 0
assert acc.size == 0
assert acc.peek() == -1
def test_accum_init_empty():
values = np.empty(0)
acc = Accumulator(values, 10)
assert acc is not None
assert acc.size == 0
assert acc.peek() < 0
assert acc.remove() < 0
assert len(acc.top_keys()) == 0
def test_accum_add_a_few():
values = np.array([1.5, 2, -1])
acc = Accumulator(values, 10)
assert acc is not None
assert acc.size == 0
acc.add(1)
acc.add(0)
acc.add(2)
assert acc.size == 3
assert acc.peek() == 2
assert acc.remove() == 2
assert acc.size == 2
assert acc.remove() == 0
assert acc.remove() == 1
assert acc.size == 0