diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/modeltests/lookup/tests.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tests/modeltests/lookup/tests.py b/tests/modeltests/lookup/tests.py index cf18a83945..3f40bf10f5 100644 --- a/tests/modeltests/lookup/tests.py +++ b/tests/modeltests/lookup/tests.py @@ -115,7 +115,8 @@ class LookupTests(TestCase): self.assertEqual(Article.objects.in_bulk((self.a3.id,)), {self.a3.id: self.a3}) self.assertEqual(Article.objects.in_bulk([1000]), {}) self.assertEqual(Article.objects.in_bulk([]), {}) - self.assertRaises(AssertionError, Article.objects.in_bulk, 'foo') + self.assertEqual(Article.objects.in_bulk(iter([self.a1.id])), {self.a1.id: self.a1}) + self.assertEqual(Article.objects.in_bulk(iter([])), {}) self.assertRaises(TypeError, Article.objects.in_bulk) self.assertRaises(TypeError, Article.objects.in_bulk, headline__startswith='Blah') |