def test_k_shingles_generator_empty_doc_in_generator_yield_None(): # set up size = 4 faux_generator = generator_empty() expected_results = None # execute actual_results = next(shgl._k_shingles_generator(faux_generator, size)) # assert nt.eq_(actual_results, expected_results)
def test_k_shingles_generator(): # set up size = 4 faux_generator = generator_string() expected_results = get_faux_list_of_k_shingles(), "abcd efghij cklmn op." # execute actual_results = next(shgl._k_shingles_generator(faux_generator, size)) # assert #sort as the list of shingles produced maybe in different order than mocked list nt.eq_(actual_results[0].sort(), expected_results[0].sort()) nt.eq_(actual_results[1], expected_results[1])