diff options
Diffstat (limited to 'tests/dates')
| -rw-r--r-- | tests/dates/tests.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tests/dates/tests.py b/tests/dates/tests.py index 82f21a2634..9f12d997e0 100644 --- a/tests/dates/tests.py +++ b/tests/dates/tests.py @@ -98,11 +98,12 @@ class DatesTests(TestCase): def test_dates_fails_when_given_invalid_kind_argument(self): msg = "'kind' must be one of 'year', 'month', 'week', or 'day'." - with self.assertRaisesMessage(AssertionError, msg): + with self.assertRaisesMessage(ValueError, msg): Article.objects.dates("pub_date", "bad_kind") def test_dates_fails_when_given_invalid_order_argument(self): - with self.assertRaisesMessage(AssertionError, "'order' must be either 'ASC' or 'DESC'."): + msg = "'order' must be either 'ASC' or 'DESC'." + with self.assertRaisesMessage(ValueError, msg): Article.objects.dates("pub_date", "year", order="bad order") @override_settings(USE_TZ=False) |