ref: slight improvement to auto generated types

Author: asottile-sentry

tests/sentry/issues/endpoints/test_organization_group_suspect_flags.py

@@ -1,10 +1,16 @@
 import datetime
 import time
 import uuid
+from typing import TypedDict
 
 from sentry.testutils.cases import APITestCase, SnubaTestCase
 
 
+class _FlagResult(TypedDict):
+    flag: str
+    result: bool
+
+
 class OrganizationGroupSuspectFlagsTestCase(APITestCase, SnubaTestCase):
     endpoint = "sentry-api-0-organization-group-suspect-flags"
 
@@ -73,7 +79,7 @@ def _mock_event(
         hash: str = "a" * 32,
         group_id: int | None = None,
         project_id: int = 1,
-        flags: list[dict[str, object]] | None = None,
+        flags: list[_FlagResult] | None = None,
     ) -> None:
         self.snuba_insert(
             (

tests/sentry/issues/test_suspect_flags.py

@@ -1,6 +1,7 @@
 import datetime
 import time
 import uuid
+from typing import TypedDict
 
 from sentry.issues.suspect_flags import (
     get_suspect_flag_scores,
@@ -10,14 +11,19 @@
 from sentry.testutils.cases import SnubaTestCase, TestCase
 
 
+class _FlagResult(TypedDict):
+    flag: str
+    result: bool
+
+
 class SnubaTest(TestCase, SnubaTestCase):
     def mock_event(
         self,
         ts: datetime.datetime,
         hash: str = "a" * 32,
         group_id: int | None = None,
         project_id: int = 1,
-        flags: list[dict[str, object]] | None = None,
+        flags: list[_FlagResult] | None = None,
     ) -> None:
         self.snuba_insert(
             (