def test_find_with_regex_filters_results_from_walk_using_regex_nonlazy(exists): ('Node#find_with_regex returns an evaluated list of nodes') nd = Node('/foo/bar') nd.walk = Mock() nd.walk.return_value = [ "/foo/wisdom/aaa.py", "/foo/wisdom/bbb.txt", "/foo/wisdom/ccc.php", "/foo/wisdom/ddd.py", ] ret = nd.find_with_regex('[.]\w{3}$', lazy=False) ret.should.be.a(list) ret.should.equal([ Node("/foo/wisdom/bbb.txt"), Node("/foo/wisdom/ccc.php"), ]) nd.walk.assert_called_once_with(lazy=False)
def test_find_with_regex_filters_results_from_walk_using_regex(exists): ('Node#find_with_regex returns a lazy list of nodes') nd = Node('/foo/bar') nd.walk = Mock() nd.walk.return_value = [ "/foo/wisdom/aaa.py", "/foo/wisdom/bbb.txt", "/foo/wisdom/ccc.php", "/foo/wisdom/ddd.py", ] ret = nd.find_with_regex('[.]\w{3}$', lazy='passed-to-walk') ret.should.be.a('types.GeneratorType') list(ret).should.equal([ Node("/foo/wisdom/bbb.txt"), Node("/foo/wisdom/ccc.php"), ]) nd.walk.assert_called_once_with(lazy='passed-to-walk')