Apply suggestions from code review

Co-authored-by: Mario Vega <[email protected]>
ipsilon · Sep 26, 2024 · b3ad2ff · b3ad2ff
1 parent 912fc8e
commit b3ad2ff
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/tests/prague/eip7692_eof_v1/eip7069_extcall/test_calls.py b/tests/prague/eip7692_eof_v1/eip7069_extcall/test_calls.py
@@ -632,16 +632,17 @@ def test_callee_context(
  value=tx_value,
  )
 
+ expected_bytes: Address | int
  if expected_result == "destination":
- expected_bytes: bytes = destination_contract_address
+ expected_bytes = destination_contract_address
  elif expected_result == "caller":
  expected_bytes = calling_contract_address
  elif expected_result == "sender":
  expected_bytes = sender
  elif expected_result == "tx_value":
- expected_bytes = to_fixed_size_bytes(tx_value, 32)
+ expected_bytes = tx_value
  elif isinstance(expected_result, int):
- expected_bytes = to_fixed_size_bytes(expected_result, 32)
+ expected_bytes = expected_result
  else:
  raise TypeError("Unexpected expected_result", expected_result)
 

diff --git a/tests/prague/eip7692_eof_v1/eip7620_eof_create/test_eofcreate.py b/tests/prague/eip7692_eof_v1/eip7620_eof_create/test_eofcreate.py
@@ -640,14 +640,15 @@ def test_eofcreate_context(
 
  tx = Transaction(sender=sender, to=calling_contract_address, gas_limit=1000000, value=value)
 
+ expected_bytes: Address | int
  if expected_result == "destination":
- expected_bytes: bytes = destination_contract_address
+ expected_bytes = destination_contract_address
  elif expected_result == "caller":
  expected_bytes = calling_contract_address
  elif expected_result == "sender":
  expected_bytes = sender
  elif expected_result == "eofcreate_value":
- expected_bytes = to_fixed_size_bytes(eofcreate_value, 32)
+ expected_bytes = eofcreate_value
  else:
  raise TypeError("Unexpected expected_result", expected_result)