def test_window_gap(): w = Window(STREAM, before=1) w.seek() print(list(w.iter_docs())) assert sorted(STREAM[0][1] + STREAM[1][1]) == sorted(w.iter_docs()) assert w.seek() assert sorted(STREAM[1][1] + STREAM[2][1]) == sorted(w.iter_docs()) assert not w.seek()
def test_window(): w = Window(STREAM) w.seek() assert STREAM[0][1] == list(w.iter_docs()) print(list(w.iter_docs())) w.seek() assert STREAM[1][1] == list(w.iter_docs()) print(list(w.iter_docs()))
def test_empty_seek(): w = Window(STREAM, before=1) assert w.seek(datetime.date(2014, 1, 4)) print(list(w.iter_docs())) assert list(w.iter_docs()) print(w.date) assert not w.seek() print(w.date) print(list(w.iter_docs())) assert not list(w.iter_docs())