def test_iter_query_string(): assert iter_query("lolwut") == ["lolwut"]
def test_iter_query_file(tmpdir): filename = str(tmpdir.join('test.txt')) with open(filename, 'w') as f: f.write("lolwut") assert iter_query(filename) == ["lolwut"]