Added tests for aggregates with Django

Author: ankane

tests/test_django.py

@@ -2,6 +2,7 @@
 from django.conf import settings
 from django.core import serializers
 from django.db import connection, migrations, models
+from django.db.models import Avg, Sum
 from django.db.migrations.loader import MigrationLoader
 from django.forms import ModelForm
 from math import sqrt
@@ -132,6 +133,22 @@ def test_filter(self):
         items = Item.objects.alias(distance=distance).filter(distance__lt=1)
         assert [v.id for v in items] == [1]
 
+    def test_avg(self):
+        avg = Item.objects.aggregate(Avg('embedding'))['embedding__avg']
+        assert avg is None
+        Item(embedding=[1, 2, 3]).save()
+        Item(embedding=[4, 5, 6]).save()
+        avg = Item.objects.aggregate(Avg('embedding'))['embedding__avg']
+        assert np.array_equal(avg, np.array([2.5, 3.5, 4.5]))
+
+    def test_sum(self):
+        avg = Item.objects.aggregate(Sum('embedding'))['embedding__sum']
+        assert avg is None
+        Item(embedding=[1, 2, 3]).save()
+        Item(embedding=[4, 5, 6]).save()
+        avg = Item.objects.aggregate(Sum('embedding'))['embedding__sum']
+        assert np.array_equal(avg, np.array([5, 7, 9]))
+
     def test_serialization(self):
         create_items()
         items = Item.objects.all()