From daa208f72be171927f04702fbaa7f1d4ac480d88 Mon Sep 17 00:00:00 2001 From: Gui Heise Date: Tue, 17 Aug 2021 12:57:16 -0400 Subject: [PATCH] Enforce data typpe for helpers --- tests/helpers.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/tests/helpers.py b/tests/helpers.py index 0ebcf0f..0f3677e 100644 --- a/tests/helpers.py +++ b/tests/helpers.py @@ -1,7 +1,11 @@ from typing import List from mev_inspect.schemas.blocks import TraceType -from mev_inspect.schemas.classified_traces import Classification, ClassifiedTrace +from mev_inspect.schemas.classified_traces import ( + Classification, + ClassifiedTrace, + ClassifiedCallData, +) def make_transfer_trace( @@ -13,7 +17,7 @@ def make_transfer_trace( token_address: str, amount: int, ): - return ClassifiedTrace( + return ClassifiedCallData( transaction_hash=transaction_hash, block_number=block_number, trace_type=TraceType.call, @@ -38,7 +42,7 @@ def make_swap_trace( recipient_address: str, recipient_input_key: str, ): - return ClassifiedTrace( + return ClassifiedCallData( transaction_hash=transaction_hash, block_number=block_number, trace_type=TraceType.call,