gte/tests/test_rpnsymbol.py

import unittest

from gte.turmit import Turmit
from gte.rpnlib import _op_list, RpnSymbol

class RpnSymbolTestCase(unittest.TestCase):
    
    def test_operations(self):
        add_idx = list(_op_list.keys()).index('add')

        sym = RpnSymbol(add_idx, RpnSymbol.OPERATION)
        self.assertEqual(str(sym), 'ADD')

    def test_value(self):
        sym = RpnSymbol(42, RpnSymbol.VALUE)
        self.assertEqual(str(sym), '0x2A')

    def test_op(self):
        sym = RpnSymbol(0, RpnSymbol.OPERATION)
        self.assertEqual(str(sym), 'MEM_SZ')