Add test for opcodes

Relates to #1111

Author: cburgdorf

tests/core/opcodes/test_opcodes.py

@@ -0,0 +1,100 @@
+import pytest
+
+from eth_utils import (
+    to_canonical_address,
+)
+
+from eth import (
+    constants
+)
+from eth.vm import (
+    opcode_values
+)
+from eth.vm.forks.byzantium.opcodes import (
+    BYZANTIUM_OPCODES
+)
+from eth.vm.forks import (
+    ByzantiumVM,
+    SpuriousDragonVM,
+    TangerineWhistleVM,
+    HomesteadVM,
+    FrontierVM,
+)
+
+from eth.vm.message import (
+    Message,
+)
+
+
+NORMALIZED_ADDRESS_A = "0x0f572e5295c57f15886f9b263e2f6d2d6c7b5ec6"
+NORMALIZED_ADDRESS_B = "0xcd1722f3947def4cf144679da39c4c32bdc35681"
+CANONICAL_ADDRESS_A = to_canonical_address("0x0f572e5295c57f15886f9b263e2f6d2d6c7b5ec6")
+CANONICAL_ADDRESS_B = to_canonical_address("0xcd1722f3947def4cf144679da39c4c32bdc35681")
+
+
+def prepare_computation(vm_class):
+
+    message = Message(
+        to=CANONICAL_ADDRESS_A,
+        sender=CANONICAL_ADDRESS_B,
+        value=100,
+        data=b'',
+        code=b'',
+        gas=800,
+    )
+
+    tx_context = vm_class._state_class.transaction_context_class(
+        gas_price=1,
+        origin=CANONICAL_ADDRESS_B,
+    )
+
+    computation = vm_class._state_class.computation_class(
+        state=None,
+        message=message,
+        transaction_context=tx_context,
+    )
+    return computation
+
+
+@pytest.mark.parametrize(
+    'vm_class, val1, val2, expected',
+    (
+        (ByzantiumVM, 2, 4, 6,),
+        (SpuriousDragonVM, 2, 4, 6,),
+        (TangerineWhistleVM, 2, 4, 6,),
+        (HomesteadVM, 2, 4, 6,),
+        (FrontierVM, 2, 4, 6,),
+    )
+)
+def test_add(vm_class, val1, val2, expected):
+    computation = prepare_computation(vm_class)
+    computation.stack_push(val1)
+    computation.stack_push(val2)
+    computation.opcodes[opcode_values.ADD](computation)
+
+    result = computation.stack_pop(type_hint=constants.UINT256)
+
+    assert result == expected
+
+
+@pytest.mark.parametrize(
+    'vm_class, val1, val2, expected',
+    (
+        (ByzantiumVM, 2, 2, 4,),
+        (SpuriousDragonVM, 2, 2, 4,),
+        (TangerineWhistleVM, 2, 2, 4,),
+        (HomesteadVM, 2, 2, 4,),
+        (FrontierVM, 2, 2, 4,),
+    )
+)
+def test_mul(vm_class, val1, val2, expected):
+    computation = prepare_computation(vm_class)
+    computation.stack_push(val1)
+    computation.stack_push(val2)
+    computation.opcodes[opcode_values.MUL](computation)
+
+    result = computation.stack_pop(type_hint=constants.UINT256)
+
+    assert result == expected
+
+