def test_read():
assert read(StringIO("one\ntwo\r\nthree\rfour\n")) == [
"one",
"two",
"three",
"four",
]
def test_read():
assert extract(read(StringIO("one\ntwo\r\nthree\rfour\n"))) == [
"one",
"two",
"three",
"four",
]
def time_search(number_of_lines, line_fn, term, max_results):
results = []
for _ in range(NUMBER_OF_MEASUREMENTS):
stdin = StringIO.StringIO("\n".join([
line_fn(index)
for index
in range(number_of_lines)
]))
t1 = timeit.default_timer()
list(itertools.islice(search(read(stdin), term), max_results))
t2 = timeit.default_timer()
results.append(t2 - t1)
return min(results)
def test_read_skips_duplicates():
assert read(StringIO("dup\ndup")) == [
"dup",
]
def test_read_skips_duplicates():
assert extract(read(StringIO("dup\ndup"))) == [
"dup",
]
