import unittest

from vm.chunk import Chunk
from vm.value import Value
from vm.instruction import OpCode
from vm.machine import VM


class VMTestCase(unittest.TestCase):
    def test_chunk_basic(self):
        chk = Chunk('test chunk')
        chk.add_constant(Value.new_number(1.2), 1)
        chk.write(OpCode.POP, 2)
        chk.disassemble()

    def test_vm_basic(self):
        chk = Chunk('test chunk')
        chk.add_constant(Value.new_number(1.2), 1)
        chk.add_constant(Value.new_number(3.4), 1)
        chk.write(OpCode.ADD, 1)
        chk.add_constant(Value.new_number(5.6), 1)
        chk.write(OpCode.DIVIDE, 1)
        chk.write(OpCode.NEGATE, 1)
        chk.write(OpCode.RETURN, 1)

        m = VM(chk)

        m.run()



if __name__ == '__main__':
    unittest.main()