Function naming

This commit is contained in:
Gui Heise 2021-09-29 09:29:27 -04:00
parent d348490ce5
commit f48d373cf3

View File

@ -33,7 +33,7 @@ def test_single_weth_liquidation():
classified_traces = trace_classifier.classify(block.traces) classified_traces = trace_classifier.classify(block.traces)
result = get_liquidations(classified_traces) result = get_liquidations(classified_traces)
_assert_n_equals(result, liquidations) _assert_actual_and_expected_are_equal(result, liquidations)
def test_single_liquidation(): def test_single_liquidation():
@ -62,7 +62,7 @@ def test_single_liquidation():
classified_traces = trace_classifier.classify(block.traces) classified_traces = trace_classifier.classify(block.traces)
result = get_liquidations(classified_traces) result = get_liquidations(classified_traces)
_assert_n_equals(result, liquidations) _assert_actual_and_expected_are_equal(result, liquidations)
def test_multiple_liquidations_in_block(): def test_multiple_liquidations_in_block():
@ -114,9 +114,11 @@ def test_multiple_liquidations_in_block():
result = get_liquidations(classified_traces) result = get_liquidations(classified_traces)
liquidations = [liquidation1, liquidation2, liquidation3] liquidations = [liquidation1, liquidation2, liquidation3]
_assert_n_equals(result, liquidations) _assert_actual_and_expected_are_equal(result, liquidations)
def _assert_n_equals(result: List[Liquidation], liquidations: List[Liquidation]): def _assert_actual_and_expected_are_equal(
for i in range(len(liquidations)): actual_liquidations: List[Liquidation], expected_liquidations: List[Liquidation]
assert result[i] == liquidations[i] ):
for i in range(len(actual_liquidations)):
assert actual_liquidations[i] == expected_liquidations[i]