Skip to content

Commit c693a90

Browse files
Using generics types + enabling ruff
1 parent 0d7c4de commit c693a90

File tree

2 files changed

+11
-2
lines changed

2 files changed

+11
-2
lines changed

tests/functional/event_handler/required_dependencies/appsync/test_appsync_batch_resolvers.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,19 @@
11
from __future__ import annotations
22

3+
from typing import TYPE_CHECKING
4+
35
import pytest
46

57
from aws_lambda_powertools.event_handler import AppSyncResolver
68
from aws_lambda_powertools.event_handler.graphql_appsync.exceptions import InvalidBatchResponse, ResolverNotFoundError
79
from aws_lambda_powertools.event_handler.graphql_appsync.router import Router
8-
from aws_lambda_powertools.utilities.data_classes import AppSyncResolverEvent
910
from aws_lambda_powertools.utilities.typing import LambdaContext
1011
from aws_lambda_powertools.warnings import PowertoolsUserWarning
1112
from tests.functional.utils import load_event
1213

14+
if TYPE_CHECKING:
15+
from aws_lambda_powertools.utilities.data_classes import AppSyncResolverEvent
16+
1317

1418
# TESTS RECEIVING THE EVENT PARTIALLY AND PROCESS EACH RECORD PER TIME.
1519
def test_resolve_batch_processing_with_related_events_one_at_time():

tests/functional/event_handler/required_dependencies/test_api_middlewares.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,7 @@
11
from __future__ import annotations
22

3+
from typing import TYPE_CHECKING
4+
35
import pytest
46

57
from aws_lambda_powertools.event_handler import content_types
@@ -20,9 +22,12 @@
2022
from aws_lambda_powertools.event_handler.middlewares.schema_validation import (
2123
SchemaValidationMiddleware,
2224
)
23-
from aws_lambda_powertools.event_handler.types import EventHandlerInstance
2425
from tests.functional.utils import load_event
2526

27+
if TYPE_CHECKING:
28+
from aws_lambda_powertools.event_handler.types import EventHandlerInstance
29+
30+
2631
API_REST_EVENT = load_event("apiGatewayProxyEvent.json")
2732
API_RESTV2_EVENT = load_event("apiGatewayProxyV2Event_GET.json")
2833

0 commit comments

Comments
 (0)