diff --git a/tests/instructions/extcodehash_test.py b/tests/instructions/extcodehash_test.py index e4586285..c677d5ef 100644 --- a/tests/instructions/extcodehash_test.py +++ b/tests/instructions/extcodehash_test.py @@ -11,12 +11,13 @@ from mythril.support.support_utils import get_code_hash from mythril.laser.smt import symbol_factory + def test_extcodehash_concrete(): # Arrange world_state = WorldState() account = world_state.create_account(balance=10, address=101) account.code = Disassembly("60606040") - world_state.create_account(balance = 10, address = 1000) + world_state.create_account(balance=10, address=1000) environment = Environment(account, None, None, None, None, None) og_state = GlobalState( world_state, environment, None, MachineState(gas_limit=8000000) @@ -24,9 +25,9 @@ def test_extcodehash_concrete(): og_state.transaction_stack.append( (MessageCallTransaction(world_state=WorldState(), gas_limit=8000000), None) ) - + instruction = Instruction("extcodehash", dynamic_loader=None) - + # If account does not exist, return 0 og_state.mstate.stack = [symbol_factory.BitVecVal(1, 256)] new_state = instruction.evaluate(og_state)[0] @@ -35,9 +36,12 @@ def test_extcodehash_concrete(): # If account code does not exist, return hash of empty set. og_state.mstate.stack = [symbol_factory.BitVecVal(1000, 256)] new_state = instruction.evaluate(og_state)[0] - assert new_state.mstate.stack[-1] == '0xc5d2460186f7233c927e7db2dcc703c0e500b653ca82273b7bfad8045d85a470' + assert ( + new_state.mstate.stack[-1] + == "0xc5d2460186f7233c927e7db2dcc703c0e500b653ca82273b7bfad8045d85a470" + ) # If account code exists, return hash of the code. og_state.mstate.stack = [symbol_factory.BitVecVal(101, 256)] new_state = instruction.evaluate(og_state)[0] - assert new_state.mstate.stack[-1] == get_code_hash("60606040") \ No newline at end of file + assert new_state.mstate.stack[-1] == get_code_hash("60606040")