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