Commit 4b8a6952 by Chris Jerdonek

Changed the test_context.TestCase class to a mixin class.

parent 3e284008
...@@ -12,9 +12,9 @@ from pystache.context import _get_item ...@@ -12,9 +12,9 @@ from pystache.context import _get_item
from pystache.context import Context from pystache.context import Context
class TestCase(unittest.TestCase): class AssertIsMixin:
"""A TestCase class with support for assertIs().""" """A mixin for adding assertIs() to a unittest.TestCase."""
# unittest.assertIs() is not available until Python 2.7: # unittest.assertIs() is not available until Python 2.7:
# http://docs.python.org/library/unittest.html#unittest.TestCase.assertIsNone # http://docs.python.org/library/unittest.html#unittest.TestCase.assertIsNone
...@@ -48,7 +48,7 @@ class MappingObject(object): ...@@ -48,7 +48,7 @@ class MappingObject(object):
return self._dict[key] return self._dict[key]
class GetItemTestCase(TestCase): class GetItemTestCase(unittest.TestCase, AssertIsMixin):
"""Test context._get_item().""" """Test context._get_item()."""
...@@ -168,7 +168,7 @@ class GetItemTestCase(TestCase): ...@@ -168,7 +168,7 @@ class GetItemTestCase(TestCase):
self.assertRaises(AttributeError, _get_item, obj, "foo") self.assertRaises(AttributeError, _get_item, obj, "foo")
class ContextTests(TestCase): class ContextTests(unittest.TestCase, AssertIsMixin):
""" """
Test the Context class. Test the Context class.
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment