def test_calls(self): call_stack = CallStack(self.fake) c = Call(self.fake) c.return_val = 1 call_stack.add_call(c) c = Call(self.fake) c.return_val = 2 call_stack.add_call(c) eq_(call_stack(), 1) eq_(call_stack(), 2)
def test_with_initial_calls(self): c = Call(self.fake) c.return_val = 1 call_stack = CallStack(self.fake, initial_calls=[c]) eq_(call_stack(), 1)