fix: obey mypy (#25416)

PostHog · Oct 6, 2024 · f529a78 · f529a78
1 parent f7eaea6
commit f529a78
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/posthog/hogql_queries/test/test_events_query_runner.py b/posthog/hogql_queries/test/test_events_query_runner.py
@@ -131,7 +131,9 @@ def test_person_id_expands_to_distinct_ids(self):
         query_ast = EventsQueryRunner(query=query, team=self.team).to_query()
         where_expr = cast(ast.CompareOperation, cast(ast.And, query_ast.where).exprs[0])
         right_expr = cast(ast.Tuple, where_expr.right)
-        self.assertEqual([x.args[0].value for x in right_expr.exprs], ["id1", "id2"])
+        self.assertEqual(
+            [cast(ast.Constant, cast(ast.Call, x).args[0]).value for x in right_expr.exprs], ["id1", "id2"]
+        )
 
         # another team
         another_team = Team.objects.create(organization=Organization.objects.create())