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", ]