Additional test/assertions for subq in expr.

coleifer · Dec 14, 2018 · 9ed66f6 · 9ed66f6
1 parent fb41bfc
commit 9ed66f6
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/models.py b/tests/models.py
@@ -997,6 +997,13 @@ def test_subquery_in_select_expression(self):
             (10, 10, 30),
             (10, 20, 30)])
 
+        query = (Point
+                 .select(Point.x, (Point.y + subq).alias('sy'))
+                 .order_by(Point.x, Point.y))
+        self.assertEqual(list(query.tuples()), [
+            (1, 4), (1, 5),
+            (10, 40), (10, 50)])
+
     @requires_models(User, Tweet)
     def test_filtering(self):
         with self.database.atomic():