def test_push_pop(self): """ Covers Context.push(), Context.pop(), Context.top() """ ctx = Context(self) Context.push(ctx) self.assertIs(ctx, Context.top()) Context.pop(ctx)
def test_raises_unbalanced_context(self): """ Raises if context push and pop calls are not balanced. """ ctx1 = Context(self) ctx2 = Context(self) Context.push(ctx1) Context.push(ctx2) self.assertRaises(AssertionError, Context.pop, ctx1)