diff options
Diffstat (limited to 'tests/test_template_spec.py')
-rw-r--r-- | tests/test_template_spec.py | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/tests/test_template_spec.py b/tests/test_template_spec.py index b3d7093..5bb8210 100644 --- a/tests/test_template_spec.py +++ b/tests/test_template_spec.py @@ -30,7 +30,7 @@ class Thing(object): pass -class ViewTestCase(unittest.TestCase): +class ViewTestCase(unittest.TestCase, AssertStringMixin): def test_init(self): """ @@ -119,8 +119,9 @@ class ViewTestCase(unittest.TestCase): self.assertEquals(view.render(), 'nopqrstuvwxyznopqrstuvwxyz') def test_inverted(self): - view = Inverted() - self.assertEquals(view.render(), """one, two, three, empty list""") + renderer = Renderer() + expected = renderer.render(Inverted()) + self.assertString(expected, u"""one, two, three, empty list""") def test_accessing_properties_on_parent_object_from_child_objects(self): parent = Thing() @@ -132,8 +133,9 @@ class ViewTestCase(unittest.TestCase): self.assertEquals(view.render(), 'derp') def test_inverted_lists(self): - view = InvertedLists() - self.assertEquals(view.render(), """one, two, three, empty list""") + renderer = Renderer() + expected = renderer.render(InvertedLists()) + self.assertString(expected, u"""one, two, three, empty list""") class SpecLoaderTests(unittest.TestCase, AssertIsMixin, AssertStringMixin): |