Added average to QuerySet

This commit is contained in:
Harry Marr
2009-12-30 16:31:33 +00:00
parent 30d4a0379f
commit e9254f471f
2 changed files with 31 additions and 0 deletions

View File

@@ -220,6 +220,19 @@ class QuerySetTest(unittest.TestCase):
BlogPost.drop_collection()
def test_average(self):
"""Ensure that field can be averaged correctly.
"""
ages = [23, 54, 12, 94, 27]
for i, age in enumerate(ages):
self.Person(name='test%s' % i, age=age).save()
avg = float(sum(ages)) / len(ages)
self.assertAlmostEqual(int(self.Person.objects.average('age')), avg)
self.Person(name='ageless person').save()
self.assertEqual(int(self.Person.objects.average('age')), avg)
def test_sum(self):
"""Ensure that field can be summed over correctly.
"""