def test_insert_after_current_seen_again():
test = [1, 2]
it = ListIterator(test)
it.next()
it.next()
it.insert_after_current([42, 1337])
assert it.next()
assert it.current() == 42
assert it.next()
assert it.current() == 1337
def test_insert_current():
test_list = [42, 1337]
it = ListIterator(test_list)
it.next()
it.next()
it.insert_before([1, 1, 2])
assert it.current() == 1337
def test_insert_previous():
test_list = [42, 1337]
it = ListIterator(test_list)
it.next()
it.next()
it.insert_before([1, 3, 5])
assert it.previous() == 5
def test_iteration_end():
test_list = [1, 2, 3]
it = ListIterator(test_list)
it.next()
it.next()
it.next()
assert not it.next()
def test_insert_offset():
test_list = [42, 1337]
it = ListIterator(test_list)
it.next()
it.next()
it.insert_before([1, 3, 5], 1)
assert all([a == b for a, b in zip(test_list, [1, 3, 5, 42, 1337])])
def test_insert_after_current_end():
test = [1, 2]
it = ListIterator(test)
it.next()
it.next()
it.insert_after_current([42, 1337])
assert test == [1, 2, 42, 1337]
def test_previous_value():
test_list = [1, 2]
it = ListIterator(test_list)
it.next()
it.next()
assert it.previous() == 1
def test_no_has_previous():
test_list = [1]
it = ListIterator(test_list)
assert not it.has_previous()
def test_has_previous():
test_list = [1, 2]
it = ListIterator(test_list)
it.next()
it.next()
assert it.has_previous()
def test_empty_list_no_next():
test_list = []
it = ListIterator(test_list)
assert not it.next()
def test_peek_mid():
test = [1, 2, 3]
it = ListIterator(test)
it.next()
assert it.peek() == 2
def test_current_index():
test = [1, 2]
it = ListIterator(test)
it.next()
assert it.current_index() == 0
def test_insert_after_current_empty():
test = []
it = ListIterator(test)
it.insert_after_current([1, 2])
assert test == [1, 2]
def test_peek_two():
test = [1, 2]
it = ListIterator(test)
assert it.peek(2) == 2
def test_peek_no_more():
test = [1]
it = ListIterator(test)
it.next()
with pytest.raises(AssertionError):
it.peek()
def test_iteration():
test_list = [1, 2, 3]
it = ListIterator(test_list)
for i in test_list:
it.next()
assert it.current() == i
def test_can_peek(test_list, result):
it = ListIterator(test_list)
assert it.can_peek() == result
def test_peek_first():
test = [1]
it = ListIterator(test)
assert it.peek() == 1