array

PostHog · Apr 18, 2023 · 0136a50 · 0136a50
1 parent 9232749
commit 0136a50
Showing 1 changed file with 14 additions and 12 deletions.
diff --git a/posthog/hogql/test/test_resolver.py b/posthog/hogql/test/test_resolver.py
@@ -631,15 +631,17 @@ def test_resolve_union_all(self):
     def test_call_type(self):
         node = parse_select("select max(timestamp) from events")
         resolve_types(node, self.database)
-        expected = ast.Call(
-            name="max",
-            # NB! timestamp was resolved to a DateTimeType for the Call's arg type.
-            type=ast.CallType(name="max", arg_types=[ast.DateTimeType()], return_type=ast.UnknownType()),
-            args=[
-                ast.Field(
-                    chain=["timestamp"],
-                    type=ast.FieldType(name="timestamp", table_type=ast.TableType(table=self.database.events)),
-                )
-            ],
-        )
-        self.assertEqual(node.select[0], expected)
+        expected = [
+            ast.Call(
+                name="max",
+                # NB! timestamp was resolved to a DateTimeType for the Call's arg type.
+                type=ast.CallType(name="max", arg_types=[ast.DateTimeType()], return_type=ast.UnknownType()),
+                args=[
+                    ast.Field(
+                        chain=["timestamp"],
+                        type=ast.FieldType(name="timestamp", table_type=ast.TableType(table=self.database.events)),
+                    )
+                ],
+            ),
+        ]
+        self.assertEqual(node.select, expected)