def test_trip_at_when_not_lazy_relative(os): ("Node#trip_at(path, lazy=False) returns a list when lazy=False " "(testing with relative path)") os.walk.return_value = [ ("/foo/bar/somewhere", [], ["file1.py", "file2.py"]), ] nd = Node("/foo/bar/") nd.trip_at('somewhere', lazy=False).should.equal([ "/foo/bar/somewhere/file1.py", "/foo/bar/somewhere/file2.py", ])
def test_trip_at_when_not_lazy_absolute(os): ("Node#trip_at(path, lazy=False) returns a list when lazy=False " "(testing with absolute path)") os.walk.return_value = [ ("/dummy/", [], ["file1.py", "file2.py"]), ] nd = Node("/foo/bar/") nd.trip_at('/dummy/', lazy=False).should.equal([ "/dummy/file1.py", "/dummy/file2.py", ])
def test_trip_at_when_lazy_relative(os): ("Node#trip_at(path, lazy=True) returns a generator when lazy=True " "(testing with relative path)") os.walk.return_value = [ ("/dummy/", [], ["file1.py", "file2.py"]), ] nd = Node("/foo/bar/") nd.trip_at('/dummy/', lazy=True).should.be.a('types.GeneratorType') list(nd.trip_at('/dummy/', lazy=True)).should.equal([ "/dummy/file1.py", "/dummy/file2.py", ])
def test_walk_trips_at_node_path(): ("Node#walk() trips at node.path") nd = Node("/foo/bar/") nd.trip_at = Mock() nd.walk() nd.walk(lazy=True) nd.trip_at.assert_has_calls([ call('/foo/bar', lazy=False), call('/foo/bar', lazy=True), ])