From 96ead6cc929110569888901f6949d4b29cf150d5 Mon Sep 17 00:00:00 2001 From: Luke Van Seters Date: Thu, 12 Aug 2021 12:04:56 -0400 Subject: [PATCH] Rename remove_inner_transfers => remove_child_transfers_of_transfers --- mev_inspect/swaps.py | 6 +++--- mev_inspect/transfers.py | 2 +- tests/test_transfers.py | 6 +++--- 3 files changed, 7 insertions(+), 7 deletions(-) diff --git a/mev_inspect/swaps.py b/mev_inspect/swaps.py index 94bf9c5..437dfdd 100644 --- a/mev_inspect/swaps.py +++ b/mev_inspect/swaps.py @@ -10,7 +10,7 @@ from mev_inspect.schemas.transfers import Transfer from mev_inspect.transfers import ( get_child_transfers, filter_transfers, - remove_inner_transfers, + remove_child_transfers_of_transfers, ) @@ -52,8 +52,8 @@ def _get_swaps_for_transaction(traces: List[ClassifiedTrace]) -> List[Swap]: swap = _parse_swap( trace, - remove_inner_transfers(prior_transfers), - remove_inner_transfers(child_transfers), + remove_child_transfers_of_transfers(prior_transfers), + remove_child_transfers_of_transfers(child_transfers), ) if swap is not None: diff --git a/mev_inspect/transfers.py b/mev_inspect/transfers.py index 991e3fe..4237063 100644 --- a/mev_inspect/transfers.py +++ b/mev_inspect/transfers.py @@ -38,7 +38,7 @@ def filter_transfers( return filtered_transfers -def remove_inner_transfers(transfers: List[Transfer]) -> List[Transfer]: +def remove_child_transfers_of_transfers(transfers: List[Transfer]) -> List[Transfer]: updated_transfers = [] transfer_addresses_by_transaction: Dict[str, List[List[int]]] = {} diff --git a/tests/test_transfers.py b/tests/test_transfers.py index d65a753..cca856d 100644 --- a/tests/test_transfers.py +++ b/tests/test_transfers.py @@ -1,8 +1,8 @@ from mev_inspect.schemas.transfers import Transfer -from mev_inspect.transfers import remove_inner_transfers +from mev_inspect.transfers import remove_child_transfers_of_transfers -def test_remove_inner_transfers(get_transaction_hashes, get_addresses): +def test_remove_child_transfers_of_transfers(get_transaction_hashes, get_addresses): [transaction_hash, other_transaction_hash] = get_transaction_hashes(2) [ @@ -61,7 +61,7 @@ def test_remove_inner_transfers(get_transaction_hashes, get_addresses): separate_transaction_transfer, ] - removed_transfers = remove_inner_transfers(transfers) + removed_transfers = remove_child_transfers_of_transfers(transfers) assert _equal_ignoring_order(removed_transfers, expected_transfers)