diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/modeltests/many_to_many/tests.py | 5 | ||||
| -rw-r--r-- | tests/modeltests/many_to_one/tests.py | 5 |
2 files changed, 6 insertions, 4 deletions
diff --git a/tests/modeltests/many_to_many/tests.py b/tests/modeltests/many_to_many/tests.py index b00d7da140..9fa524f993 100644 --- a/tests/modeltests/many_to_many/tests.py +++ b/tests/modeltests/many_to_many/tests.py @@ -1,4 +1,4 @@ -from __future__ import absolute_import +from __future__ import absolute_import, with_statement from django.test import TestCase @@ -52,7 +52,8 @@ class ManyToManyTests(TestCase): ]) # Adding an object of the wrong type raises TypeError - self.assertRaises(TypeError, a6.publications.add, a5) + with self.assertRaisesRegexp(TypeError, "'Publication' instance expected, got <Article.*"): + a6.publications.add(a5) # Add a Publication directly via publications.add by using keyword arguments. p4 = a6.publications.create(title='Highlights for Adults') self.assertQuerysetEqual(a6.publications.all(), diff --git a/tests/modeltests/many_to_one/tests.py b/tests/modeltests/many_to_one/tests.py index 922506eac0..bc9fe64be4 100644 --- a/tests/modeltests/many_to_one/tests.py +++ b/tests/modeltests/many_to_one/tests.py @@ -1,4 +1,4 @@ -from __future__ import absolute_import +from __future__ import absolute_import, with_statement from copy import deepcopy from datetime import datetime @@ -68,7 +68,8 @@ class ManyToOneTests(TestCase): self.assertQuerysetEqual(self.r2.article_set.all(), ["<Article: Paul's story>"]) # Adding an object of the wrong type raises TypeError. - self.assertRaises(TypeError, self.r.article_set.add, self.r2) + with self.assertRaisesRegexp(TypeError, "'Article' instance expected, got <Reporter.*"): + self.r.article_set.add(self.r2) self.assertQuerysetEqual(self.r.article_set.all(), [ "<Article: John's second story>", |