diff options
| author | Curtis Maloney <curtis@tinbrain.net> | 2013-07-16 21:11:32 +1000 |
|---|---|---|
| committer | Tim Graham <timograham@gmail.com> | 2013-07-17 13:32:32 -0400 |
| commit | a3e7d73ed7d90d31de46c60d40424267f62e411c (patch) | |
| tree | c14e422ce6b5e4669b059d5f5ef111322f5840a7 /tests/template_tests/test_context.py | |
| parent | 828359e52dd8f190a577442a88517e1ba9ba8da0 (diff) | |
Allowed Context.push to behave as a context mananger.
Thanks Loic Bistuer for the review.
Diffstat (limited to 'tests/template_tests/test_context.py')
| -rw-r--r-- | tests/template_tests/test_context.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/tests/template_tests/test_context.py b/tests/template_tests/test_context.py index 224b94d060..ca167a73f3 100644 --- a/tests/template_tests/test_context.py +++ b/tests/template_tests/test_context.py @@ -16,3 +16,12 @@ class ContextTests(TestCase): self.assertEqual(c.pop(), {"a": 2}) self.assertEqual(c["a"], 1) self.assertEqual(c.get("foo", 42), 42) + + with c.push(): + c['a'] = 2 + self.assertEqual(c['a'], 2) + self.assertEqual(c['a'], 1) + + with c.push(a=3): + self.assertEqual(c['a'], 3) + self.assertEqual(c['a'], 1) |